python实现多个pdf文件合并

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

实现代码:

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

python 复制代码
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)
相关推荐
叽哥6 分钟前
Kotlin学习第 3 课:Kotlin 流程控制:掌握逻辑分支与循环的艺术
android·java·kotlin
杨杨杨大侠6 分钟前
第5章:实现Spring Boot集成
java·github·eventbus
华仔啊9 分钟前
工作5年没碰过分布式锁,是我太菜还是公司太稳?网友:太真实了!
java·后端
尚久龙13 分钟前
安卓学习 之 图片控件和图片按钮
android·java·学习·手机·android studio·安卓
摸鱼仙人~14 分钟前
深入理解 MyBatis-Plus 的 `BaseMapper`
java·开发语言·mybatis
杨杨杨大侠16 分钟前
第6章:高级特性与性能优化
java·github·eventbus
Dcs38 分钟前
代码评审还能更好!
java
刃神太酷啦1 小时前
C++ 异常处理机制:从基础到实践的全面解析----《Hello C++ Wrold!》(20)--(C/C++)
java·c语言·开发语言·c++·qt·算法·leetcode
蓝倾9761 小时前
小红书获取用户作品列表API接口操作指南
java·服务器·前端·python·电商开放平台·开放api接口
Seven971 小时前
剑指offer-28、数组中出现次数超过⼀半的数字
java