#首先導入模塊
import os
import os.path
#定義兩個參數(shù)，一個是文件夾名稱，另外一個是后綴
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):
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? #添加到這個展示列表
? ? ? ? ? ? ? ? ? ? ? ? ?fileList.append(tempPath)
? ? ? ? ? ? ? ?else:
? ? ? ? ? ? ? ? ? ? ? ? ?findFile(tempPath,suffix)
#定義主函數(shù)
def main():
? ? ?#定義全局變量fileList
? ? global fileList
? ? fileList = []
? ? path = input('請您輸入文件夾的路徑:')
? ? suffix = input('請您輸入想查詢以什么后綴結(jié)尾的文件:')
? ? #判斷路徑是否是目錄
? ? if os.path.isdir(path):
? ? ? ? findFile(path,suffix)
? ? ? ? print(fileList)
? ?else:
? ? ? print('文件夾不存在')
main()
