Add preliminary support for downloading objects

Added command and library to download a single object from swift

Change-Id: I3dc47b414ff37b526e6f633aa83ac3aa4b5be0ae
implements: bp swift-client

2 files changed, 62 insertions, 0 deletions

diff --git a/openstackclient/object/v1/lib/object.py b/openstackclient/object/v1/lib/object.py
index 5000c79d..38b3c14e 100644
--- a/openstackclient/object/v1/lib/object.py
+++ b/openstackclient/object/v1/lib/object.py
@@ -150,6 +150,32 @@ def list_objects(
     return session.get(requrl, params=params).json()
 
 
+def save_object(
+    session,
+    url,
+    container,
+    obj,
+    file=None
+):
+    """Save an object stored in a container
+
+    :param session: an authenticated keystoneclient.session.Session object
+    :param url: endpoint
+    :param container: name of container that stores object
+    :param object: name of object to save
+    :param file: local name of object
+    """
+
+    if not file:
+        file = obj
+
+    response = session.get("%s/%s/%s" % (url, container, obj), stream=True)
+    if response.status_code == 200:
+        with open(file, 'wb') as f:
+            for chunk in response.iter_content():
+                f.write(chunk)
+
+
 def show_object(
     session,
     url,
diff --git a/openstackclient/object/v1/object.py b/openstackclient/object/v1/object.py
index 812ad6e1..d7ccf6c6 100644
--- a/openstackclient/object/v1/object.py
+++ b/openstackclient/object/v1/object.py
@@ -183,6 +183,42 @@ class ListObject(lister.Lister):
                 ) for s in data))
 
 
+class SaveObject(command.Command):
+    """Save an object locally"""
+
+    log = logging.getLogger(__name__ + ".SaveObject")
+
+    def get_parser(self, prog_name):
+        parser = super(SaveObject, self).get_parser(prog_name)
+        parser.add_argument(
+            "--file",
+            metavar="<filename>",
+            help="Downloaded object filename [defaults to object name]",
+        )
+        parser.add_argument(
+            'container',
+            metavar='<container>',
+            help='Container name that has the object',
+        )
+        parser.add_argument(
+            "object",
+            metavar="<object>",
+            help="Name of the object to save",
+        )
+        return parser
+
+    def take_action(self, parsed_args):
+        self.log.debug("take_action(%s)", parsed_args)
+
+        lib_object.save_object(
+            self.app.client_manager.session,
+            self.app.client_manager.object_store.endpoint,
+            parsed_args.container,
+            parsed_args.object,
+            parsed_args.file,
+        )
+
+
 class ShowObject(show.ShowOne):
     """Show object information"""