#1575 added account bandwidth-pools and updated reports bandwidth to support the --pool option

Author: allmightyspiff

SoftLayer/CLI/account/bandwidth_pools.py

@@ -0,0 +1,43 @@
+"""Displays information about the accounts bandwidth pools"""
+# :license: MIT, see LICENSE for more details.
+import click
+
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import formatting
+from SoftLayer.managers.account import AccountManager as AccountManager
+from SoftLayer import utils
+
+from pprint import pprint as pp
+
+@click.command()
+@environment.pass_env
+def cli(env):
+    """Lists billing items with some other useful information.
+
+    Similiar to https://cloud.ibm.com/billing/billing-items
+    """
+
+    manager = AccountManager(env.client)
+    items = manager.get_bandwidth_pools()
+    # table = item_table(items)
+    pp(items)
+    table = formatting.Table([
+        "Pool Name",
+        "Region",
+        "Servers",
+        "Allocation",
+        "Current Usage",
+        "Projected Usage"
+    ], title="Bandwidth Pools")
+    table.align = 'l'
+
+    for item in items:
+        name = item.get('name')
+        region = utils.lookup(item, 'locationGroup', 'name')
+        servers = manager.get_bandwidth_pool_counts(identifier=item.get('id'))
+        allocation = item.get('totalBandwidthAllocated', 0)
+        current = item.get('billingCyclePublicUsageTotal', 0)
+        projected = item.get('projectedPublicBandwidthUsage', 0)
+
+        table.add_row([name, region, servers, allocation, current, projected,])
+    env.fout(table)

SoftLayer/CLI/report/bandwidth.py

@@ -9,6 +9,7 @@
 from SoftLayer.CLI import formatting
 from SoftLayer import utils
 
+from pprint import pprint as pp 
 
 # pylint: disable=unused-argument
 def _validate_datetime(ctx, param, value):
@@ -47,23 +48,25 @@ def _get_pooled_bandwidth(env, start, end):
                            label='Calculating for bandwidth pools',
                            file=sys.stderr) as pools:
         for pool in pools:
-            if not pool.get('metricTrackingObjectId'):
-                continue
-
-            yield {
-                'id': pool['id'],
+            pool_detail = {
+                'id': pool.get('id'),
                 'type': 'pool',
-                'name': pool['name'],
-                'data': env.client.call(
+                'name': pool.get('name'),
+                'data': []
+            }
+            if pool.get('metricTrackingObjectId'):
+                bw_data = env.client.call(
                     'Metric_Tracking_Object',
                     'getSummaryData',
                     start.strftime('%Y-%m-%d %H:%M:%S %Z'),
                     end.strftime('%Y-%m-%d %H:%M:%S %Z'),
                     types,
                     300,
-                    id=pool['metricTrackingObjectId'],
-                ),
-            }
+                    id=pool.get('metricTrackingObjectId'),
+                )
+                pool_detail['data'] = bw_data
+
+            yield pool_detail
 
 
 def _get_hardware_bandwidth(env, start, end):
@@ -172,28 +175,20 @@ def _get_virtual_bandwidth(env, start, end):
 
 
 @click.command(short_help="Bandwidth report for every pool/server")
-@click.option(
-    '--start',
-    callback=_validate_datetime,
-    default=(datetime.datetime.now() - datetime.timedelta(days=30)
-             ).strftime('%Y-%m-%d'),
-    help="datetime in the format 'YYYY-MM-DD' or 'YYYY-MM-DD HH:MM:SS'")
-@click.option(
-    '--end',
-    callback=_validate_datetime,
-    default=datetime.datetime.now().strftime('%Y-%m-%d'),
-    help="datetime in the format 'YYYY-MM-DD' or 'YYYY-MM-DD HH:MM:SS'")
-@click.option('--sortby', help='Column to sort by',
-              default='hostname',
-              show_default=True)
-@click.option('--virtual', is_flag=True,
-              help='Show the all bandwidth summary for each virtual server',
-              default=False)
-@click.option('--server', is_flag=True,
-              help='show the all bandwidth summary for each hardware server',
-              default=False)
+@click.option('--start', callback=_validate_datetime,
+              default=(datetime.datetime.now() - datetime.timedelta(days=30)).strftime('%Y-%m-%d'),
+              help="datetime in the format 'YYYY-MM-DD' or 'YYYY-MM-DD HH:MM:SS'")
+@click.option('--end', callback=_validate_datetime, default=datetime.datetime.now().strftime('%Y-%m-%d'),
+              help="datetime in the format 'YYYY-MM-DD' or 'YYYY-MM-DD HH:MM:SS'")
+@click.option('--sortby', help='Column to sort by', default='hostname', show_default=True)
+@click.option('--virtual', is_flag=True, default=False,
+              help='Show only the bandwidth summary for each virtual server')
+@click.option('--server', is_flag=True, default=False,
+              help='Show only the bandwidth summary for each hardware server')
+@click.option('--pool', is_flag=True, default=False,
+              help='Show only the bandwidth pool summary.')
 @environment.pass_env
-def cli(env, start, end, sortby, virtual, server):
+def cli(env, start, end, sortby, virtual, server, pool):
     """Bandwidth report for every pool/server.
 
     This reports on the total data transfered for each virtual sever, hardware
@@ -243,6 +238,9 @@ def _input_to_table(item):
             for item in itertools.chain(_get_pooled_bandwidth(env, start, end),
                                         _get_hardware_bandwidth(env, start, end)):
                 _input_to_table(item)
+        elif pool:
+            for item in _get_pooled_bandwidth(env, start, end):
+                _input_to_table(item)
         else:
             for item in itertools.chain(_get_pooled_bandwidth(env, start, end),
                                         _get_hardware_bandwidth(env, start, end),

SoftLayer/CLI/routes.py

@@ -22,6 +22,7 @@
     ('account:item-detail', 'SoftLayer.CLI.account.item_detail:cli'),
     ('account:cancel-item', 'SoftLayer.CLI.account.cancel_item:cli'),
     ('account:orders', 'SoftLayer.CLI.account.orders:cli'),
+    ('account:bandwidth-pools', 'SoftLayer.CLI.account.bandwidth_pools:cli'),
 
     ('virtual', 'SoftLayer.CLI.virt'),
     ('virtual:bandwidth', 'SoftLayer.CLI.virt.bandwidth:cli'),

SoftLayer/managers/account.py

@@ -326,3 +326,23 @@ def get_active_account_licenses(self):
         _mask = """billingItem,softwareDescription"""
 
         return self.client['SoftLayer_Account'].getActiveAccountLicenses(mask=_mask)
+
+    def get_bandwidth_pools(self, mask=None):
+        """Gets all the bandwidth pools on an account"""
+
+        if mask is None:
+            mask = """mask[totalBandwidthAllocated,locationGroup, id, name, billingCyclePublicUsageTotal,
+                      projectedPublicBandwidthUsage]
+                   """
+
+        return self.client.call('SoftLayer_Account', 'getBandwidthAllotments', mask=mask, iter=True)
+
+    def get_bandwidth_pool_counts(self, identifier):
+        """Gets a count of all servers in a bandwidth pool"""
+        mask = "mask[id, bareMetalInstanceCount, hardwareCount, virtualGuestCount]"
+        counts = self.client.call('SoftLayer_Network_Bandwidth_Version1_Allotment', 'getObject',
+                                  id=identifier, mask=mask)
+        total = counts.get('bareMetalInstanceCount', 0) + \
+                counts.get('hardwareCount', 0) + \
+                counts.get('virtualGuestCount', 0)
+        return total