Add endpoint_export_inventory CLI command

Signed-off-by: Mihaela Balutoiu <mbalutoiu@cloudbasesolutions.com>

Author: mihaelabalutoiu

coriolisclient/cli/endpoints.py

@@ -17,12 +17,14 @@
 """
 import argparse
 import json
+import sys
 
 from cliff import command
 from cliff import lister
 from cliff import show
 
 from coriolisclient.cli import formatter
+from coriolisclient.cli import utils as cli_utils
 from coriolisclient import exceptions
 
 
@@ -249,6 +251,36 @@ def take_action(self, args):
         return EndpointFormatter().list_objects(obj_list)
 
 
+class ExportEndpointInventory(command.Command):
+    """Export the VM inventory of an endpoint in CSV format"""
+
+    def get_parser(self, prog_name):
+        parser = super(ExportEndpointInventory, self).get_parser(prog_name)
+        parser.add_argument('id', help='The endpoint\'s id or name')
+        cli_utils.add_args_for_json_option_to_parser(parser, 'environment')
+        parser.add_argument(
+            '--output-file',
+            help='Path to write the CSV to. Defaults to stdout.')
+        return parser
+
+    def take_action(self, args):
+        endpoints = self.app.client_manager.coriolis.endpoints
+        endpoint_id = endpoints.get_endpoint_id_for_name(args.id)
+        source_environment = cli_utils.get_option_value_from_args(
+            args, 'environment', error_on_no_value=False)
+
+        csv_content = endpoints.get_inventory_csv(
+            endpoint_id, source_environment=source_environment)
+
+        if args.output_file:
+            with open(args.output_file, 'w') as fout:
+                fout.write(csv_content)
+            self.app.stdout.write(
+                'Inventory written to %s\n' % args.output_file)
+        else:
+            sys.stdout.write(csv_content)
+
+
 class EndpointValidateConnection(command.Command):
     """validates an edpoint's connection"""
 

coriolisclient/cli/providers.py

@@ -41,7 +41,8 @@
     256: "Listing VM Flavors",
     512: "Listing Available Options",
     524288: "Source Minion Pool Operations",
-    1048576: "Destination Minion Pool Operations"
+    1048576: "Destination Minion Pool Operations",
+    2097152: "Endpoint Inventory Export",
 }
 
 PROVIDER_SCHEMA_TYPE_MAP = {

coriolisclient/v1/endpoints.py

@@ -16,6 +16,7 @@
 from coriolisclient import base
 from coriolisclient.cli import utils
 from coriolisclient import exceptions
+from coriolisclient.v1 import common
 
 
 class ConnectionInfo(base.Resource):
@@ -64,6 +65,17 @@ def update(self, endpoint, updated_values):
     def delete(self, endpoint):
         return self._delete('/endpoints/%s' % base.getid(endpoint))
 
+    def get_inventory_csv(self, endpoint, source_environment=None):
+        url = '/endpoints/%s/inventory' % base.getid(endpoint)
+        if source_environment:
+            if not isinstance(source_environment, dict):
+                raise ValueError("'source_environment' param must be a dict")
+            encoded_env = common.encode_base64_param(
+                source_environment, is_json=True)
+            url = '%s?env=%s' % (url, encoded_env)
+        resp = self.client.get(url, headers={'Accept': 'text/csv'})
+        return resp.text
+
     def validate_connection(self, endpoint):
         data = self.client.post(
             '/endpoints/%s/actions' % base.getid(endpoint),

setup.cfg

@@ -33,6 +33,7 @@ coriolis.v1 =
     endpoint_list = coriolisclient.cli.endpoints:ListEndpoint
     endpoint_show = coriolisclient.cli.endpoints:ShowEndpoint
     endpoint_validate_connection = coriolisclient.cli.endpoints:EndpointValidateConnection
+    endpoint_export_inventory = coriolisclient.cli.endpoints:ExportEndpointInventory
 
     endpoint_instance_list = coriolisclient.cli.endpoint_instances:ListEndpointInstance
     endpoint_instance_show = coriolisclient.cli.endpoint_instances:ShowEndpointInstance