python实现多个pdf文件合并

打印发票时，需要将pdf合并成一个，单页两张打印。网上一些pdf合并逐渐收费，这玩意儿都能收费？自己写一个脚本使用。

实现代码：

输入pdf文件夹路径data_dir，统计目录下的"合并后的PDF"文件夹下，输出合并后的pdf文件。

import os
import PyPDF2
import argparse

def merge_pdfs(paths, output):
    pdf_writer = PyPDF2.PdfWriter()

    for path in paths:
        pdf_reader = PyPDF2.PdfReader(path)
        for page in range(len(pdf_reader.pages)):
            pdf_writer.add_page(pdf_reader.pages[page])

    with open(output, 'wb') as out:
        pdf_writer.write(out)

def main(data_dir):
    names = os.listdir(data_dir)
    print("names:", names)

    file_paths = []
    for name in names:
        file_paths.append(os.path.join(data_dir, name))

    output = os.path.join(data_dir, "合并后的PDF/merge_%02d.pdf" % len(names))  # 合并后的PDF文件路径
    os.makedirs("./合并后的PDF", exist_ok=True)

    merge_pdfs(file_paths, output)
    print("合并后文件：", output)

def argv_parse():
    parser = argparse.ArgumentParser(usage="it's usage PyPDF2.", description="pdf merge.")
    parser.add_argument("--data_dir", type=str, help="pdf file dir.")
    return parser.parse_args()

if __name__ == "__main__":
    argv = argv_parse()
    print("argv:", argv)
    main(argv.data_dir)