python调用多级目录中的文件_python多级目录文件查询

多层次目录下的文件查询，返回文件路径。

这个脚本是通过输入关键字查询文件的路径,脚本仅可查询“mp4、avi、mkv、rmvb、mov”格式的文件。

#!/usr/bin/python

# Author:Playon

# -*- coding:utf-8 -*-

# Time:2018/6/15 18:15

import os,re

rec="\\\\film-server\\project\\Refer\\Movie\\"

def findAllFile(base):

for root, ds, fs in os.walk(rec):

for f in fs:

rex = re.compile('mp4$|avi$|mkv$|rmvb$|mov$')

if re.findall(rex,f):

fullname = os.path.join(root, f)

yield fullname

def main(base):

for i in findAllFile(base):

print(i)

if __name__ == '__main__':

while True:

flag='xxx'

sear_name = input('请输入查询影片关键词 : ').lower()

for i in findAllFile(rec):

if sear_name in os.path.basename(i).lower():

print(i)

flag=666

if flag==666:

break

else:

print('毛都没有...')

请输入查询影片关键词 : 肖申克

\\film-server\project\Refer\Movie\肖申克的救赎.The.Shawshank.Redemption.1994.BD1080P.mp4