✨ feat(s3): allow for specifying bucketname withing the url

sladg · Dec 12, 2023 · 8b8d57e · 8b8d57e
1 parent d4cfb07
commit 8b8d57e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/imaginex_lambda/lib/img_lib.py b/imaginex_lambda/lib/img_lib.py
@@ -179,6 +179,9 @@ def download_and_optimize(url: str,
     with TemporaryFile() as buffer:
         if is_absolute(url):
             buffer, _ = download_image(buffer, url, chunk_size)
+        if url.startswith('s3://'):
+            bucket_name, key = url[len('s3://'):].split('/', 1)
+            buffer, _ = get_s3_image(buffer, bucket_name, key, chunk_size)
         else:
             key = url.strip('/')
             buffer, _ = get_s3_image(buffer, bucket_name, key, chunk_size)