support additional headers in upload

jpmorganchase · Sep 12, 2024 · a44b9fc · a44b9fc
1 parent 1f7e5de
commit a44b9fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py_src/fusion/fusion_filesystem.py b/py_src/fusion/fusion_filesystem.py
@@ -736,8 +736,7 @@ async def _meth(url: Any, kw: Any) -> None:
                 headers[k] = v
 
         lpath.seek(0)
-        kw = self.kwargs.copy()
-        kw.update({"headers": headers})
+
         kw_op = self.kwargs.copy()
         if "File-Name" in headers:  # noqa: SIM102
             kw_op.setdefault("headers", {})
@@ -752,6 +751,8 @@ async def _meth(url: Any, kw: Any) -> None:
         headers["Digest"] = "SHA-256=" + base64.b64encode(hash_sha256.digest()).decode()
         kw = self.kwargs.copy()
         kw.update({"headers": headers})
+        if additional_headers:
+            kw["headers"].update(additional_headers)
         sync(self.loop, _finish_operation, operation_id, kw)
 
     def put(  # noqa: PLR0913