使用可从命令行运行的 Python 脚本将指定数量的行复制到多个文档

问题描述

我正在尝试构建一个脚本，将指定数量的行从一个文档复制到多个其他文档。复制的行应该附加到文档的末尾。如果我想删除文档末尾的行，脚本还必须能够删除指定数量的行。

我希望能够从命令行运行脚本并希望传递两个参数：

“添加”或“删除”
行数（从文档末尾算起）命令可能如下所示： py doccopy.py add 2 会将最后两行复制到其他文档，或者： py doccopy.py del 4 将从所有文档中删除最后 4 行。

到目前为止，我已经编写了一个函数，可以从原始文档中复制我想要的行数，

def copy_last_lines(number_of_lines):
    line_offset = [0]
    offset = 0
    for line in file_to_copy_from:
        line_offset.append(offset)
        offset += len(line)
    file_to_copy_from.seek(line_offset[number_of_lines])
    changedlines = file_to_copy_from.read()

将所述行粘贴到文档的函数

def add_to_file():
    doc = open(files_to_write[file_number],"a")
    doc.write("\n")
    doc.write(changedlines.strip())
    doc.close()

和一个主函数：

def main(action,number_of_lines):
    if action == "add":
        for files in files_to_write:
            add_to_file()
    elif action == "del":
        for files in files_to_write:
            del_from_file()
    else:
        print("Not a valid action.")

main 函数当然还没搞定，del_from_file 函数的实现我还没有搞清楚。我在遍历所有文档时也遇到问题。

我的想法是制作一个列表，其中包含我想要写入的文档的所有路径，然后遍历此列表并为“原始”文档创建一个变量，但我不知道这是否是可能是我想要的方式。如果可能，也许有人知道如何使用单个列表实现所有这些，将“原始”文档作为第一个条目，并在写入其他文档时循环以“1”开头的列表。

我意识到我到目前为止所做的代码完全是一团糟，我问了很多问题，所以我会很感激每一点帮助。我对编程完全陌生，过去 3 天我刚刚参加了 Python 速成课程，而我自己的第一个项目正变得比我想象的要复杂得多。

解决方法

我认为这应该可以满足您的要求。

# ./doccopy.py add src N dst...
#    Appends the last N lines of src to all of the dst files.
# ./doccopy.py del N dst...
#    Removes the last N lines from all of the dst files.

import sys

def process_add(args):
    # Fetch the last N lines of src.
    src = argv[0]
    count = int(args[1])
    lines = open(src).readlines()[-count:]

    # Copy to dst list.
    for dst in args[2:}
        open(dst,'a').write(''.join(lines))

def process_del(args):
    # Delete the last N lines of each dst file.
    count = int(args[0])
    for dst in args[1:]:
        lines = open(dst).readlines()[:-count]
        open(dst,'w').write(''.join(lines))

def main():
    if sys.argv[1] == 'add':
        process_add( sys.argv[2:] )
    elif sys.argv[1] == 'del':
        process delete( sys.argv[2:] )
    else:
        print( "What?" )

if __name__ == "__main__":
    main()

command-line python python-3.x