Просмотр исходного кода

Renamed InstanceType to VMType and added missing list method to internet gateway

Nuwan Goonasekera 8 лет назад
Родитель
Сommit
c161041e37

+ 1 - 1
cloudbridge/cloud/providers/azure/azure_client.py

@@ -349,7 +349,7 @@ class AzureClient(object):
                                  'location': self.region_name
                              }).result()
 
-    def list_instance_types(self):
+    def list_vm_types(self):
         return self.compute_client.virtual_machine_sizes. \
             list(self.region_name)
 

+ 6 - 6
cloudbridge/cloud/providers/azure/resources.py

@@ -1157,7 +1157,7 @@ class AzureInstance(BaseInstance):
         'Creating': InstanceState.PENDING,
         'VM running': InstanceState.RUNNING,
         'Updating': InstanceState.CONFIGURING,
-        'Deleted': InstanceState.TERMINATED,
+        'Deleted': InstanceState.DELETED,
         'Stopping': InstanceState.CONFIGURING,
         'Deleting': InstanceState.CONFIGURING,
         'Stopped': InstanceState.STOPPED,
@@ -1260,19 +1260,19 @@ class AzureInstance(BaseInstance):
         return self._private_ips
 
     @property
-    def instance_type_id(self):
+    def vm_type_id(self):
         """
         Get the instance type name.
         """
         return self._vm.hardware_profile.vm_size
 
     @property
-    def instance_type(self):
+    def vm_type(self):
         """
         Get the instance type.
         """
-        return self._provider.compute.instance_types.find(
-            name=self.instance_type_id)[0]
+        return self._provider.compute.vm_types.find(
+            name=self.vm_type_id)[0]
 
     def reboot(self):
         """
@@ -1280,7 +1280,7 @@ class AzureInstance(BaseInstance):
         """
         self._provider.azure_client.restart_vm(self.id)
 
-    def terminate(self):
+    def delete(self):
         """
         Permanently terminate this instance.
         After deleting the VM. we are deleting the network interface

+ 12 - 8
cloudbridge/cloud/providers/azure/services.py

@@ -373,7 +373,7 @@ class AzureSnapshotService(BaseSnapshotService):
 class AzureComputeService(BaseComputeService):
     def __init__(self, provider):
         super(AzureComputeService, self).__init__(provider)
-        self._instance_type_svc = AzureVMTypeService(self.provider)
+        self._vm_type_svc = AzureVMTypeService(self.provider)
         self._instance_svc = AzureInstanceService(self.provider)
         self._region_svc = AzureRegionService(self.provider)
         self._images_svc = AzureImageService(self.provider)
@@ -383,8 +383,8 @@ class AzureComputeService(BaseComputeService):
         return self._images_svc
 
     @property
-    def instance_types(self):
-        return self._instance_type_svc
+    def vm_types(self):
+        return self._vm_type_svc
 
     @property
     def instances(self):
@@ -399,7 +399,7 @@ class AzureInstanceService(BaseInstanceService):
     def __init__(self, provider):
         super(AzureInstanceService, self).__init__(provider)
 
-    def create(self, name, image, instance_type, subnet=None, zone=None,
+    def create(self, name, image, vm_type, subnet=None, zone=None,
                key_pair=None, vm_firewalls=None, user_data=None,
                launch_config=None, **kwargs):
 
@@ -419,8 +419,8 @@ class AzureInstanceService(BaseInstanceService):
         image = (self.provider.compute.images.get(image)
                  if isinstance(image, str) else image)
 
-        instance_size = instance_type.id if \
-            isinstance(instance_type, VMType) else instance_type
+        instance_size = vm_type.id if \
+            isinstance(vm_type, VMType) else vm_type
 
         if not subnet:
             subnet = self.provider.networking.subnets.get_or_create_default()
@@ -712,7 +712,7 @@ class AzureVMTypeService(BaseVMTypeService):
         """
         Fetch info about the available instances.
         """
-        r = self.provider.azure_client.list_instance_types()
+        r = self.provider.azure_client.list_vm_types()
         return r
 
     def list(self, limit=None, marker=None):
@@ -1032,13 +1032,17 @@ class AzureRouterService(BaseRouterService):
 class AzureGatewayService(BaseGatewayService):
     def __init__(self, provider):
         super(AzureGatewayService, self).__init__(provider)
+        # Singleton returned by the list method
+        self.gateway_singleton = AzureInternetGateway(self.provider, None)
 
     def get_or_create_inet_gateway(self, name):
         AzureInternetGateway.assert_valid_resource_name(name)
-
         gateway = AzureInternetGateway(self.provider, None)
         gateway.name = name
         return gateway
 
+    def list(self, limit=None, marker=None):
+        return [self.gateway_singleton]
+
     def delete(self, gateway):
         pass