python 进度条

港控/mmm° 2022-08-31 11:24 227阅读 0赞

# -*- coding: utf-8 -*-
    
    import getpass
    import time
    
    import requests
    from tqdm import tqdm
    
    from requests.packages.urllib3 import disable_warnings
    disable_warnings()
    
    
    #下载 展示进度条方法
    def downFromUrl(url, filepath, session, headers):
        with session.get(url, headers=headers, stream=True, verify=False) as resq:
            fileSize = int(resq.headers['content-length'])
            headers["Range"] = f"bytes=0-{fileSize}"
            #pbar = tqdm(total=fileSize, initial=0, unit='B', unit_scale=True, desc=filepath, ascii=True)
            pbar = tqdm(total=fileSize, initial=0, unit='B', unit_scale=True,  ascii=True)
            with open(filepath, 'ab+') as f:
                for chunk in resq.iter_content(chunk_size=1024):  # (6)
                    if chunk:
                        f.write(chunk)
                        pbar.update(1024)
            pbar.close()
    
    #打印时间方法
    def printTime(takeTime):
        if takeTime <= 60.0:
            print(f"共耗时:{takeTime:.0f}秒")
        else:
            minute = int(takeTime // 60)
            second = takeTime - ((minute << 6) - (minute << 2))
            print(f"共耗时:{minute:.0f}分钟{second:.0f}秒")
        #input()
    
    
    def downUrl(url, filename, session, headers):
        #桌面路径
        deskDir = f'C:\\Users\\{getpass.getuser()}\\Desktop\\'
        print()
        print('开始下载 => ' + filename)
        startTime = time.time()
        downFromUrl(url, f'{deskDir}{filename}', session, headers)
        endTime = time.time()
        takeTime = endTime - startTime
        printTime(takeTime)
        print()
    
    
    if __name__ == '__main__':
        url = ''
        baseUrl = "https://upos-sz-mirrorkodoo1.bilivideo.com/upgcxcode/08/21/367132108/367132108_nb2-1-30080.m4s?e=ig8euxZM2rNcNbdlhoNvNC8BqJIzNbfqXBvEqxTEto8BTrNvN0GvT90W5JZMkX_YN0MvXg8gNEV4NC8xNEV4N03eN0B5tZlqNxTEto8BTrNvNeZVuJ10Kj_g2UB02J0mN0B5tZlqNCNEto8BTrNvNC7MTX502C8f2jmMQJ6mqF2fka1mqx6gqj0eN0B599M=&uipk=5&nbs=1&deadline=1627188415&gen=playurlv2&os=kodoo1bv&oi=3704185415&trid=e110fc7627614951a6faf69bf62f0d95u&platform=pc&upsig=a9215445eb953f5d2e2fe4adc9e519c6&uparams=e,uipk,nbs,deadline,gen,os,oi,trid,platform&mid=43864164&bvc=vod&nettype=0&orderid=0,3&agrr=1&logo=80000000"
        headers = { 
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:67.0) Gecko/20100101 Firefox/67.0',
            'referer': 'https://www.bilibili.com/'
        }
        downUrl(baseUrl, "test.mp4", requests.session(), headers)