list and detail support for capacity groups

Author: allmightyspiff

SoftLayer/CLI/virt/capacity/detail.py

@@ -0,0 +1,33 @@
+"""Shows the details of a reserved capacity group"""
+# :license: MIT, see LICENSE for more details.
+
+import click
+
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import formatting
+from SoftLayer.managers.vs_capacity import CapacityManager as CapacityManager
+
+
+from pprint import pprint as pp
+
+@click.command()
+@click.argument('identifier')
+@environment.pass_env
+def cli(env, identifier):
+    """Reserved Capacity Group Details"""
+    manager = CapacityManager(env.client)
+    result = manager.get_object(identifier)
+    try:
+        flavor = result['instances'][0]['billingItem']['description']
+    except KeyError:
+        flavor = "Pending Approval..."
+
+    table = formatting.Table(
+        ["ID", "Created"], 
+        title= "%s - %s" % (result.get('name'), flavor)
+    )
+    for rci in result['instances']:
+        table.add_row([rci['guestId'], rci['createDate']])
+    env.fout(table)
+

SoftLayer/CLI/virt/capacity/list.py

@@ -17,4 +17,23 @@ def cli(env):
     """List Reserved Capacity"""
     manager = CapacityManager(env.client)
     result = manager.list()
-    pp(result)
+    table = formatting.Table(
+        ["ID", "Name", "Capacity", "Flavor", "Instance Cost", "Created"], 
+        title="Reserved Capacity"
+    )
+    for rc in result:
+        occupied_string = "#" * int(rc.get('occupiedInstancesCount',0))
+        available_string = "-" * int(rc.get('availableInstanceCount',0))
+
+        try:
+            flavor = rc['instances'][0]['billingItem']['description']
+            cost = float(rc['instances'][0]['billingItem']['hourlyRecurringFee'])
+            instance_count = int(rc.get('instanceCount',0)) 
+            cost_string = "%s * %s = %s" % (cost, instance_count, cost * instance_count)
+        except KeyError:
+            flavor = "Unknown Billing Item"
+            cost_string = "-"
+        capacity = "%s%s" % (occupied_string, available_string)
+        table.add_row([rc['id'], rc['name'], capacity, flavor, cost_string, rc['createDate']])
+    env.fout(table)
+    # pp(result)

SoftLayer/managers/vs_capacity.py

@@ -33,14 +33,21 @@ def __init__(self, client, ordering_manager=None):
         self.client = client
         self.account = client['Account']
         self.capacity_package = 'RESERVED_CAPACITY'
+        self.rcg_service = 'Virtual_ReservedCapacityGroup'
 
         if ordering_manager is None:
             self.ordering_manager = ordering.OrderingManager(client)
 
     def list(self):
-        results = self.client.call('Account', 'getReservedCapacityGroups')
+        mask = "mask[availableInstanceCount, occupiedInstanceCount, instances[billingItem], instanceCount]"
+        results = self.client.call('Account', 'getReservedCapacityGroups', mask=mask)
         return results
 
+    def get_object(self, identifier):
+        mask = "mask[instances[billingItem]]"
+        result = self.client.call(self.rcg_service, 'getObject', id=identifier, mask=mask)
+        return result
+
     def get_create_options(self):
         mask = "mask[attributes,prices[pricingLocationGroup]]"
         # mask = "mask[id, description, capacity, units]"