#首先導(dǎo)入模塊

import os

import os.path

#定義兩個(gè)參數(shù)，一個(gè)是文件夾名稱，另外一個(gè)是后綴

def findFile(path,suffix):

? ? '''目的就是找到文件夾中以suffix結(jié)尾的文件'''

? ? ls = os.listdir(path)

? ? for temp in ls:

? ? ?#路徑已經(jīng)路徑下的文件拼接

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

? ? ? ? #判斷路徑是否是文件

? ? ? ? ? ? ? ?if os.path.isfile(tempPath):

? ? ? ? ? ? ? ? ? #后綴以什么結(jié)尾

? ? ? ? ? ? ? ? ? ? ?if temp.endswith(suffix):

? ? ? ? ? ? ? ? ? ? ? ? ? ? ? #添加到這個(gè)展示列表

? ? ? ? ? ? ? ? ? ? ? ? ?fileList.append(tempPath)

? ? ? ? ? ? ? ?else:

? ? ? ? ? ? ? ? ? ? ? ? ?findFile(tempPath,suffix)

#定義主函數(shù)

def main():

? ? ?#定義全局變量fileList

? ? global fileList

? ? fileList = []

? ? path = input('請(qǐng)您輸入文件夾的路徑:')

? ? suffix = input('請(qǐng)您輸入想查詢以什么后綴結(jié)尾的文件:')

? ? #判斷路徑是否是目錄

? ? if os.path.isdir(path):

? ? ? ? findFile(path,suffix)

? ? ? ? print(fileList)

? ?else:

? ? ? print('文件夾不存在')

main()