如何合并Python文本替换代码中的if语句以提升可维护性

我写了下面这段Python代码，想提升它的可维护性：

from tkinter import *
from tkinter import filedialog
import os
from docx import document


def replace_text(file_path, b1o, b1a, b1p, b1e):
  document = document(file_path)
   for paragraph in document.paragraphs:
    if 'B1O' in paragraph.text:
        paragraph.text = paragraph.text.replace('B1O', b1o)
    if 'B1A' in paragraph.text:
        paragraph.text = paragraph.text.replace('B1A', b1a)
    if 'B1P' in paragraph.text:
        paragraph.text = paragraph.text.replace('B1P', b1p)
    if 'B1E' in paragraph.text:
        paragraph.text = paragraph.text.replace('B1E', b1e)
return document

请问怎么合并这些if语句，让代码更好管理？我查过相关的合并if/elif语句的资料，但不确定怎么用到这个场景里。求建议！

编辑：以下是程序的其余代码片段：

def generate_file():
    # Open a file dialog to select a template file
    root = Tk()
    root.withdraw()
    file_path = filedialog.askopenfilename(
        title="Select a template file",
        filetypes=[("Word files", "*.docx")],
        initialdir=r"C://desktop/Templates"
    )

# Get user input for variables
    b1o = input("Enter the name of the owner: ")
    b1a = input("Enter the address: ")
    b1p = input("Enter the phone number: ")
    b1e = input("Enter the email address: ")

# Replace placeholders with user input
    document = replace_text(file_path, b1o, b1a, b1p, b1e)

# Open a file dialog to select a folder to save the new file
    folder_path = filedialog.askdirectory(title="Select a folder to save the new file")
    file_name = os.path.splitext(os.path.basename(file_path))[0] + "_" + b1p + "_" + b1o + ".docx"
    file_path = os.path.join(folder_path, file_name)

# Save the modified document as a new file
    document.save(file_path)
    print("File saved at:", file_path)

    root = Tk()
    root.title("Generate New Document")

# Create a button to generate a new file
    generate_button = Button(root, text="Select Template and Generate", command=generate_file)
    generate_button.pack()

root.mainloop()

1. 合并替换逻辑，移除冗余if判断 #

核心思路是用字典存储占位符与对应值的映射关系，然后遍历字典统一执行替换操作。这样既消除了重复的if判断，后续新增占位符时只需在字典里添加键值对，无需修改循环逻辑，扩展性大幅提升。

修改后的replace_text函数：

from docx import Document  # 修复原代码导入错误，避免类名与变量名冲突

def replace_text(file_path, replacements):
    doc = Document(file_path)  # 变量名改为doc，避免与Document类重名
    for paragraph in doc.paragraphs:
        # 遍历替换字典，统一处理所有占位符
        for placeholder, value in replacements.items():
            paragraph.text = paragraph.text.replace(placeholder, value)
    return doc

2. 优化调用逻辑，提升代码可读性 #

将原函数中零散的参数改为字典传递，让代码结构更清晰，同时增加用户取消操作的容错处理：

修改后的完整代码：

from tkinter import *
from tkinter import filedialog
import os
from docx import Document

def replace_text(file_path, replacements):
    doc = Document(file_path)
    for paragraph in doc.paragraphs:
        for placeholder, value in replacements.items():
            paragraph.text = paragraph.text.replace(placeholder, value)
    return doc

def generate_file():
    # 选择模板文件
    root = Tk()
    root.withdraw()
    file_path = filedialog.askopenfilename(
        title="选择模板文件",
        filetypes=[("Word文件", "*.docx")],
        initialdir=r"C://desktop/Templates"
    )
    if not file_path:  # 处理用户取消选择的情况
        return

    # 获取用户输入并构建替换字典
    replacements = {
        'B1O': input("输入所有者姓名："),
        'B1A': input("输入地址："),
        'B1P': input("输入电话号码："),
        'B1E': input("输入邮箱地址：")
    }

    # 执行文本替换
    doc = replace_text(file_path, replacements)

    # 选择保存文件夹
    folder_path = filedialog.askdirectory(title="选择保存文件夹")
    if not folder_path:  # 处理用户取消选择的情况
        return

    # 生成保存路径并保存文件
    original_name = os.path.splitext(os.path.basename(file_path))[0]
    file_name = f"{original_name}_{replacements['B1P']}_{replacements['B1O']}.docx"
    save_path = os.path.join(folder_path, file_name)
    doc.save(save_path)
    print(f"文件已保存至：{save_path}")

# 主窗口初始化
root = Tk()
root.title("生成新文档")
generate_button = Button(root, text="选择模板并生成", command=generate_file)
generate_button.pack()

root.mainloop()

额外优化说明 #

• 修复了原代码中from docx import document的导入错误，正确导入应为Document类，避免与变量名冲突
• 增加了用户取消文件/文件夹选择时的判断，防止后续代码因空路径报错
• 使用f-string格式化文件名，比传统字符串拼接更简洁易读
• 调整了代码缩进，修复原replace_text函数中的缩进错误
• 将零散参数改为字典传递，函数接口更简洁，后续新增占位符只需扩展字典即可

内容的提问来源于stack exchange，提问作者Devon