GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-03 23:09:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace with open file handler, avoid reading files into memory

Browse Source
This commit is contained in:
debugtalk
2019-12-13 11:17:26 +08:00
parent 4c5636f44c
commit 7ba3c99f76
4 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
httprunner/ext/uploader/__init__.py
View File

@@ -128,9 +128,10 @@ def multipart_encoder(**kwargs):
if is_exists_file:
# value is file path to upload
filename = os.path.basename(_file_path)
with open(_file_path, 'rb') as f:
mime_type = get_filetype(_file_path)
fields_dict[key] = (filename, f.read(), mime_type)
mime_type = get_filetype(_file_path)
# TODO: fix ResourceWarning for unclosed file
file_handler = open(_file_path, 'rb')
fields_dict[key] = (filename, file_handler, mime_type)
else:
fields_dict[key] = value