Fix upload file Processing callback

terricain · May 27, 2024 · 4ad23e3 · 4ad23e3
1 parent e654bd6
commit 4ad23e3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aioboto3/s3/inject.py b/aioboto3/s3/inject.py
@@ -351,6 +351,8 @@ async def upload_fileobj(
     initial_data = Fileobj.read(Config.multipart_threshold)
     if inspect.isawaitable(initial_data):
         initial_data = await initial_data
+    if Processing:
+        initial_data = Processing(initial_data)
 
     if len(initial_data) < Config.multipart_threshold:
         # Do put_object
@@ -360,6 +362,8 @@ async def upload_fileobj(
             Body=initial_data,
             **kwargs
         )
+        if Callback:
+            Callback(len(initial_data))
         return
 
     # File bigger than threshold, start multipart upload