added update_virtual_machine and bug fixes

Author: Thomas

docs/netbox.rst

@@ -0,0 +1,61 @@
+netbox package
+==============
+
+Submodules
+----------
+
+netbox\.connection module
+-------------------------
+
+.. automodule:: netbox.connection
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+netbox\.dcim module
+-------------------
+
+.. automodule:: netbox.dcim
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+netbox\.exceptions module
+-------------------------
+
+.. automodule:: netbox.exceptions
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+netbox\.ipam module
+-------------------
+
+.. automodule:: netbox.ipam
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+netbox\.netbox module
+---------------------
+
+.. automodule:: netbox.netbox
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+netbox\.virtualization module
+---------------------
+
+.. automodule:: netbox.virtualization
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: netbox
+    :members:
+    :undoc-members:
+    :show-inheritance:

netbox/virtualization.py

@@ -8,16 +8,20 @@ def get_clusters(self, **kwargs):
         """Return all clusters"""
         return self.netbox_con.get('/virtualization/clusters/', **kwargs)
 
-    # https://github.com/digitalocean/netbox/issues/1910
-    def create_cluster(self, name, type_id, **kwargs):
+    def create_cluster(self, name, type, **kwargs):
         """Create a new cluster
 
         :param name: name of the cluster
         :param type: cluster type
         :param kwargs: optional arguments
         :return: bool True if successful otherwise exception raised
         """
-        required_fields = {"name": name, "type": type_id}
+        try:
+            cluster_type_id = self.get_cluster_types(name=type)[0]['id']
+        except IndexError:
+            raise exceptions.NotFoundException('cluster-type: {}'.format(name)) from None
+
+        required_fields = {"name": name, "type": cluster_type_id}
         return self.netbox_con.post('/virtualization/clusters/', required_fields, **kwargs)
 
     def delete_cluster(self, name):
@@ -27,7 +31,7 @@ def delete_cluster(self, name):
         :return: bool True if succesful otherwase delete exception
         """
         try:
-            cluster_id = self.get_cluster(name=name)[0]['id']
+            cluster_id = self.get_clusters(name=name)[0]['id']
         except IndexError:
             raise exceptions.NotFoundException('cluster {}'.format(name)) from None
         return self.netbox_con.delete('/virtualization/clusters/', cluster_id)
@@ -40,7 +44,7 @@ def update_cluster(self, name, **kwargs):
         :return: bool True if successful otherwise raise UpdateException
         """
         try:
-            cluster_id = self.get_cluster(name=name)[0]['id']
+            cluster_id = self.get_clusters(name=name)[0]['id']
         except IndexError:
             raise exceptions.NotFoundException('cluster: {}'.format(name)) from None
         return self.netbox_con.patch('/virtualization/clusters/', cluster_id, **kwargs)
@@ -67,7 +71,7 @@ def update_cluster_type(self, name, **kwargs):
         :return: bool True if successful otherwise raise UpdateException
         """
         try:
-            type_id = self.get_cluster_type(name)[0]['id']
+            type_id = self.get_cluster_types(name=name)[0]['id']
         except IndexError:
             raise exceptions.NotFoundException(name) from None
         return self.netbox_con.patch('/virtualization/cluster-types/', type_id, **kwargs)
@@ -79,7 +83,7 @@ def delete_cluster_type(self, name):
         :return: bool True if succesful otherwase delete exception
         """
         try:
-            cluster_type_id = self.get_cluster_type(name=name)[0]['id']
+            cluster_type_id = self.get_cluster_types(name=name)[0]['id']
         except IndexError:
             raise exceptions.NotFoundException('cluster-type: {}'.format(name)) from None
         return self.netbox_con.delete('/virtualization/cluster-types/', cluster_type_id)
@@ -154,3 +158,17 @@ def delete_virtual_machine(self, virtual_machine_name):
         except IndexError:
             raise exceptions.NotFoundException('virtual-machine: {}'.format(virtual_machine_name)) from None
         return self.netbox_con.delete('/virtualization/virtual-machines/', virtual_machine_id)
+
+    def update_virtual_machine(self, virtual_machine_name, **kwargs):
+        """Update virtual-machine
+
+        :param name: name of the virtual-machine to update
+        :param kwargs: update data
+        :return: bool True if successful otherwise raise UpdateException
+        """
+        try:
+            virtual_machine_id = self.get_virtual_machines(name=virtual_machine_name)[0]['id']
+        except IndexError:
+            raise exceptions.NotFoundException('virtual-machine: {}'
+                                               .format(virtual_machine_name)) from None
+        return self.netbox_con.patch('/virtualization/virtual-machines/', virtual_machine_id, **kwargs)