Skip to content

Commit 9737ecc

Browse files
committed
Use header-derived TUS offset variable
1 parent 67407bf commit 9737ecc

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

transloadit/async_client.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -325,11 +325,11 @@ async def upload_tus_assembly(self, file_count: int, content: bytes, fieldname:
325325
if uploadResponse.status != 204:
326326
raise RuntimeError(f"TUS upload returned HTTP {uploadResponse.status}, expected 204")
327327
try:
328-
remote_offset = int(uploadResponse.headers.get("Upload-Offset", ""))
328+
upload_offset = int(uploadResponse.headers.get("Upload-Offset", ""))
329329
except ValueError as error:
330330
raise RuntimeError("TUS upload returned an invalid Upload-Offset header") from error
331-
if remote_offset != len(content):
332-
raise RuntimeError(f"TUS upload offset {remote_offset}, expected {len(content)}")
331+
if upload_offset != len(content):
332+
raise RuntimeError(f"TUS upload offset {upload_offset}, expected {len(content)}")
333333

334334
completedAssembly = await self.wait_for_assembly(createdAssembly.data.get("assembly_ssl_url"))
335335

transloadit/client.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -333,11 +333,11 @@ def upload_tus_assembly(self, file_count: int, content: bytes, fieldname: str, f
333333
if uploadResponse.status_code != 204:
334334
raise RuntimeError(f"TUS upload returned HTTP {uploadResponse.status_code}, expected 204")
335335
try:
336-
remote_offset = int(uploadResponse.headers.get("Upload-Offset", ""))
336+
upload_offset = int(uploadResponse.headers.get("Upload-Offset", ""))
337337
except ValueError as error:
338338
raise RuntimeError("TUS upload returned an invalid Upload-Offset header") from error
339-
if remote_offset != len(content):
340-
raise RuntimeError(f"TUS upload offset {remote_offset}, expected {len(content)}")
339+
if upload_offset != len(content):
340+
raise RuntimeError(f"TUS upload offset {upload_offset}, expected {len(content)}")
341341

342342
completedAssembly = self.wait_for_assembly(createdAssembly.data.get("assembly_ssl_url"))
343343

0 commit comments

Comments
 (0)