Python实现简单拆分PDF文件的方法

本文实例讲述了Python实现简单拆分PDF文件的方法。分享给大家供大家参考。具体如下：

依赖pyPdf处理PDF文件
切分pdf文件

使用方法：
1）将要切分的文件放在input_dir目录下
2）在configure.txt文件中设置要切分的份数（如要切分4份，则设置part_num=4）
3）执行程序
4）切分后的文件保存在output_dir目录下
5）运行日志写在pp_log.txt中

P.S. 本程序可以批量切割多个pdf文件

from pyPdf import PdfFileWriter, PdfFileReader
import os
import time
import sys
def part_pdf(input_file, output_file, config_count, f_w, now, file_name):
  file1 = file(input_file, \'rb\')
  pdf = PdfFileReader(file1)
  pdf_pages_len = len(pdf.pages)
  if config_count <= pdf_pages_len:
    ye = pdf_pages_len / config_count
    lst_ye = pdf_pages_len % config_count
    part_count = 0
    part_count_ye = 0
    for fen in range(config_count):
      part_count += 1
      if part_count == config_count:
        part_ye = ye + lst_ye
      else:
        part_ye = ye
      write_pdf(pdf, part_count_ye, part_count_ye+part_ye, fen, output_file)
      part_count_ye += ye
  else:
    f_w.writelines(\'time: \'+now+\' file name: \'+file_name+\' status: part_num > pdf pages [error]\\n\')
    sys.exit(1)
def write_pdf(pdf, part_count_ye, part_count_ye_end, fen, output_file):
  out = PdfFileWriter()
  for pp in range(part_count_ye, part_count_ye_end):
    out.addPage(pdf.getPage(pp))
  ous = file(output_file+\'_\'+str(fen+1)+\'.pdf\', \'wb\')
  out.write(ous)
  ous.close()
def pdf_main():
  f = open(\'configure.txt\', \'r\')
  f_w = open(\'pp_log.txt\', \'a\')
  now = time.strftime(\'%Y-%m-%d %H:%M:%S\')
  for i in f:
    i_ = i.strip()
    aa = i_.split(\'=\')[1]
    if i_.find(\'part_num=\') != -1 and aa.isdigit():
      config_count = int(aa)
    else:
      f_w.writelines(\'time: \'+now+\' status: part_num in configure.txt is error [error]\\n\')
      sys.exit(1)
  files = os.listdir(\'input_dir/\')
  for each in files:
    input_file = \'input_dir/\'+each
    file_name = input_file[input_file.index(\'/\'):input_file.index(\'.\')]
    output_file = \'output_dir/\'+file_name
    part_pdf(input_file, output_file, config_count, f_w, now, file_name)
    f_w.writelines(\'time: \'+now+\' file name: \'+file_name+\' status: success\\n\')
pdf_main()

希望本文所述对大家的Python程序设计有所帮助。