合并 pdf

  1. """
  2. 合并多个 pdf 到一个，使用方式：放到当前文件夹下运行
  3. """
  5. import os.path
  7. from pyPdf import PdfFileReader, PdfFileWriter # pip install pyPdf
  10. def get_pdf_files(dst_dir):
  11.     paths = []
  12.     for root, dirs, files in os.walk(dst_dir):
  13.         for filespath in files:
  14.             if filespath.endswith('.pdf'):  # pdf file
  15.                 abspath = os.path.join(root, filespath)
  16.                 paths.append(abspath)
  17.     return paths
  20. ##########################合并同一个文件夹下所有PDF文件########################
  21. def merge_pdf(dst_dir, outfile, sort=True):
  22.     output = PdfFileWriter()
  23.     curpage = 0
  24.     pdf_paths = sorted(get_pdf_files(dst_dir)) if sort else get_pdf_files(dst_dir)
  25.     for each in pdf_paths:
  26.         print(each)
  27.         reader = PdfFileReader(file(each, "rb"))
  28.         # 如果pdf文件已经加密，必须首先解密才能使用pyPdf
  29.         if reader.isEncrypted == True:
  30.             reader.decrypt("map")
  31.         # 获得源pdf文件中页面总数
  32.         page_count = reader.getNumPages()
  33.         curpage += page_count
  34.         print(page_count)
  36.         for iPage in range(0, page_count):
  37.             output.addPage(reader.getPage(iPage))
  39.     print("All Pages Number:" + str(curpage))
  40.     outputStream = file(dst_dir + outfile, "wb")
  41.     output.write(outputStream)
  42.     outputStream.close()
  45. def main():
  46.     merged = "all.pdf"
  47.     merge_pdf("./", merged)
  50. if __name__ == '__main__':
  51.     main()