Python實現下載界面(帶進度條，斷點續傳，多線程多任務下載等)

開發環境：

Windows 7 64位，Python 3.6.2

實現功能：

進度條，下載速度和下載進度的顯示，斷點續傳(暫停繼續下載功能)，取消下載等功能
下載界面，如圖所示

點擊'新建任務'，彈出輸入下載鏈接的窗口，如圖所示

點擊'開始下載'，可以自動獲取下載文件名和選擇存儲路徑，如圖所示：

關鍵代碼：

因爲斷點續傳是在之前的文件繼續追加，所以open(filename,'ab')
這裏打開文件的模式要爲ab
實現斷點續傳的代碼如下：

headers={'Range': 'bytes=%d-' %os.path.getsize(filename) }
r = requests.get(self.url,stream=True,headers=headers)
with open(filename, 'ab') as code:
    for chunk in r.iter_content(chunk_size=1024): #邊下載邊存硬盤
        if chunk :
            code.write(chunk)

顯示進度條的代碼(通過改變self.value的值顯示下載進度，範圍爲0-100)：

from tkinter import ttk
self.value=IntVar()
pb=ttk.Progressbar(self.fm4,length=200,variable=self.value)
pb.grid(row=0,column=1)

附上下載文件的關鍵代碼：

import os
import requests
import time
import re
import urllib
class Getfile():  #下載文件
    def __init__(self,url):
        self.url=url
        self.flag=True  #當self.flag=False，暫停或取消下載，也就是結束下載線程
        self.header_flag=False #當爲True時，設置header，斷點續傳
        self.re=requests.head(url,allow_redirects=True,timeout=20)  #運行head方法時重定向
    def getsize(self):
        try:
            self.file_total=int(self.re.headers['Content-Length']) #獲取下載文件大小    
            return self.file_total
        except:
            return 0
    def getfilename(self):  #獲取默認下載文件名
        if 'Content-Disposition' in self.re.headers:
            n=self.re.headers.get('Content-Disposition').split('name=')[1]
            filename=urllib.parse.unquote(n,encoding='utf8')
        else :
            filename=os.path.basename(self.re.url.split('?')[0])
        if filename=='':
            filename='index.html'
        return filename
    def downfile(self,filename):  #下載文件
        self.headers={}
        self.mode='wb'
        if os.path.exists(filename) and self.header_flag:
            self.headers={'Range': 'bytes=%d-' %os.path.getsize(filename) }
            self.mode='ab'
        self.r = requests.get(self.url,stream=True,headers=self.headers)
        with open(filename, self.mode) as code:
            for chunk in self.r.iter_content(chunk_size=1024): #邊下載邊存硬盤
                if chunk and self.flag:
                    code.write(chunk)
                else:
                    break
        time.sleep(1)
    def cancel(self,filename):  #取消下載
        self.flag=False
        time.sleep(1)
        if os.path.isfile(filename):
            os.remove(filename)

實現界面的代碼有點多，這裏就不給出來了。所有源碼已經上傳到 http://down.51cto.com/data/2445977 ，有需要的可以去下載。

思考與總結

在取消下載那裏花了不少時間，一直在糾結着怎麼強制退出下載文件的線程，最後發現不用強制退出，直接在下載文件的函數增加self.flag標誌位，當爲false時就可以使線程退出了。
這次的小項目又多學習到了多線程的一些知識。