python 按关键词查找遍历所有文件和目录和子目录

python 按关键词查找遍历所有文件和目录， 当前目录以及当前目录的所有子目录下查找文件名包含指定字符串的文件

其中最下面的if语句为测试代码部分，当输入'123'时，会打印当前目录和当前目录的所有子目录下包含'123'的所有文件
当输入'.jpg'时，会打印当前目录和当前目录的所有子目录下包含'.jpg'格式的的所有文件
当输入'.'时，会打印当前目录和当前目录的所有子目录下 的所有文件

# coding=utf-8
# QQ2737499951
import os

mypath = "G:\\python学习"

"""
其中最下面的if语句为测试代码部分，当输入'123'时，会打印当前目录和当前目录的所有子目录下包含'123'的所有文件
当输入'.jpg'时，会打印当前目录和当前目录的所有子目录下包含'.jpg'格式的的所有文件
当输入'.'时，会打印当前目录和当前目录的所有子目录下 的所有文件
"""


def search(a, b):#a为路径，b为关键词
    for file in os.listdir(a):
        if os.path.isfile(a + '\\' + file):
            if b in file:
                print(file, '==>', a + '\\' + file)
        else:
            search(a + '\\' + file, b)


if __name__ == "__main__":
    search(os.path.abspath(mypath), '.jpg') #jpg格式查找 #'.'全部文件  '123'