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