Python中通过下一段标记删除指定段落并插入新内容的问题

原代码存在的问题 #

1. 参数无效：函数定义的file_name参数未实际使用，硬编码了input.txt，导致调用时传参失效。
2. 行匹配不严谨：仅用line.strip('\n')匹配目标行，未处理行前后的空格，若目标行带有空格则无法匹配成功。
3. 逻辑不符合需求：当前逻辑是遇到start_key开启删除、stop_key关闭删除，跳过两者之间的内容，但未实现定位对应闭合括号的逻辑，也没有插入指定文本的步骤。

修正后的代码（匹配需求版本） #

以下代码实现：删除从start_key（如paragraph_a）到对应闭合括号')'的整个段落；若提前遇到stop_key（如下一段落名称paragraph_b）则终止删除，同时在删除位置插入指定文本。

def erase_and_insert(file_name: str, start_key: str, stop_key: str, insert_text: str):
    try:
        # 读取文件所有行
        with open(file_name, 'r') as fr:
            lines = fr.readlines()
        
        processed_lines = []
        delete_mode = False
        found_start = False
        
        for line in lines:
            stripped_line = line.strip()
            
            if not delete_mode:
                # 未进入删除模式，正常写入行，直到找到起始标记
                if stripped_line == start_key:
                    delete_mode = True
                    found_start = True
                else:
                    processed_lines.append(line)
            else:
                # 删除模式下，检查终止条件
                if ')' in stripped_line:
                    # 找到对应闭合括号，退出删除模式并插入文本
                    delete_mode = False
                    processed_lines.append(insert_text + '\n')
                elif stripped_line == stop_key:
                    # 提前遇到下一段落名称，退出删除模式，插入文本后保留当前行
                    delete_mode = False
                    processed_lines.append(insert_text + '\n')
                    processed_lines.append(line)
                # 删除模式下的其他行直接跳过
        
        # 兜底处理：找到起始标记但未找到终止条件的情况
        if found_start and delete_mode:
            print("警告：未找到对应闭合括号或下一段落标记，已插入指定文本")
            processed_lines.append(insert_text + '\n')
        
        # 写入处理后的内容到输出文件
        with open('output.txt', 'w') as fw:
            fw.writelines(processed_lines)
            
    except Exception as ex:
        print(f"处理错误:\n\t{ex}")

def main():
    # 调用示例：文件名、起始段落名、下一段落名、插入文本
    erase_and_insert('input.txt', 'paragraph_a', 'paragraph_b', 'story "A is deleted"')

if __name__ == "__main__":
    main()

代码逻辑说明 #

• 状态管理：用delete_mode标记是否处于删除段落的状态，found_start记录是否找到起始段落标记。
• 终止条件：支持两种终止删除的情况：找到对应闭合括号，或提前遇到下一段落名称stop_key。
• 文本插入：在删除段落的位置插入指定文本，确保输出结构符合需求。
• 容错处理：对未找到终止条件的情况添加警告并兜底插入文本。

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