robotframework 测试工具添加PDF文件内容匹配插件

# -*- coding: UTF-8 -*-
# coding=utf-8
#from __future__ import unicode_literals
import sys
import os

import subprocess
import time
import re
from robot.libraries.BuiltIn import BuiltIn
from mailcommon import mailcommon
from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdfparser import PDFParser
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.pdfdevice import PDFDevice, TagExtractor
from pdfminer.pdfpage import PDFPage
from pdfminer.converter import XMLConverter, HTMLConverter, TextConverter
from pdfminer.cmapdb import CMapDB
from pdfminer.layout import LAParams
from pdfminer.image import ImageWriter
import cStringIO
#reload(sys)
#sys.setdefaultencoding(‘utf-8‘)

class pdfContentKeywords(object):
    ROBOT_LIBRARY_SCOPE = ‘Global‘

    root = os.path.abspath(os.path.join(__file__, ‘..‘))


    def __init__(self):
        self.builtin = BuiltIn()


    

    def handlepdf(self, path, param_dict,email, fail=True):
        ‘‘‘ Reads a specified directory pdf file and diff ...

        `path`: pdf file path

        `param_dict`: Require comparison data

        `fail`:  If there are differences it will throw an exception and test will fail
                 defaults to True, if False test‘s will continue ‘‘‘
        param_dict=eval(param_dict)
        self.builtin.log("pdf: %s" %path)
        self.builtin.log("param: %s" %param_dict)

        fail = self.builtin.convert_to_boolean(fail)
        result=dict
        if fail:
            result=self._diffContent(path, param_dict)
        else:
            try:
                result=self._diffContent(path, param_dict)
            except Exception, e:
                self.builtin.log(e)
        msg=‘‘
        for i in result:
            print u"result:%s=" % i,result[i]
            #print self.builtin.log(i) 
            msg+="=>".join([i,str("检验通过" if result[i]==1 else "检验没通过")])
        mail_obj=mailcommon()
        print u"mail MSG:%s" % msg
        maillist=[]
        maillist.append(email)
        mail_obj.send_mail(maillist,‘测试结果分析‘,msg);
        self.builtin.log("End")

    def _readPdf(self,path):
        caching = False
        password = ‘‘
        pagenos = set()
        maxpages = 0
        # output option
        outfile = None
        outtype = None
        imagewriter = None
        rotation = 0
        layoutmode = ‘normal‘
        codec = ‘utf-8‘
        pageno = 1
        scale = 1
        showpageno = True
        laparams = LAParams()
        content=‘‘
        outfp=cStringIO.StringIO()
        try:
            rsrcmgr = PDFResourceManager(caching=caching)
            
            device = TextConverter(rsrcmgr, outfp, codec=codec, laparams=laparams,
                                       imagewriter=imagewriter)
            fp = file(path, ‘rb‘)
            interpreter = PDFPageInterpreter(rsrcmgr, device)
            for page in PDFPage.get_pages(fp, pagenos,
                                            maxpages=maxpages, password=password,
                                            caching=caching, check_extractable=True):
                page.rotate = (page.rotate+rotation) % 360
                interpreter.process_page(page)
            fp.close()  
            device.close()
            content= outfp.getvalue()
            outfp.close()
        except Exception, e:
            print "Exception:%s",e
            self.builtin.log(e)
        content=content.replace(‘\n‘,‘‘).replace(‘\t‘,‘‘).strip(‘\n‘)
        print u"pdf file content:%s"% content#.encode(‘utf-8‘).strip()
        return content

        
    def _diffContent(self,path,p_dict):
        result=dict()
        try:
            pdfContent=self._readPdf(path)
            fail=0
            for i in p_dict: 
              print "dict[%s]=" % i,p_dict[i] 
              if re.search(r‘‘+i+p_dict[i] ,pdfContent,re.I | re.M|re.X):
                fail=1   
              result[i+p_dict[i]]=fail
        except Exception, e:
            print "Exception:%s",e
            self.builtin.log(e)
        return result

    

if __name__ == "__main__":
    d = pdfContentKeywords()
    print d.handlepdf("C:\\Print.pdf",{"船名/航次:":"CMACGMFIDELIO/B14WESVLC/VALENCIA"},‘roger_he@5uzh.com‘,False)
    #d._readPdf("")