python学习------指定目录下搜索特定类型的文件

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zhaozhirongfree1111/article/details/80207509

import os
def searchFile(start_dir,target):
    os.chdir(start_dir);
    for each_file in  os.listdir(os.curdir):
        ext = os.path.splitext(each_file)[1]
        if ext in target:
            py_list.append(os.getcwd()+os.sep+each_file+os.linesep)
        if os.path.isdir(each_file):
            searchFile(each_file,target);
            os.chdir(os.pardir)


start_dir="F:\python\studyPython\day1";
program_dir=os.getcwd();
target = ['.py']
py_list=[];
searchFile(start_dir,target)

f= open(program_dir+os.sep+"py.txt","w")
f.writelines(py_list)
f.close()