python 将pdf分页后插入至word中

所用技术

　　1. python编程基础

　　2. 使用pyPdf

　　3. 使用python操作word

　　4. 正则表达式的使用

　　5. windows的bat编程

 

下面是一个pyPdf库使用的示例：

    from pyPdf import PdfFileWriter, PdfFileReader



    output = PdfFileWriter()

    input1 = PdfFileReader(file("document1.pdf", "rb"))



    # add page 1 from input1 to output document, unchanged

    output.addPage(input1.getPage(0))



    # add page 2 from input1, but rotated clockwise 90 degrees

    output.addPage(input1.getPage(1).rotateClockwise(90))



    # add page 3 from input1, rotated the other way:

    output.addPage(input1.getPage(2).rotateCounterClockwise(90))

    # alt: output.addPage(input1.getPage(2).rotateClockwise(270))



    # add page 4 from input1, but first add a watermark from another pdf:

    page4 = input1.getPage(3)

    watermark = PdfFileReader(file("watermark.pdf", "rb"))

    page4.mergePage(watermark.getPage(0))



    # add page 5 from input1, but crop it to half size:

    page5 = input1.getPage(4)

    page5.mediaBox.upperRight = (

        page5.mediaBox.getUpperRight_x() / 2,

        page5.mediaBox.getUpperRight_y() / 2

    )

    output.addPage(page5)



    # print how many pages input1 has:

    print "document1.pdf has %s pages." % input1.getNumPages())



    # finally, write "output" to document-output.pdf

    outputStream = file("document-output.pdf", "wb")

    output.write(outputStream)

有了该库，就可以很容易将现有的pdf做分割。

因为我的需求是要将pdf中的关键字提取出来，用它来作为文件名。pyPdf中提供了将pdf中的文字全部提取出来。

inputfile.getPage(0).extractText()

这里返回的unicode，需要转为str

inputfile.getPage(0).extractText().encode("utf-8")

然后将每页的关键字提取出来，增加函数如下：

p_sheetName = re.compile('Blattname: (.+?)project')

def getSheetName(str):

    m = p_sheetName.search(str)

    if m:

        return m.group(1)

    else:

        return None;

最终代码如下：

from pyPdf import PdfFileWriter, PdfFileReader

import re,os



p_sheetName = re.compile('Blattname: (.+?)project')

def getSheetName(str):

    m = p_sheetName.search(str)

    if m:

        return m.group(1)

    else:

        return None;



def splitpdf(srcFile):

        input1 = file(srcFile,"rb")

        inputfile = PdfFileReader(input1)

        numofpages = inputfile.getNumPages()

        print "pages: %d" % numofpages

        #new directory

        folderName,ext_ = os.path.splitext(srcFile)

        if not os.path.isdir(folderName):

            os.makedirs(folderName)

        for page_index in range(1,numofpages+1):

            output = PdfFileWriter()

            output.addPage(inputfile.getPage(page_index-1))

            

            sheetName = getSheetName(inputfile.getPage(page_index-1).extractText().encode("utf-8"))

            #save file

            saveFileName = os.path.join(folderName,"%d %s.pdf" % (page_index,sheetName))

            print saveFileName

            outputFile = file(saveFileName,"wb")

            output.write(outputFile)

            outputFile.close()

        input1.close()





splitpdf("E:\\test.pdf")

下一步，将pdf参数化

from pyPdf import PdfFileWriter, PdfFileReader

import re,sys,os,string



def translator(frm='', to='', delete='', keep=None):

    if len(to) == 1 :

        to = to * len(frm)

    trans = string.maketrans(frm,to)

    if keep is not None:

        allchars = string.maketrans('','')

        delete = allchars.translate(allchars,keep.translate(allchars,delete))

    def translate(s):

        return s.translate(trans,delete)

    return translate



delete_some_speicl = translator(delete="/:\\?*><|")



p_sheetName = re.compile('Blattname: (.+?)project')

def getSheetName(str):

    m = p_sheetName.search(str)

    return delete_some_speicl(m.group(1))



def splitpdf(srcFile):

    try:

        folderName,ext_ = os.path.splitext(srcFile)

        if ext_ != '.pdf':

            raise Exception(os.path.basename(srcFile) + " is not pdf!")

        input1 = file(srcFile,"rb")

        inputfile = PdfFileReader(input1)

        numofpages = inputfile.getNumPages()

        print "pages: %d" % numofpages

        #new directory

        if not os.path.isdir(folderName):

            os.makedirs(folderName)

        for page_index in range(1,numofpages+1):

            output = PdfFileWriter()

            output.addPage(inputfile.getPage(page_index-1))

            

            sheetName = getSheetName(inputfile.getPage(page_index-1).extractText().encode("utf-8"))

            #save file

            saveFileName = os.path.join(folderName,"%d %s.pdf" % (page_index,sheetName))

            print saveFileName

            outputFile = file(saveFileName,"wb")

            output.write(outputFile)

            outputFile.close()

        input1.close()

        print "Split success!"

        print "please find them at " + folderName

    except Exception,e:

        print e



if __name__ == '__main__':

    if len(sys.argv) < 2:

        print 'usage: %s filename' % os.path.basename(sys.argv[0])

        exit(0)

    #print sys.argv[1]

    splitpdf(sys.argv[1])   

这里translator函数是将关键字中的特殊字符过滤掉，因为新建文件时可能会出错。

 

其实分开pdf也还需要一些手动操作，不然还需用vba导入到word中，我想直接用python干完这些事，如果就用到了win32com来操作word

下面是使用操作word的一个示例：

import win32com

from win32com.client import Dispatch, constants



w = win32com.client.Dispatch('Word.Application')

# 或者使用下面的方法，使用启动独立的进程：

# w = win32com.client.DispatchEx('Word.Application')



# 后台运行，不显示，不警告

w.Visible = 0

w.DisplayAlerts = 0



# 打开新的文件

doc = w.Documents.Open( FileName = filenamein )

# worddoc = w.Documents.Add() # 创建新的文档



# 插入文字

myRange = doc.Range(0,0)

myRange.InsertBefore('Hello from Python!')



# 使用样式

wordSel = myRange.Select()

wordSel.Style = constants.wdStyleHeading1



# 正文文字替换

w.Selection.Find.ClearFormatting()

w.Selection.Find.Replacement.ClearFormatting()

w.Selection.Find.Execute(OldStr, False, False, False, False, False, True, 1, True, NewStr, 2)



# 页眉文字替换

w.ActiveDocument.Sections[0].Headers[0].Range.Find.ClearFormatting()

w.ActiveDocument.Sections[0].Headers[0].Range.Find.Replacement.ClearFormatting()

w.ActiveDocument.Sections[0].Headers[0].Range.Find.Execute(OldStr, False, False, False, False, False, True, 1, False, NewStr, 2)



# 表格操作

doc.Tables[0].Rows[0].Cells[0].Range.Text ='123123'

worddoc.Tables[0].Rows.Add() # 增加一行



# 转换为html

wc = win32com.client.constants

w.ActiveDocument.WebOptions.RelyOnCSS = 1

w.ActiveDocument.WebOptions.OptimizeForBrowser = 1

w.ActiveDocument.WebOptions.BrowserLevel = 0 # constants.wdBrowserLevelV4

w.ActiveDocument.WebOptions.OrganizeInFolder = 0

w.ActiveDocument.WebOptions.UseLongFileNames = 1

w.ActiveDocument.WebOptions.RelyOnVML = 0

w.ActiveDocument.WebOptions.AllowPNG = 1

w.ActiveDocument.SaveAs( FileName = filenameout, FileFormat = wc.wdFormatHTML )



# 打印

doc.PrintOut()



# 关闭

# doc.Close()

w.Documents.Close(wc.wdDoNotSaveChanges)

w.Quit()

仿照上例，修改前面的代码如下：

from pyPdf import PdfFileWriter, PdfFileReader

import re,sys,os,string,win32com

from win32com.client import Dispatch, constants

win32com.client.gencache.EnsureDispatch('Word.Application')





def translator(frm='', to='', delete='', keep=None):

    if len(to) == 1 :

        to = to * len(frm)

    trans = string.maketrans(frm,to)

    if keep is not None:

        allchars = string.maketrans('','')

        delete = allchars.translate(allchars,keep.translate(allchars,delete))

    def translate(s):

        return s.translate(trans,delete)

    return translate



delete_some_speicl = translator(delete="/:\\?*><|")



p_sheetName = re.compile('Blattname: (.+?)project')

def getSheetName(str):

    m = p_sheetName.search(str)

    return m.group(1)



def splitPdfToWord(srcFile):

    try:

        folderName,ext_ = os.path.splitext(srcFile)

        if ext_ != '.pdf':

            raise Exception(os.path.basename(srcFile) + " is not pdf!")

        input1 = file(srcFile,"rb")

        inputfile = PdfFileReader(input1)

        numofpages = inputfile.getNumPages()

        print "Total Pages: %d" % numofpages

        wordApp = win32com.client.Dispatch('Word.Application')

        wordApp.Visible = False

        wordApp.DisplayAlerts = 0

        doc = wordApp.Documents.Add()

        sel = wordApp.Selection

        #new directory

        if not os.path.isdir(folderName):

            os.makedirs(folderName)

        for page_index in range(1,numofpages+1):

            output = PdfFileWriter()

            output.addPage(inputfile.getPage(page_index-1))

            

            sheetName = getSheetName(inputfile.getPage(page_index-1).extractText().encode("utf-8"))

            sel.Style = constants.wdStyleHeading1

            sel.TypeText("Page%d %s" % (page_index,sheetName))

            sheetName = delete_some_speicl(sheetName)

            #save file

            saveFileName = os.path.join(folderName,"%d %s.pdf" % (page_index,sheetName))

            print "Add Page %d" % page_index

            #print saveFileName

            outputFile = file(saveFileName,"wb")

            output.write(outputFile)

            outputFile.close()

            sel.TypeParagraph()

            sel.Style = constants.wdStyleBodyText

            sel.InlineShapes.AddOLEObject(ClassType="AcroExch.Document.11",FileName=saveFileName)

            sel.InsertBreak(Type=constants.wdPageBreak)

        input1.close()

        doc.SaveAs(folderName+".doc")

        print "Split success!"

        print "please find them at " + folderName

        print "create word document success!"

        print "Location:" + folderName + ".doc"

    except Exception,e:

        print e

    finally:

        wordApp.Quit()



if __name__ == '__main__':

    if len(sys.argv) < 2:

        print 'usage: %s filename' % os.path.basename(sys.argv[0])

        sys.exit(1)

    splitPdfToWord(sys.argv[1])