python在给定的目录下查找文件中是否包含指定的字符串

写作防止遗忘。

这个小demo的作用是在给定的目录下，查找所有的文件中是否包含你需要查找的字符串。类似于Linux的grep -rn <string> 命令。demo中主要用到了os模块，第一个输入为你要查找的字符串，第二个给定一个路径，可以是绝对路径，也可以是相对路径，实现起来比较简单。

---

import os

text = input("input text : ")
path = input("path : ")
flag = 0
def getfiles(path):
    # 将flag设置全局变量，防止递归查找时信息提示错误。
    global flag
    # 到给定的路径下
    os.chdir(path)
    # 返回一个列表，列表中的元素为path下的文件和子文件夹
    files = os.listdir()
    # 依次遍历
    for file_name in files:
        # 返回文件或子文件夹的绝对路径
        abs_path = os.path.abspath(file_name)
        # 如果为文件则进行查找
        if os.path.isfile(abs_path):
            f = open(file_name, "r")
            # 在文件中找到要查到的字符串后进行打印，并将flag置为1
            if text in f.read():
                flag = 1
                print(text + " found in ")
                print(abs_path)
        # 如果为文件夹则重复前面的动作
        if os.path.isdir(abs_path):
            getfiles(abs_path)

    if flag == 0:
        print(text + " not found! ")
        return False
    return True

getfiles(path)