从PDF文件中突出显示的注释中提取文本

时间:2014-01-10 17:24:16

标签: python qt pdf poppler

从昨天开始,我正在尝试使用python-poppler-qt4从一个pdf中的一些突出显示的注释中提取文本。

根据this documentation,看起来我必须使用Page.text()方法获取文本,从高亮注释传递一个Rectangle参数,我使用Annotation.boundary()。但我只得到空白文本。有人能帮我吗?我复制了下面的代码,并为我正在使用的PDF添加了一个链接。谢谢你的帮助!

import popplerqt4
import sys
import PyQt4


def main():

    doc = popplerqt4.Poppler.Document.load(sys.argv[1])
    total_annotations = 0
    for i in range(doc.numPages()):
        page = doc.page(i)
        annotations = page.annotations()
        if len(annotations) > 0:
            for annotation in annotations:
                if  isinstance(annotation, popplerqt4.Poppler.Annotation):
                    total_annotations += 1
                    if(isinstance(annotation, popplerqt4.Poppler.HighlightAnnotation)):
                        print str(page.text(annotation.boundary()))
    if total_annotations > 0:
        print str(total_annotations) + " annotation(s) found"
    else:
        print "no annotations found"

if __name__ == "__main__":
    main()

测试pdf: https://www.dropbox.com/s/10plnj67k9xd1ot/test.pdf

1 个答案:

答案 0 :(得分:5)

查看the documentation for Annotations似乎边界属性在规范化坐标中返回此注释的边界矩形。虽然这似乎是一个奇怪的决定,但我们可以简单地按比例缩放坐标。 page.pageSize().width().height()值。

import popplerqt4
import sys
import PyQt4


def main():

    doc = popplerqt4.Poppler.Document.load(sys.argv[1])
    total_annotations = 0
    for i in range(doc.numPages()):
        #print("========= PAGE {} =========".format(i+1))
        page = doc.page(i)
        annotations = page.annotations()
        (pwidth, pheight) = (page.pageSize().width(), page.pageSize().height())
        if len(annotations) > 0:
            for annotation in annotations:
                if  isinstance(annotation, popplerqt4.Poppler.Annotation):
                    total_annotations += 1
                    if(isinstance(annotation, popplerqt4.Poppler.HighlightAnnotation)):
                        quads = annotation.highlightQuads()
                        txt = ""
                        for quad in quads:
                            rect = (quad.points[0].x() * pwidth,
                                    quad.points[0].y() * pheight,
                                    quad.points[2].x() * pwidth,
                                    quad.points[2].y() * pheight)
                            bdy = PyQt4.QtCore.QRectF()
                            bdy.setCoords(*rect)
                            txt = txt + unicode(page.text(bdy)) + ' '

                        #print("========= ANNOTATION =========")
                        print(unicode(txt))

    if total_annotations > 0:
        print str(total_annotations) + " annotation(s) found"
    else:
        print "no annotations found"

if __name__ == "__main__":
    main()

此外,我决定连接.highlightQuads()以更好地表示实际突出显示的内容。

请注意我已附加到文字的每个四边形区域的显式<space>

在示例文档中,无法将返回的QString直接传递给print()str(),解决方法是使用unicode()代替。

我希望这有助于帮助我的人。

注意:页面旋转可能会影响缩放值,我无法对此进行测试。