# !/usr/bin/python3
# -*-coding:UTF-8-*-
# FileName: os_searchBykey.py

import os

def search_key(dir, key):
    '''
        在当前目录以及当前目录的所有子目录下,
        查找文件名包含指定字符串的文件，并打印出相对路径。
    '''
    path_file = ''
    if dir == None or dir == '.':
        dir = os.getcwd()
    if key =='': return 0
    try:
        for root, dirs, files in os.walk(dir, topdown=True, onerror=OSError(FileNotFoundError)):
            # onerror=OSError(FileNotFoundError)：当所请求的文件或目录不存在时将被引发
            for name in files:
                if key in name:         # 查找文件名包含指定的关键字
                    path_file = os.path.join(root, name)
                    yield os.path.relpath(path_file)    # relpath(path, start=os.curdir),返回相对路径
    except Exception as e:
        print(f'输入目录参数dir是: “{dir}”, 输入文件类型参数key是: “{key}”')
        print(f'输入错误：“{e.__context__}”，请核对后再次运行程序。')

def main():
    _dir = str(input("请输入要查找文件的路径，如果直接回车则查找当前路径："))
    _ext = str(input("请输入要查找文件句的字符串(例如‘str’)，如果直接回车退出："))
    # _dir = r'D:\Python37\Code'
    # _ext = 'str'
    for s in search_key(_dir, _ext):
        print(s)

if __name__ == '__main__':
    main()