Python脚本-extract_lines

2025-01-08

该脚本用于从输入文件中提取指定行范围的内容，并将提取的内容保存到一个新的输出文件中。

Source Code

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Script Description:
  This script extracts lines from an input file based on a specified range (inclusive of start and end lines),
  and saves the extracted content to an output file.
  Author: ZLin
  Date: 2025-01-08

Usage:
  python extract_lines.py -i <input_file_path> -s <start_line_number> -e <end_line_number> -o <output_file_path>
"""

import argparse

class LineExtractor:
  def __init__(self, input_file, start_row, end_row, output_file):
    self.input_file = input_file
    self.output_file = output_file
    self.start_row = start_row
    self.end_row = end_row
    

  def extract_lines(self):
    try:
      with open(self.input_file, 'r') as infile:
        lines = infile.readlines()

      # Ensure the line numbers are within a valid range
      if self.start_row < 1 or self.end_row > len(lines) or self.start_row > self.end_row:
        raise ValueError("Line numbers are out of range")

      # Extract the specified lines
      extracted_lines = lines[self.start_row - 1:self.end_row]

      # Write the extracted lines to the output file
      with open(self.output_file, 'w') as outfile:
        outfile.writelines(extracted_lines)

      print(f"Successfully extracted lines {self.start_row} to {self.end_row} and saved them to '{self.output_file}'.")

    except Exception as e:
      print(f"Error occurred: {e}")

def main():
  parser = argparse.ArgumentParser(description="Extract specified range of lines from an input file and save to an output file.")
  parser.add_argument('-i', '--input', type=str, required=True, help="Path to the input file")
  parser.add_argument('-o', '--output', type=str, required=True, help="Path to the output file")
  parser.add_argument('-s', '--start_row', type=int, required=True, help="Start line number")
  parser.add_argument('-e', '--end_row', type=int, required=True, help="End line number")
  

  args = parser.parse_args()

  extractor = LineExtractor(args.input, args.start_row, args.end_row, args.output)
  extractor.extract_lines()

if __name__ == "__main__":
  main()

脚本概述

该脚本用于从输入文件中提取指定行范围的内容，并将提取的内容保存到一个新的输出文件中。用户通过命令行传递输入文件路径、起始行号、结束行号和输出文件路径等参数。脚本会验证行号是否在合理的范围内，并将指定行写入到输出文件。

命令行参数

脚本使用 argparse 库解析命令行参数，以下是支持的参数说明：

-i / --input：输入文件路径（必填参数）。
-s / --start_row：起始行号（必填参数，1-based）。
-e / --end_row：结束行号（必填参数，1-based）。
-o / --output：输出文件路径（必填参数）。

使用示例

假设有一个文件 input.txt，并希望从第 10 行到第 20 行提取内容并保存到 output.txt：

1	python extract_lines.py -i input.txt -s 10 -e 20 -o output.txt

错误处理

脚本会进行如下错误检查：

行号范围检查：确保 start_row 小于等于 end_row，并且行号在文件总行数范围内。如果超出范围，脚本会抛出 ValueError。
文件读取异常：如果打开输入文件或输出文件时发生错误（例如文件不存在），会抛出异常并打印错误信息。

脚本输出

成功：当提取的行成功写入到输出文件时，脚本会输出：

1	Successfully extracted lines <start_row> to <end_row> and saved them to '<output_file>'.

失败：当出现错误时，脚本会输出类似以下的错误信息：
1
Error occurred: <错误信息>

总结

该脚本提供了一个简单的方法来提取文件中指定范围的行，并将它们保存到一个新的文件。用户只需通过命令行传递必要的参数，脚本会自动完成文件读取、行提取和文件写入的工作。

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