When listing datacenters/pods, mark those that are closing soon.

Author: caberos

SoftLayer/CLI/hardware/create_options.py

@@ -7,6 +7,7 @@
 from SoftLayer.CLI import formatting
 from SoftLayer.managers import account
 from SoftLayer.managers import hardware
+from SoftLayer.managers import network
 
 
 @click.command()
@@ -22,14 +23,39 @@ def cli(env, prices, location=None):
     account_manager = account.AccountManager(env.client)
     options = hardware_manager.get_create_options(location)
     routers = account_manager.get_routers(location=location)
+    network_manager = network.NetworkManager(env.client)
+
+    closing_filter = {
+        'capabilities': {
+            'operation': 'in',
+            'options': [{'name': 'data', 'value': ['CLOSURE_ANNOUNCED']}]
+        },
+        'name': {
+            'operation': 'orderBy',
+            'options': [{'name': 'sort', 'value': ['DESC']}]
+        }
+    }
+
+    pods_mask = """mask[name, datacenterLongName, frontendRouterId, capabilities, datacenterId, backendRouterId,
+        backendRouterName, frontendRouterName]"""
+    pods = network_manager.get_pods(mask=pods_mask, filter=closing_filter)
     tables = []
 
     # Datacenters
-    dc_table = formatting.Table(['Datacenter', 'Value'], title="Datacenters")
+    dc_table = formatting.Table(['Datacenter', 'Value', 'note'], title="Datacenters")
     dc_table.sortby = 'Value'
     dc_table.align = 'l'
     for location_info in options['locations']:
-        dc_table.add_row([location_info['name'], location_info['key']])
+        closure = []
+        for pod in pods:
+            if ((location_info['key'] in str(pod['name']))):
+                closure.append(pod['name'])
+
+        if len(closure) == 0:
+            closure = ''
+        else:
+            closure = 'closed soon: %s' % (str(closure))
+        dc_table.add_row([location_info['name'], location_info['key'], str(closure)])
     tables.append(dc_table)
 
     tables.append(_preset_prices_table(options['sizes'], prices))

SoftLayer/CLI/virt/create_options.py

@@ -2,6 +2,7 @@
 # :license: MIT, see LICENSE for more details.
 # pylint: disable=too-many-statements
 import click
+from SoftLayer.managers import network
 
 import SoftLayer
 from SoftLayer.CLI import environment
@@ -22,16 +23,41 @@ def cli(env, vsi_type, prices, location=None):
     """Virtual server order options."""
 
     vsi = SoftLayer.VSManager(env.client)
+    network_manager = network.NetworkManager(env.client)
     options = vsi.get_create_options(vsi_type, location)
 
+    closing_filter = {
+        'capabilities': {
+            'operation': 'in',
+            'options': [{'name': 'data', 'value': ['CLOSURE_ANNOUNCED']}]
+        },
+        'name': {
+            'operation': 'orderBy',
+            'options': [{'name': 'sort', 'value': ['DESC']}]
+        }
+    }
+
+    pods_mask = """mask[name, datacenterLongName, frontendRouterId, capabilities, datacenterId, backendRouterId,
+            backendRouterName, frontendRouterName]"""
+    pods = network_manager.get_pods(mask=pods_mask, filter=closing_filter)
+
     tables = []
 
     # Datacenters
-    dc_table = formatting.Table(['datacenter', 'Value'], title="Datacenters")
+    dc_table = formatting.Table(['Datacenter', 'Value', 'note'], title="Datacenters")
     dc_table.sortby = 'Value'
     dc_table.align = 'l'
     for location_info in options['locations']:
-        dc_table.add_row([location_info['name'], location_info['key']])
+        closure = []
+        for pod in pods:
+            if ((location_info['key'] in str(pod['name']))):
+                closure.append(pod['name'])
+
+        if len(closure) == 0:
+            closure = ''
+        else:
+            closure = 'closed soon: %s' % (str(closure))
+        dc_table.add_row([location_info['name'], location_info['key'], str(closure)])
     tables.append(dc_table)
 
     if vsi_type == 'CLOUD_SERVER':

SoftLayer/managers/network.py

@@ -779,16 +779,16 @@ def cancel_item(self, identifier, cancel_immediately,
                                 customer_note,
                                 id=identifier)
 
-    def get_pods(self, datacenter=None):
+    def get_pods(self, mask=None, filter=None, datacenter=None):
         """Calls SoftLayer_Network_Pod::getAllObjects()
 
         returns list of all network pods and their routers.
         """
-        _filter = None
+
         if datacenter:
-            _filter = {"datacenterName": {"operation": datacenter}}
+            filter = {"datacenterName": {"operation": datacenter}}
 
-        return self.client.call('SoftLayer_Network_Pod', 'getAllObjects', filter=_filter)
+        return self.client.call('SoftLayer_Network_Pod', 'getAllObjects', mask=mask, filter=filter)
 
     def get_list_datacenter(self):
         """Calls SoftLayer_Location::getDatacenters()