python合并PDF文件

今天群里有个人提了这么一个需求：不同文件夹下的同名PDF能不能合并。

这个其实很简单我写了代码实现了。就是把同名文件先收集起来，然后形式一个同名文件的列表（全路径不一样，文件名一样），然后再把这个列表中的文件合并成一个PDF文件写到一个总的目录中，好了直接上代码吧。

#coding=utf-8
import os
import os.path
from pyPdf import PdfFileReader,PdfFileWriter

class MergePDF():
def __init__(self):
self.pdf_name = ""
        self.pdf_list = []

def Merge(self):
        out = PdfFileWriter()

for i in range(len(self.pdf_list)):
            src_pdf = self.pdf_list[i]
            pdf = PdfFileReader(file(src_pdf, 'rb'))

for page in pdf.pages:
                out.addPage(page)

        ous = file("C:/test_merge/Merge/"+self.pdf_name,'wb')
        out.write(ous)
        ous.close()

class Opt():
def __init__(self):
self.srcfolderlist = []
self.dstfolder = ""

        self.OptList = {}

def add_srcfolder(self,srcfolder):
self.srcfolderlist.append(srcfolder)

def add_dstfolder(self,dstfolder):
self.dstfolder = dstfolder

def deal(self):
        foldernum = len(self.srcfolderlist)
for i in range(foldernum):
            folder = self.srcfolderlist[i]
for parent,dirnames,filenames in os.walk(folder):
for filename in filenames:
self.createOpt(filename,parent + "/" + filename)

def createOpt(self,filename,fullname):
        isexist_key = self.OptList.has_key(filename)

if isexist_key == False :
            m = MergePDF()
            m.pdf_name = filename
            m.pdf_list.append(fullname)
self.OptList[filename] = m
else:
            m = self.OptList[filename]
            m.pdf_list.append(fullname)


def doMerge(self):
for key,obj in self.OptList.items():
            obj.Merge()

if __name__ == "__main__":
    optObj = Opt()
    optObj.add_srcfolder("C:/test_merge/1")
    optObj.add_srcfolder("C:/test_merge/2")
    optObj.add_srcfolder("C:/test_merge/3")
    optObj.add_srcfolder("C:/test_merge/4")
    optObj.add_srcfolder("C:/test_merge/5")
    optObj.add_srcfolder("C:/test_merge/6")

    optObj.deal()
    optObj.doMerge()

备注：C:\test_merge 这里里面是6个目录，每个目录下有PDF文件。

然后合并后的文件是在C:\test_merge\Merge目录中。

用到了pyPdf这个读写PDF的库。