Python模块os.listdir()查找目录中指定关键字的文件示例(Crossin教室实例23)

一、示例代码：

import os


def findfile(key, inputdir='.'):
    found_list = []
    all_files = os.listdir(inputdir)
    for name in all_files:                      # 遍历目录下所有文件
        full_name = inputdir + '/' + name
        if key in name:                             # 文件名中有关键字
            found_list.append(full_name)
        else:
            try:                                    # 加上try，避免打开非文本文件报错退出
                with open(full_name) as file:
                    for line in file:               # 对每一行进行遍历
                        if key in line:             # 若当前行中有关键字，就加入列表并跳出该文件
                            found_list.append(full_name + ':' + line.strip())
                            break
            except:
                pass
    return found_list


keyword = input('匹配关键字：')
path = input('搜索目录（不填写默认为当前目录）：')
if not path.strip():                                # 若没有有效字符
    path = '.'                                      # 表示当前目录
result = findfile(keyword, path)
for r in result:
    print(r)

二、运行测试结果：