有精通易语言的MJJ嘛

yjsx86

ficefox 发表于 2020-2-27 21:42
pm一下文件给我 大佬 我看看

没有做错误处理的！！！

1. import requests
   
2. import re
   
3. import urllib.parse
   
4. 
   
5. class FileInfo():
   
6.     def __init__(self, headers):
   
7.         self.headers = headers
   
8. 
   
9.     @property
   
10.     def name(self):
    
11.         rs = re.search(r'filename="(.*?)"', str(self.headers))
    
12.         if rs:
    
13.             return urllib.parse.unquote(rs.groups()[0])
    
14.         else:
    
15.             return None
    
16. 
    
17.     @property
    
18.     def size(self):
    
19.         rs =  self.headers.get('Content-Length', None)
    
20.         return int(rs) if rs else rs
    
21. 
    
22. class Down():
    
23.     def __init__(self, url):
    
24.         self.url = url
    
25.    
    
26.     def view_bar(self, num, total):
    
27.         rate = num/total
    
28.         rate_num = int(rate * 100)
    
29.         number = int(50*rate)
    
30.         r = '\r[%s%s]  %d%%' % ("#"*number, " "*(50-number), rate_num, )
    
31.         print("{}".format(r), end="\r")
    
32. 
    
33.     def Run(self):
    
34.         r = requests.get(self.url, stream=True)
    
35.         file = FileInfo(r.headers)
    
36.         print("fileName: ", file.name)
    
37.         print("fileSize: ", file.size)
    
38.         downloadedSize = 0
    
39.         with open(file.name, "wb") as f:
    
40.             for chunk in r.iter_content(chunk_size=1024):
    
41.                 if chunk:
    
42.                     f.write(chunk)
    
43.                     downloadedSize += 1024
    
44.                     self.view_bar(downloadedSize, file.size)
    
45.         r.close()
    
46.         
    
47. 
    
48. if __name__ == "__main__":
    
49.     url = input("Download Url >>> ")
    
50.     d = Down(url)
    
51.     d.Run()

复制代码