61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import requests
|
||
import json
|
||
import os
|
||
from tqdm import tqdm
|
||
import threading
|
||
from time import sleep
|
||
|
||
|
||
print('在使用在线更新时，请安装steam++加速github才可以加速下载')
|
||
sleep(1)
|
||
|
||
url = 'https://api.github.com/repos/dengrb1/chatgpt/releases/latest'
|
||
|
||
response = requests.get(url, headers={'Accept': 'application/vnd.github.v3+json'})
|
||
release_info = json.loads(response.text)
|
||
|
||
new_version = release_info['tag_name']
|
||
current_version = '1.3' # 修改为你的已有版本号
|
||
|
||
if new_version != current_version:
|
||
print('发现新版本: {}'.format(new_version))
|
||
print('准备更新')
|
||
|
||
# 下载最新程序并保存到本地
|
||
download_url = release_info['assets'][0]['browser_download_url'] # 假设发布包第一个asset为我们要下载的程序
|
||
file_name = download_url.split('/')[-1]
|
||
file_path = os.path.join(os.getcwd(), file_name)
|
||
|
||
print('开始下载：{}'.format(file_name))
|
||
response = requests.get(download_url, stream=True)
|
||
content_size = int(response.headers['Content-Length'])
|
||
|
||
def write_data(start, end, url, file_path):
|
||
headers = {'Range': f'bytes={start}-{end}', 'Accept-Encoding': None}
|
||
res = requests.get(url, headers=headers, stream=True)
|
||
with open(file_path, 'rb+') as f:
|
||
f.seek(start)
|
||
var = f.tell()
|
||
f.write(res.content)
|
||
|
||
thread_num = 10
|
||
threads = []
|
||
step = content_size // thread_num
|
||
for i in range(thread_num):
|
||
start = step * i
|
||
if i == thread_num -1:
|
||
end = content_size - 1
|
||
else:
|
||
end = (i+1) * step - 1
|
||
t = threading.Thread(target=write_data, kwargs={"start": start, "end": end, "url": download_url, "file_path": file_path})
|
||
threads.append(t)
|
||
t.start()
|
||
for t in threads:
|
||
t.join()
|
||
|
||
print('下载完成！')
|
||
# 打开更新程序
|
||
os.startfile(file_path)
|
||
else:
|
||
print('当前已是最新版本')
|