開發環境:
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時就可以使線程退出了。
這次的小項目又多學習到了多線程的一些知識。