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))