pdf合并代码

想合个pdf，发现市场上的软件都要钱，故写了个python代码解决，因本人没搞过python，故用ai写的，先下环境，然后安装pypdf库，把合并文件改一下名字放入同一目录，运行代码即可

from pypdf import PdfReader, PdfWriter

import sys

def insert_pdf(in1, in2, out, insert_after_page):

    reader1 = PdfReader(in1)

    reader2 = PdfReader(in2)

    writer = PdfWriter()

    # 先写 in1 的前 insert_after_page 页

    for i in range(insert_after_page):

        writer.add_page(reader1.pages[i])

    # 插入 in2 的所有页

    for page in reader2.pages:

        writer.add_page(page)

    # 写 in1 剩余的页

    for i in range(insert_after_page, len(reader1.pages)):

        writer.add_page(reader1.pages[i])

    with open(out, "wb") as f:

        writer.write(f)

    print(f"合并完成 -> {out}")

if __name__ == "__main__":

    if len(sys.argv) != 5:

        print("用法: python merge_insert.py in1.pdf in2.pdf out.pdf 插入页码")

    else:

        _, in1, in2, out, page = sys.argv

        insert_pdf(in1, in2, out, int(page))