我不知道如何让此脚本将受感染文件的列表打印到输出文件中

import os

# 获取受感染文件的列表
def get_infected_files(directory):
    infected_files = []
    for root, dirs, files in os.walk(directory):
        for file in files:
            # 判断文件是否受感染的逻辑
            if is_infected(file):
                infected_files.append(os.path.join(root, file))
    return infected_files

# 判断文件是否受感染的逻辑
def is_infected(file):
    # 在这里添加您的判断逻辑
    # 返回True表示文件受感染，返回False表示文件未受感染
    pass

# 将受感染文件的列表写入输出文件
def write_infected_files(infected_files, output_file):
    with open(output_file, 'w') as file:
        for infected_file in infected_files:
            file.write(infected_file + '\n')

# 主函数
def main():
    directory = '/path/to/directory'  # 替换为您要检查的目录路径
    output_file = '/path/to/output/file.txt'  # 替换为输出文件的路径

    infected_files = get_infected_files(directory)
    write_infected_files(infected_files, output_file)

if __name__ == '__main__':
    main()