Python脚本-remove_line

2025-01-01

这个脚本检查给定文件中的每一行，如果某行包含指定的字符串（如 sim_level ?=），则删除这一行。

Source Code


# -*- coding: utf-8 -*-
"""
Script to remove or replace lines containing a specific string in a file or multiple files within a directory.
Author: ZLin
Date: 2025-01-01
Description: This script checks if a line in a file contains the string "sim_level ?=" and either deletes or replaces that line with a new string if found. It can handle both single files and directories.
"""

import argparse
import os

class FileProcessor:
    def __init__(self, file_path, verbose=False):
        self.file_path = file_path
        self.verbose = verbose

    def process_file(self, target_string, replacement_string=None):
        try:
            with open(self.file_path, 'r') as file:
                lines = file.readlines()

            if replacement_string:
                # Replace lines containing the target string with the replacement string
                new_lines = [
                    replacement_string + '\n' if target_string in line else line
                    for line in lines
                ]
                if self.verbose:
                    print(f"Lines containing '{target_string}' have been replaced in {self.file_path}.")
            else:
                # Remove lines containing the target string
                new_lines = [line for line in lines if target_string not in line]
                if self.verbose:
                    print(f"Lines containing '{target_string}' have been removed in {self.file_path}.")

            # Rewrite the file with the modified lines
            with open(self.file_path, 'w') as file:
                file.writelines(new_lines)

        except Exception as e:
            print(f"Error while processing the file {self.file_path}: {e}")


def process_directory(directory, target_string, replacement_string=None, verbose=False):
    try:
        for root, _, files in os.walk(directory):
            for file_name in files:
                file_path = os.path.join(root, file_name)
                processor = FileProcessor(file_path, verbose)
                processor.process_file(target_string, replacement_string)
    except Exception as e:
        print(f"Error while processing the directory {directory}: {e}")


def main():
    parser = argparse.ArgumentParser(description="Remove or replace lines containing a specific string in a file or directory.")
    parser.add_argument('-f', '--file', type=str, help="The path to the file to process (if specified).")
    parser.add_argument('-d', '--directory', type=str, help="The path to the directory to process (if specified).")
    parser.add_argument('-s', '--string', type=str, required=True, help="The string to search for in the file(s).")
    parser.add_argument('-r', '--replace', type=str, help="The string to replace the matching line with (if specified).")
    parser.add_argument('-v', '--verbose', action='store_true', help="Enable verbose output (prints messages).")
    args = parser.parse_args()

    if args.directory:
        process_directory(args.directory, args.string, args.replace, verbose=args.verbose)
    elif args.file:
        processor = FileProcessor(args.file, verbose=args.verbose)
        processor.process_file(args.string, args.replace)
    else:
        print("Please provide either a file or a directory to process.")


if __name__ == "__main__":
    main()

脚本说明

功能：这个脚本检查给定文件中的每一行，如果某行包含指定的字符串（如 sim_level ?=），则删除这一行。
命令行参数
- -f 或 --file：要处理的文件路径。
- -s 或 --string：要查找的字符串。
- -r 或 --replace：如果提供了这个参数，脚本会用你指定的新字符串来替换包含目标字符串的行。如果没有提供替换字符串，则会删除包含目标字符串的行。
- -v 或 --verbose 参数，如果该参数被指定，脚本才会打印处理过程中的信息。
- -d 或 --directory 参数：如果指定了这个参数，脚本将遍历给定目录中的所有文件，执行删除或替换操作。
- process_file 方法：根据是否提供了替换字符串来决定是删除还是替换行。
默认行为：删除所有包含目标字符串的行。

使用方法

使用方法：

删除包含特定字符串的行
1
python remove_line.py -f test.txt -s "sim_level ?="
这个命令会删除 test.txt 中所有包含 sim_level ?= 的行。

替换包含特定字符串的行

1	python remove_line.py -f test.txt -s "sim_level ?=" -r "sim_level = 1"

处理单个文件

1	python remove_line.py -f test.txt -s "sim_level ?=" -r "sim_level = 1" -v

这个命令会处理 test.txt 文件，替换其中包含 sim_level ?= 的行，且输出详细信息。

1	python remove_line.py -d /path/to/directory -s "sim_level ?=" -r "sim_level = 1" -v

本文作者： ICXNM-ZLin
本文链接： https://talent-tudou.github.io/2025/01/01/Programming/Python脚本-remove_line/
版权声明： 本作品采用 CC BY-NC-SA 4.0 进行许可。转载请注明出处！