1

我想从我拥有的一堆 PDF 中提取表格。为此，我使用 AWS Textract Python 管道。

请告知我如何在没有 SNS 和 SQS 的情况下做到这一点？我希望它是同步的：为我的管道提供一个 PDF 文件，调用 AWS Textract 并获取结果。

这是我同时使用的，请告知我应该更改什么：

import boto3
import time

def startJob(s3BucketName, objectName):
    response = None
    client = boto3.client('textract')
    response = client.start_document_text_detection(
    DocumentLocation={
        'S3Object': {
            'Bucket': s3BucketName,
            'Name': objectName
        }
    })

    return response["JobId"]

def isJobComplete(jobId):
    # For production use cases, use SNS based notification 
    # Details at: https://docs.aws.amazon.com/textract/latest/dg/api-async.html
    time.sleep(5)
    client = boto3.client('textract')
    response = client.get_document_text_detection(JobId=jobId)
    status = response["JobStatus"]
    print("Job status: {}".format(status))

    while(status == "IN_PROGRESS"):
        time.sleep(5)
        response = client.get_document_text_detection(JobId=jobId)
        status = response["JobStatus"]
        print("Job status: {}".format(status))

    return status

def getJobResults(jobId):

    pages = []

    client = boto3.client('textract')
    response = client.get_document_text_detection(JobId=jobId)

    pages.append(response)
    print("Resultset page recieved: {}".format(len(pages)))
    nextToken = None
    if('NextToken' in response):
        nextToken = response['NextToken']

    while(nextToken):

        response = client.get_document_text_detection(JobId=jobId, NextToken=nextToken)

        pages.append(response)
        print("Resultset page recieved: {}".format(len(pages)))
        nextToken = None
        if('NextToken' in response):
            nextToken = response['NextToken']

    return pages

# Document
s3BucketName = "ki-textract-demo-docs"
documentName = "Amazon-Textract-Pdf.pdf"

jobId = startJob(s3BucketName, documentName)
print("Started job with id: {}".format(jobId))
if(isJobComplete(jobId)):
    response = getJobResults(jobId)

#print(response)

# Print detected text
for resultPage in response:
    for item in resultPage["Blocks"]:
        if item["BlockType"] == "LINE":
            print ('\033[94m' +  item["Text"] + '\033[0m')
4

1 回答 1

2

您目前无法直接与 Textract 同步处理 PDF 文档。从文本文档：

Amazon Textract 同步操作 (DetectDocumentText和AnalyzeDocument) 支持 PNG 和 JPEG 图像格式。异步操作 ( StartDocumentTextDetection, StartDocumentAnalysis) 也支持 PDF 文件格式。

一种解决方法是将 PDF 文档转换为代码中的图像，然后对这些图像使用同步 API 操作来处理文档。

于 2020-06-03T13:43:06.363 回答