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)