python匹配包含文本的txt文件名

def find_file(content, file_dir):
    """
    匹配包含文本的txt文件
    :param content: 文本
    :param file_dir: txt所在文件目录
    :return: 文件名
    """
    file_list = []
    for file in os.listdir(file_dir):
        sep = os.path.splitext(file)
        if sep[1] == ".txt":
            file_list.append(file)
    for file in file_list:
        file_path = os.path.join(file_dir, file)
        with open(file_path, "r", encoding="utf-8") as f:
            contents = f.readline()     # 这里将txt文件当成单行文本
        if content in contents:
            return file
        
file_name = find_file(content, file_dir）