Skip to content

Commit 39f2170

Browse files
committed
fix: total_length error when use local uploader
- Fix total_length error in modules.transfer.multi_parts_uploader_from_url. - Remove logs for request and response headers.
1 parent dbe9fcd commit 39f2170

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

modules/handlers/url.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ async def url_handler(event):
9191
if total_length > 0:
9292
logger('use local uploader to upload from url')
9393
callback = Callback(event, status_message)
94-
await multi_parts_uploader_from_url(name, local_response, total_length, callback)
94+
await multi_parts_uploader_from_url(name, local_response, callback)
9595
logger("File uploaded to %s"%os.path.join(onedrive.remote_root_path, name))
9696
await status_message.finish()
9797
else:

modules/transfer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,9 @@ async def multi_parts_uploader(
7373
buffer.close()
7474

7575

76-
async def multi_parts_uploader_from_url(name, response, total_length, progress_callback=None):
76+
async def multi_parts_uploader_from_url(name, response, progress_callback=None):
77+
total_length = int(response.headers['Content-Length'])
78+
7779
upload_session = onedrive.multipart_upload_session_builder(name)
7880
uploader = onedrive.multipart_uploader(upload_session, total_length)
7981

modules/utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,8 +163,6 @@ def get_filename(url):
163163
headers['Referer'] = url
164164
headers["Connection"] = "close"
165165
response = requests.get(url, stream=True, verify=False, headers=headers)
166-
logger(response.request.headers)
167-
logger(response.headers)
168166
if response.status_code == 200:
169167
name = get_filename_from_cd(response.headers.get('Content-Disposition'))
170168
content_type = response.headers['Content-Type']

0 commit comments

Comments
 (0)