From 7b53d212868ad9d95f48aeafcf9017d2fe187e41 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Sun, 31 Dec 2017 20:55:37 +0200 Subject: Add: download a blob from API --- worker_manager | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'worker_manager') 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: -- cgit v1.2.1