We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 609661f commit 1bb3e56Copy full SHA for 1bb3e56
elorus/client.py
@@ -40,9 +40,13 @@ def _get_auth(self):
40
)
41
42
def handle_file_download(self, response):
43
- filename = response.headers.get("Content-Disposition", None)
44
- if filename:
45
- filename = filename.split("filename=")[1]
+ content_disposition = response.headers.get("Content-Disposition", "")
+ if "filename=" not in content_disposition:
+ raise ValueError("No filename found in Content-Disposition header")
46
+ filename_kv = content_disposition.split("filename=")
47
+ if len(filename_kv) < 2:
48
+ raise ValueError("Invalid Content-Disposition header format")
49
+ filename = filename_kv[1]
50
return filename, response.content
51
52
def _handle_response(self, response):
0 commit comments