python遍歷所有盤符下的圖片並拷貝下來

最近在學習python，閑着無聊就試着寫啦這個小的腳本，雖然有很多不足，但是還是收獲不少。

該腳本的功能：

①遍歷本地計算機中的所有盤符，並將名稱記錄下來；

②循環遍歷盤符下的所有圖片（當然這里可以根據自己的需求來遍歷不同類型的文件），並下載。

 

源代碼（有不足的地方請聯系我，謝謝）：

 1 #coding=utf-8
 2 import os
 3 import shutil
 4 
 5 #返回當前工作目錄
 6 beforedir = os.getcwd()
 7 
 8 def dir(way):
 9     picture = ["jpg","png"]
10     #遍歷目錄下的所有文件，文件夾
11     for root,dirs,files in os.walk(way):
12         for dirname in dirs:
13         #遞歸
14             dir(dirname)
15         for Filename in files:
16         #返回文件的完整路徑
17             path = os.path.join(root,Filename)
18             #返回文件的大小
19             size = os.path.getsize(path)
20             if ((Filename[-3:] in picture and Filename[0] != "$") and size < 2097152):
21                 shutil.copy(path,beforedir)
22                 
23 #遍歷所有盤符名稱
24 for i in range(65,91):
25     dirvename = chr(i) + ':'
26     if os.path.isdir(dirvename):
27         if dirvename != "C:":
28             print dirvename
29             dir(dirvename)