Python实践练习：选择性拷贝

题目

项目要求：编写一个程序，遍历一个目录树，查找特定扩展名的文件（诸如.pdf 或.jpg），不论这些文件的位置在哪里， 将它们拷贝到一个新的文件夹中。

代码

import os
import shutil


def choseCopy(folder, targetFolder):
    folder = os.path.abspath(folder)
    targetFolder = os.path.abspath(targetFolder)
    if not os.path.exists(targetFolder):
        os.makedirs(targetFolder)

    for root, dirs, files in os.walk(folder):
        for file in files:
            if file.endswith('.jpg'):
                shutil.copy(os.path.join(root, file), targetFolder)


choseCopy('./chose', './find')