Administrator/NANOKVM-MaixCDK-MIRROR
1
0
Fork 1
mirror of https://github.com/sipeed/MaixCDK.git synced 2026-11-09 17:04:38 -06:00
Code Issues Packages Projects Releases 1 Wiki Activity

optimize downloader

Browse Source
This commit is contained in:
Neucrack
2024-05-10 15:09:21 +08:00
parent bd5cedfff6
commit bd5ece4444
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tools/file_downloader.py
View File

@@ -98,12 +98,15 @@ class Downloader:
self.callback = callback
self.data_content = None
headers = self.headers.copy()
res = requests.head(self.url, headers=headers)
# 302 redirect
if res.status_code == 302:
self.url = res.headers['Location']
print(f"redirect to {self.url}")
while 1:
res = requests.head(self.url, headers=headers)
# 302 redirect
if res.status_code == 302 or res.status_code == 301:
self.url = res.headers['Location']
print(f"redirect to {self.url}")
res = requests.head(self.url, headers=headers)
else:
break
if res.status_code != 200:
raise Exception("download file error, code: {}, content: {}".format(res.status_code, res.content.decode()))
try: