Merge pull request #254 from coursera/sb_multipart

Fix bug in multipart upload
coursera · Jul 15, 2016 · 529832f · 529832f
2 parents b6ec25d + 17caf18
commit 529832f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dataduct/s3/utils.py b/dataduct/s3/utils.py
@@ -47,11 +47,11 @@ def read_from_s3(s3_path):
     return key.get_contents_as_string()
 
 
-def _multipart_upload(bucket, key, file_path):
+def _multipart_upload(bucket, key_name, file_path):
     """Multipart upload for really large files
     """
     filename = os.path.basename(file_path)
-    mp = bucket.initiate_multipart_upload(filename)
+    mp = bucket.initiate_multipart_upload(key_name)
     source_size = os.stat(file_path).st_size
     chunks_count = int(math.ceil(source_size / float(CHUNK_SIZE)))
 
@@ -110,7 +110,7 @@ def upload_to_s3(s3_path, file_name=None, file_text=None, acl='private'):
     key = bucket.new_key(key_name)
     if file_name:
         if source_size > LARGE_FILE_LIMIT:
-            _multipart_upload(bucket, key, file_name)
+            _multipart_upload(bucket, key_name, file_name)
         else:
             if source_size > CHUNK_SIZE:
                 bar = pyprind.ProgPercent(