Python实现PDF 转txt 和html转txt

时间：2019-01-29 18:09:52 阅读：268 评论：0 收藏：0 [点我收藏+]

标签：exce action horizon dev box obj std xtext ndt

# -*- coding: utf-8 -*-
from HTMLParser import HTMLParser
from re import sub
from sys import stderr
from traceback import print_exc
from pdfminer.pdfparser import  PDFParser
from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdfpage import PDFPage
from pdfminer.pdfdevice import PDFDevice
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import PDFPageAggregator
from pdfminer.layout import LTTextBoxHorizontal,LAParams
from pdfminer.pdfpage import PDFTextExtractionNotAllowed
import  re


class pythonNToTxt(HTMLParser):
    def __init__(self):
        HTMLParser.__init__(self)
        self.__text = []

    def handle_data(self, data):
        text = data.strip()
        if len(text) > 0:
            text = sub(‘[ \t\r\n]+‘, ‘ ‘, text)
            self.__text.append(text + ‘ ‘)

    def handle_starttag(self, tag, attrs):
        if tag == ‘p‘:
            self.__text.append(‘\n\n‘)
        elif tag == ‘br‘:
            self.__text.append(‘\n‘)

    def handle_startendtag(self, tag, attrs):
        if tag == ‘br‘:
            self.__text.append(‘\n\n‘)

    def text(self):
        return ‘‘.join(self.__text).strip()


def dehtml(text):
    try:
        parser = pythonNToTxt()
        parser.feed(text)
        parser.close()
        return parser.text()
    except:
        print_exc(file=stderr)
        return text

def html_to_txt(fileobject,saveName):
    text = r‘‘‘
        <html>
            <body>
                <b>Project:</b> DeHTML<br>
                <b>Description</b>:<br>
                <p>由HTML转换成txt文件.从HTML文件读取，存入test3.txt</p>
            </body>
        </html>
    ‘‘‘
    neirong1 = open(fileobject, ‘rb‘)
    neirong = neirong1.read()
    print neirong
    # for line in neirong:
    #     #Path = open(‘af58a19ce7b54986a7515f330a48cde3.pdf‘, ‘rb‘)
    #     print(dehtml(line))
    #     with open(‘%s‘ % (saveName), ‘a‘) as f:
    #         # results = dehtml(text).encode(‘utf-8‘)
    #         results = dehtml(text)
    #         f.write(results + "\n")
    with open(‘%s‘ % (saveName), ‘a‘) as f:
        results = dehtml(neirong)
        f.write(results + "\n")


def pdf_to_txt(filename, Save_name):
    fileobject = open(filename, ‘rb‘)
    parser = PDFParser(fileobject)
    document = PDFDocument(parser)

    if not document.is_extractable:
        raise PDFTextExtractionNotAllowed
    else:
        rsrcmgr = PDFResourceManager()
        laparams = LAParams()
        device = PDFPageAggregator(rsrcmgr, laparams=laparams)
        interpreter = PDFPageInterpreter(rsrcmgr, device)

        for page in PDFPage.create_pages(document):
            interpreter.process_page(page)
            layout = device.get_result()
            for x in layout:
                if(isinstance(x, LTTextBoxHorizontal)):
                    with open(‘%s‘ % (Save_name), ‘a‘) as f:
                        results = x.get_text().encode(‘utf-8‘)
                        f.write(results + "\n")

# 分支判断
filename = ‘test3.html‘

fenzhihouzhui = re.findall(r‘.*(\..*)‘, str(filename))[0]
if fenzhihouzhui == ‘.pdf‘ or fenzhihouzhui == ‘.PDF‘:
    pdf_to_txt(filename, ‘3.txt‘)
elif fenzhihouzhui == ‘.html‘ or fenzhihouzhui == ‘.HTML‘:
    html_to_txt(filename, ‘wenben1.txt‘)


正则说明：