如何一次替换多个项目？[python]

import os
import fileinput

def replace_in_files(directory, old_text, new_text):
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            if file.endswith(".txt"):  # 可以根据需要替换的文件类型进行修改
                with fileinput.FileInput(file_path, inplace=True) as f:
                    for line in f:
                        print(line.replace(old_text, new_text), end='')

# 替换目录下所有文件中的指定文本
replace_in_files('/path/to/directory', 'old_text', 'new_text')