Merge ~pwlars/testflinger-cli:artifact-downloading into testflinger-cli:master

[Paul Larson]

Downloading artifacts is now supported. If you want to test this, I'm also uploading a new snap to --beta right now.

[Paul Larson (pwlars)]

Minor change s/artifact/artifacts to match the dir name in the tarball and on the agent side

[Paul Larson (pwlars)]

review: Approve

diff --git a/README.rst b/README.rst
index b59ef3e..21b590b 100644
--- a/README.rst
+++ b/README.rst
@@ -56,9 +56,15 @@ job is complete.
 
   $ testflinger-cli poll <job_id>
 
-Finally, to get the full json results from the job when it is done running,
-you can use the 'results' subcommand:
+To get the full json results from the job when it is done running, you can
+use the 'results' subcommand:
 .. code-block:: console
 
   $ testflinger-cli results <job_id>
 
+Finally, to download the artifact tarball from the job, you can use the
+'artifact' subcommand:
+.. code-block:: console
+
+  $ testflinger-cli artifact [--filename <DEFAULT:artifact.tgz>] <job_id>
+
diff --git a/testflinger-cli b/testflinger-cli
index 52ca535..cb0e4b8 100755
--- a/testflinger-cli
+++ b/testflinger-cli
@@ -98,6 +98,25 @@ def results(ctx, job_id):
 
 @cli.command()
 @click.argument('job_id', nargs=1)
[e-mail]('--filename', default='artifacts.tgz')
[e-mail]_context
+def artifacts(ctx, job_id, filename):
+    conn = ctx.obj['conn']
+    print('Downloading artifacts tarball...')
+    try:
+        conn.get_artifact(job_id, filename)
+    except testflinger_cli.HTTPError as e:
+        if e.status == 204:
+            print('No artifacts tarball found for that job id.')
+        elif e.status == 400:
+            print('Invalid job id specified. Check the job id to be sure it '
+                  'is correct')
+        sys.exit(1)
+    print('Artifacts downloaded to {}'.format(filename))
+
+
[e-mail]()
[e-mail]('job_id', nargs=1)
 @click.pass_context
 def poll(ctx, job_id):
     conn = ctx.obj['conn']
diff --git a/testflinger_cli/__init__.py b/testflinger_cli/__init__.py
index 1d98424..56a1f90 100644
--- a/testflinger_cli/__init__.py
+++ b/testflinger_cli/__init__.py
@@ -92,6 +92,23 @@ class Client():
         endpoint = '/v1/result/{}'.format(job_id)
         return json.loads(self.get(endpoint))
 
+    def get_artifact(self, job_id, path):
+        """Get results for a specified test job
+
+        :param job_id:
+            ID for the test job
+        :param path:
+            Path and filename for the artifact file
+        """
+        endpoint = '/v1/result/{}/artifact'.format(job_id)
+        uri = urllib.parse.urljoin(self.server, endpoint)
+        req = requests.get(uri)
+        if req.status_code != 200:
+            raise HTTPError(req.status_code)
+        with open(path, 'wb') as artifact:
+            for chunk in req.iter_content(chunk_size=4096):
+                artifact.write(chunk)
+
     def get_output(self, job_id):
         """Get the latest output for a specified test job