add s3 prefix

Substra · Jun 1, 2020 · f7b6e91 · f7b6e91
1 parent f6c6400
commit f7b6e91
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/substrapp/tasks/tasks.py b/backend/substrapp/tasks/tasks.py
@@ -52,6 +52,7 @@
 ACCESS_KEY = os.getenv('BUCKET_TRANSFER_ID')
 SECRET_KEY = os.getenv('BUCKET_TRANSFER_SECRET')
 BUCKET_NAME = os.getenv('BUCKET_TRANSFER_NAME')
+S3_PREFIX = os.getenv('BUCKET_TRANSFER_PREFIX')
 
 
 class TasksError(Exception):
@@ -886,7 +887,7 @@ def transfer_to_bucket(tuple_key, paths):
             's3',
             aws_access_key_id=ACCESS_KEY,
             aws_secret_access_key=SECRET_KEY)
-        s3.upload_file(tar_path, BUCKET_NAME, tar_name)
+        s3.upload_file(tar_path, BUCKET_NAME, f'{S3_PREFIX}{tar_name}' if S3_PREFIX else tar_name)
 
 
 @timeit