PDF分割

PDF分割

#!/usr/bin/python3
# -*- encoding: utf-8 -*-
\'\'\'
@File        :PDF分割.py
@Time        :2020/09/01 22:32:08
@Author      :hejiang
@Software    :vsCode
\'\'\'

import os
from PyPDF2 import PdfFileWriter, PdfFileReader


def split(pdf_file, delta, output_dir):
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    if not os.path.exists(pdf_file):
        return

    print(\'----------参数信息---------------\')
    print(\'File:\' + pdf_file)
    print(\'PageNums:\' + str(delta))
    print(\'Dest:\' + output_dir)
    print(\'-----------pdf开始切分-----------\')
    file_name = pdf_file.replace(\'\\\', \'/\').split(\'/\')[-1].split(\'.\')[0]

    output_dir = os.path.join(output_dir, file_name)
    if not os.path.exists(output_dir):
        os.mkdir(output_dir)
    input_stream = open(pdf_file, \'rb\')
    pdf_input = PdfFileReader(input_stream)
    page_count = pdf_input.getNumPages()
    sum_page_count = page_count // delta if page_count % delta == 0 else (
        page_count // delta) + 1

    remind_page = page_count % delta
    for i in range(0, sum_page_count):
        start = i * delta
        end = (i + 1) * delta
        pdf_out = PdfFileWriter()
        file_path = os.path.join(output_dir, os.path.split(pdf_file)[1])  #
        if i < sum_page_count:
            full_file_name = file_name + str(start +
                                             1) + \'-\' + str(end) + ".pdf"
            file_path = os.path.join(output_dir, full_file_name)
            print(file_path + \'切分完成\')
            for j in range(start, end):
                page = pdf_input.getPage(j)
                pdf_out.addPage(page)
        else:
            full_file_name = file_name + str(start + 1) + \'-\' + str(
                start + remind_page) + ".pdf"
            file_path = os.path.join(output_dir, full_file_name)
            print(file_path + \'切分完成\')
            for j in range(delta * (sum_page_count), page_count):
                page = pdf_input.getPage(j)
                pdf_out.addPage(page)
        out_stream = open(file_path, \'wb\')
        pdf_out.write(out_stream)
        out_stream.close()
    input_stream.close()
    print(\'-----------pdf切分完成-----------\')


if __name__ == \'__main__\':

    pdf_path = r\'C:\Users\He\Desktop\伟星PP-R价格表（灰）20180101.pdf\'
    page_count = 4
    out_dir = r\'C:\Users\He\Desktop\'

    split(pdf_path, page_count, out_dir)

    # import sys

    # try:
    #     pdf_path = sys.argv[1]
    #     print(pdf_path)
    #     page_count = int(sys.argv[2])
    #     out_dir = sys.argv[3]
    #     split(pdf_path, page_count, out_dir)
    # except:
    #     pass

最新代码,点击码云查看