Python實驗:選擇性複製文件

#! python
# coding: utf-8
###編寫一個程序，遍歷一個目錄樹，查找特定擴展名的文件（諸如.pdf 或.jpg）。Python 編程快速上手——讓繁瑣工做自動化不論這些文件的位置在哪裏，將它們拷貝到一個新的文件夾中。
import shutil, os, re
filedir = 'C:\\Users\\Loyu\\Desktop\\python1'
desfiledir = 'C:\\Users\\Loyu\\Desktop\\file_copy'

###使用正則表達式########
totle = 0    #用來計算符合條件的文件數
for n, f, file in os.walk(filedir):
    if len(file) != 0:
        filestr = ' '.join(file)
        filerex = re.findall(r'\S+.pdf|\S+.txt', filestr)
        if len(filerex) != 0:
            for i in filerex:
                filetotledir = n +'\\' + i
                print(filetotledir)
                shutil.copy(filetotledir, desfiledir)
                totle += 1
    else:
        continue
print("總共找到 %d 個符合條件的文件" % totle)


###不使用正則表達式###
totle = 0    #用來計算符合條件的文件數
for n, f, file in os.walk(filedir):
    if len(file) != 0:
        for i in file:
            if '.pdf' in i or  '.txt' in i:
                filetotledir = n +'\\' + i
                print(filetotledir)
                shutil.copy(filetotledir, desfiledir)
                totle += 1
    else:
        continue
print("總共找到 %d 個符合條件的文件" % totle)