GitAutoupdater/GitAutoupdater.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import requests 
from urllib.request import urlretrieve
import os,sys
from argparse import ArgumentParser
import json


g_config = None
config_path = os.path.dirname(os.path.realpath(sys.argv[0]))+'\config.json'

def reporthook(count, block_size, total_size):
    """
    函数名可以自定义，参数 count 表示当前已经下载的数据块数目，
    block_size 表示每个数据块的大小，total_size 表示下载目标的总大小
    """
    percent = int(count * block_size * 100 / total_size)
    print('\r', f"正在下载：{count * block_size} / {total_size} bytes ({percent}%)", end='', flush=True)

def http_down(project, down_list=[], proxy=""):
    git_api = "https://api.github.com/repos/{}/releases/latest".format(project) 
    git_releases="{}/releases/download/".format(project) 

    print("获取", git_api)
    requests.packages.urllib3.disable_warnings()
    r = requests.get(git_api, verify=False)  #不检查证书
    # print(soup)
    latest_version = r.json()["tag_name"]
    print('Latest version:', latest_version)
    git_down_url=proxy+git_releases+latest_version+"/"
    print('git_down_url:', git_down_url)
    if len(down_list):
        if not os.path.exists(latest_version):
            os.makedirs(latest_version)
    for i in down_list:
        version_str = str(latest_version).replace("v", "").replace("V", "")
        i = str(i).replace("[version]", version_str)
        print('下载 {}...'.format(i))
        urlretrieve(git_down_url+i, latest_version+"/"+i, reporthook)
        print("\r")
    print("下载完成")    

if __name__ == "__main__":
    print(config_path)
    parser = ArgumentParser()
    parser.add_argument("n", help="项目，譬如 yisier/nps")
    parser.add_argument("l", help="下载列表，譬如 windows_amd64_client.tar.gz,windows_amd64_server.tar.gz, [version]可代替版本")
    parser.add_argument("-p", default=False, help="代理，譬如 https://ghproxy.com/https://github.com/")

    args = parser.parse_args()
    project = str(args.n)
    down_list = list_str = str(args.l).split(",")
    proxy = "https://github.com/"
    if args.p:
        proxy = str(args.p)

    g_config = {}
    if os.path.exists(config_path):
        with open(config_path) as file:
            g_config = json.load(file)
        if 'proxy' in g_config and not args.p: # args.p优先
            proxy = g_config['proxy']        

    http_down(project, down_list, proxy)