from PyPDF2 import PdfFileReader
 
def get_num_pages(file_path):
 """
 獲取文件總頁(yè)碼
 :param file_path: 文件路徑
 :return:
 """
 reader = PdfFileReader(file_path)
 # 不解密可能會(huì)報(bào)錯(cuò)：PyPDF2.utils.PdfReadError: File has not been decrypted
 if reader.isEncrypted:
  reader.decrypt('')
 page_num = reader.getNumPages()
 return page_num

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
 本腳本用來(lái)合并pdf文件，支持帶一級(jí)子目錄的
 每章內(nèi)容分別放在不同的目錄下，目錄名為章節(jié)名
 最終生成的pdf，按章節(jié)名生成書(shū)簽
'''

import os, sys, codecs
from PyPDF2 import PdfFileReader, PdfFileWriter, PdfFileMerger
import glob

def getFileName(filepath):
 '''
  獲取當(dāng)前目錄下的所有pdf文件
 '''
 file_list = glob.glob(filepath+"/*.pdf")
 # 默認(rèn)安裝字典序排序，也可以安裝自定義的方式排序
 # file_list.sort()
 return file_list


def get_dirs(filepath='', dirlist_out=[], dirpathlist_out=[]):
 # 遍歷filepath下的所有目錄
 for dir in os.listdir(filepath):
  dirpathlist_out.append(filepath + '\\' + dir)

 return dirpathlist_out


def merge_childdir_files(path):
 '''
  每個(gè)子目錄下合并生成一個(gè)pdf
 '''
 dirpathlist = get_dirs(path)
 if len(dirpathlist) == 0:
  print("當(dāng)前目錄不存在子目錄")
  sys.exit()
 for dir in dirpathlist:
  mergefiles(dir, dir)


def mergefiles(path, output_filename, import_bookmarks=False):
 # 遍歷目錄下的所有pdf將其合并輸出到一個(gè)pdf文件中，輸出的pdf文件默認(rèn)帶書(shū)簽，書(shū)簽名為之前的文件名
 # 默認(rèn)情況下原始文件的書(shū)簽不會(huì)導(dǎo)入，使用import_bookmarks=True可以將原文件所帶的書(shū)簽也導(dǎo)入到輸出的pdf文件中
 merger = PdfFileMerger()
 filelist = getFileName(path)
 if len(filelist) == 0:
  print("當(dāng)前目錄及子目錄下不存在pdf文件")
  sys.exit()
 for filename in filelist:
  f = codecs.open(filename, 'rb')
  file_rd = PdfFileReader(f)
  short_filename = os.path.basename(os.path.splitext(filename)[0])
  if file_rd.isEncrypted == True:
   print('不支持的加密文件：%s'%(filename))
   continue
  merger.append(file_rd, bookmark=short_filename, import_bookmarks=import_bookmarks)
  print('合并文件：%s'%(filename))
  f.close()
 # out_filename = os.path.join(os.path.abspath(path), output_filename)
 merger.write(output_filename + ".pdf")
 print('合并后的輸出文件：%s'%(output_filename))
 merger.close()

if __name__ == "__main__":
 # 每個(gè)章節(jié)一個(gè)子目錄，先分別合并每個(gè)子目錄文件為一個(gè)pdf，然后再將這些pdf合并為一個(gè)大的pdf，這樣做目的是想生成每個(gè)章節(jié)的書(shū)簽

 # 1.指定目錄
 # 原始pdf所在目錄
 path = "D:\spdf"
 # 輸出pdf路徑和文件名
 output_filename = "D:\spdf\戰(zhàn)略規(guī)劃 公司實(shí)現(xiàn)持續(xù)成功的方法、工具和實(shí)踐 羅熙昶 2018-09"

 # 2.生成子目錄的pdf
 # merge_childdir_files(path)

 # 3.子目錄pdf合并為總的pdf
 mergefiles(path, output_filename)