Skip to content

Commit

Permalink
fix: total_length error when use local uploader
Browse files Browse the repository at this point in the history
- Fix total_length error in modules.transfer.multi_parts_uploader_from_url.
- Remove logs for request and response headers.
  • Loading branch information
hlf20010508 committed Nov 25, 2023
1 parent dbe9fcd commit 39f2170
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion modules/handlers/url.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ async def url_handler(event):
if total_length > 0:
logger('use local uploader to upload from url')
callback = Callback(event, status_message)
await multi_parts_uploader_from_url(name, local_response, total_length, callback)
await multi_parts_uploader_from_url(name, local_response, callback)
logger("File uploaded to %s"%os.path.join(onedrive.remote_root_path, name))
await status_message.finish()
else:
Expand Down
4 changes: 3 additions & 1 deletion modules/transfer.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,9 @@ async def multi_parts_uploader(
buffer.close()


async def multi_parts_uploader_from_url(name, response, total_length, progress_callback=None):
async def multi_parts_uploader_from_url(name, response, progress_callback=None):
total_length = int(response.headers['Content-Length'])

upload_session = onedrive.multipart_upload_session_builder(name)
uploader = onedrive.multipart_uploader(upload_session, total_length)

Expand Down
2 changes: 0 additions & 2 deletions modules/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,8 +163,6 @@ def get_filename(url):
headers['Referer'] = url
headers["Connection"] = "close"
response = requests.get(url, stream=True, verify=False, headers=headers)
logger(response.request.headers)
logger(response.headers)
if response.status_code == 200:
name = get_filename_from_cd(response.headers.get('Content-Disposition'))
content_type = response.headers['Content-Type']
Expand Down

0 comments on commit 39f2170

Please sign in to comment.