Send files the same way for file handles and paths so retries can res…

…et the seek values properly on retries
CybercentreCanada · Jul 5, 2024 · 10ac697 · 10ac697
1 parent 35708e1
commit 10ac697
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/assemblyline_client/v4_client/module/ingest.py b/assemblyline_client/v4_client/module/ingest.py
@@ -46,7 +46,7 @@ def __call__(self, fh=None, path=None, content=None, url=None, sha256=None, fnam
  else:
  raise ClientError('Could not guess the file name, please provide an fname parameter', 400)
  fh.seek(0)
- files = {'bin': (fname, fh)}
+ files = {'bin': fh}
  request = {
  'name': fname,
  }

diff --git a/assemblyline_client/v4_client/module/submit.py b/assemblyline_client/v4_client/module/submit.py
@@ -41,7 +41,7 @@ def __call__(self, fh=None, path=None, content=None, url=None, sha256=None, fnam
  else:
  raise ClientError('Could not guess the file name, please provide an fname parameter', 400)
  fh.seek(0)
- files = {'bin': (fname, fh)}
+ files = {'bin': fh}
  request = {
  'name': fname,
  }