递归---在文件夹中查找并展示以什么为后缀的文件

2017-07-13  本文已影响0人  段家大少xi

#首先导入模块

import os

import os.path

#定义两个参数,一个是文件夹名称,另外一个是后缀

def findFile(path,suffix):

    '''目的就是找到文件夹中以suffix结尾的文件'''

    ls = os.listdir(path)

    for temp in ls:

     #路径已经路径下的文件拼接

        tempPath = os.path.join(path,temp)

        #判断路径是否是文件

               if os.path.isfile(tempPath):

                  #后缀以什么结尾

                     if temp.endswith(suffix):

                              #添加到这个展示列表

                         fileList.append(tempPath)

               else:

                         findFile(tempPath,suffix)

#定义主函数

def main():

     #定义全局变量fileList

    global fileList

    fileList = []

    path = input('请您输入文件夹的路径:')

    suffix = input('请您输入想查询以什么后缀结尾的文件:')

    #判断路径是否是目录

    if os.path.isdir(path):

        findFile(path,suffix)

        print(fileList)

   else:

      print('文件夹不存在')

main()

上一篇下一篇

猜你喜欢

热点阅读