Преглед изворни кода

Move Network and Subnet under Networking

Also fix some errors introduced when changed InstanceType to VMType.
Ehsan Chiniforooshan пре 8 година
родитељ
комит
b1725efe2c
2 измењених фајлова са 27 додато и 15 уклоњено
  1. 4 4
      cloudbridge/cloud/providers/gce/provider.py
  2. 23 11
      cloudbridge/cloud/providers/gce/services.py

+ 4 - 4
cloudbridge/cloud/providers/gce/provider.py

@@ -16,7 +16,7 @@ from oauth2client.client import GoogleCredentials
 from oauth2client.service_account import ServiceAccountCredentials
 
 from .services import GCEComputeService
-from .services import GCENetworkService
+from .services import GCENetworkingService
 from .services import GCESecurityService
 from .services import GCPStorageService
 
@@ -196,7 +196,7 @@ class GCECloudProvider(BaseCloudProvider):
         # Initialize provider services
         self._compute = GCEComputeService(self)
         self._security = GCESecurityService(self)
-        self._network = GCENetworkService(self)
+        self._networking = GCENetworkingService(self)
         self._storage = GCPStorageService(self)
 
         self._compute_resources = GCPResources(self.gce_compute)
@@ -207,8 +207,8 @@ class GCECloudProvider(BaseCloudProvider):
         return self._compute
 
     @property
-    def network(self):
-        return self._network
+    def networking(self):
+        return self._networking
 
     @property
     def security(self):

+ 23 - 11
cloudbridge/cloud/providers/gce/services.py

@@ -11,6 +11,7 @@ from cloudbridge.cloud.base.services import BaseImageService
 from cloudbridge.cloud.base.services import BaseInstanceService
 from cloudbridge.cloud.base.services import BaseKeyPairService
 from cloudbridge.cloud.base.services import BaseNetworkService
+from cloudbridge.cloud.base.services import BaseNetworkingService
 from cloudbridge.cloud.base.services import BaseRegionService
 from cloudbridge.cloud.base.services import BaseSecurityService
 from cloudbridge.cloud.base.services import BaseSnapshotService
@@ -287,9 +288,9 @@ class GCEVMTypeService(BaseVMTypeService):
                         .execute())
         return response['items']
 
-    def get(self, instance_type_id):
+    def get(self, vm_type_id):
         for inst_type in self.instance_data:
-            if inst_type.get('id') == instance_type_id:
+            if inst_type.get('id') == vm_type_id:
                 return GCEVMType(self.provider, inst_type)
         return None
 
@@ -447,7 +448,7 @@ class GCEInstanceService(BaseInstanceService):
     def __init__(self, provider):
         super(GCEInstanceService, self).__init__(provider)
 
-    def create(self, name, image, instance_type, subnet, zone=None,
+    def create(self, name, image, vm_type, subnet, zone=None,
                key_pair=None, vm_firewalls=None, user_data=None,
                launch_config=None, **kwargs):
         """
@@ -464,7 +465,7 @@ class GCEInstanceService(BaseInstanceService):
                 network_url = 'global/networks/default'
             config = {
                 'name': name,
-                'machineType': instance_type.resource_url,
+                'machineType': vm_type.resource_url,
                 'disks': [{'boot': True,
                            'autoDelete': True,
                            'initializeParams': {
@@ -575,8 +576,8 @@ class GCEComputeService(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):
@@ -587,11 +588,26 @@ class GCEComputeService(BaseComputeService):
         return self._region_svc
 
 
+class GCENetworkingService(BaseNetworkingService):
+
+    def __init__(self, provider):
+        super(GCENetworkingService, self).__init__(provider)
+        self._network_service = GCENetworkService(self.provider)
+        self._subnet_service = GCESubnetService(self.provider)
+
+    @property
+    def networks(self):
+        return self._network_service
+
+    @property
+    def subnets(self):
+        return self._subnet_service
+
+
 class GCENetworkService(BaseNetworkService):
 
     def __init__(self, provider):
         super(GCENetworkService, self).__init__(provider)
-        self._subnet_svc = GCESubnetService(self.provider)
 
     def get(self, network_id):
         if network_id is None:
@@ -668,10 +684,6 @@ class GCENetworkService(BaseNetworkService):
     def get_or_create_default(self):
         return self._create(GCEFirewallsDelegate.DEFAULT_NETWORK, True)
 
-    @property
-    def subnets(self):
-        return self._subnet_svc
-
     def floating_ips(self, network_id=None, region=None):
         if not region:
             region = self.provider.region_name