Add: download a blob from API

1 files changed, 12 insertions, 0 deletions

diff --git a/worker_manager b/worker_manager
index d29b597..47e285e 100755
--- a/worker_manager
+++ b/worker_manager
@@ -146,6 +146,12 @@ class ContainerAPI:
         headers = self.get_auth_headers()
         code = self._httpapi.put(url, headers, blob)
 
+    def download_blob(self, blob_id):
+        logging.info('Download blob %s', blob_id)
+        url = self.url('/blobs/{}'.format(blob_id))
+        headers = self.get_auth_headers()
+        return self._httpapi.get_blob(url, headers)
+
     def url(self, path):
         return '{}{}'.format(self._url, path)
 
@@ -179,6 +185,12 @@ class HttpApi:
             return None
         return r.json()
 
+    def get_blob(self, url, headers):
+        r = self._session.get(url, headers=headers, verify=False)
+        if not r.ok:
+            return None
+        return r.content
+
 
 class TokenGenerator: