使用Textract，如何从pdf文件中提取表格并通过.py脚本将其输出到csv文件中？

问题描述

我想使用textract（通过aws cli）从pdf文件（位于s3位置）中提取表格并将其导出到csv文件中。我曾尝试编写.py脚本，但正在努力读取文件。 欢迎编写.py脚本的任何建议。

这是我当前的脚本。我遇到了错误： get_table_csv_results中的文件“ extract-table.py”，第63行 bash：文件：找不到命令 blocks = response ['Blocks'] KeyError：“块”

import webbrowser,os
import json
import boto3
import io
from io import BytesIO
import sys
from pprint import pprint





def get_rows_columns_map(table_result,blocks_map):
rows = {}
for relationship in table_result['Relationships']:
    if relationship['Type'] == 'CHILD':
        for child_id in relationship['Ids']:
            cell = blocks_map[child_id]
            if cell['BlockType'] == 'CELL':
                row_index = cell['RowIndex']
                col_index = cell['ColumnIndex']
                if row_index not in rows:
                    # create new row
                    rows[row_index] = {}
                    
                # get the text value
                rows[row_index][col_index] = get_text(cell,blocks_map)
return rows
def get_text(result,blocks_map):
text = ''
if 'Relationships' in result:
    for relationship in result['Relationships']:
        if relationship['Type'] == 'CHILD':
            for child_id in relationship['Ids']:
                word = blocks_map[child_id]
                if word['BlockType'] == 'WORD':
                    text += word['Text'] + ' '
                if word['BlockType'] == 'SELECTION_ELEMENT':
                    if word['SelectionStatus'] =='SELECTED':
                        text +=  'X '    
def get_table_csv_results(file_name):



with open(file_name,'rb') as file:
    img_test = file.read()
    bytes_test = bytearray(img_test)
    print('Image loaded',file_name)

# process using image bytes
# get the results
client = boto3.client('textract')

#Response
response = client.start_document_text_detection(
DocumentLocation={
    'S3Object': {
        'Bucket': s3BucketName,'Name': documentName
    }
})

# Get the text blocks
blocks=response['Blocks']
pprint(blocks)

blocks_map = {}
table_blocks = []
for block in blocks:
    blocks_map[block['Id']] = block
    if block['BlockType'] == "TABLE":
        table_blocks.append(block)

if len(table_blocks) <= 0:
    return "<b> NO Table FOUND </b>"

csv = ''
for index,table in enumerate(table_blocks):
    csv += generate_table_csv(table,blocks_map,index +1)
    csv += '\n\n'

return csv
def generate_table_csv(table_result,table_index):
rows = get_rows_columns_map(table_result,blocks_map)



table_id = 'Table_' + str(table_index)

# get cells.
csv = 'Table: {0}\n\n'.format(table_id)

for row_index,cols in rows.items():
    
    for col_index,text in cols.items():
        csv += '{}'.format(text) + ","
    csv += '\n'
    
csv += '\n\n\n'
return csv
def main(file_name):
table_csv = get_table_csv_results(file_name)

output_file = 'output.csv'

# replace content
with open(output_file,"wt") as fout:
    fout.write(table_csv)

# show the results
print('CSV OUTPUT FILE: ',output_file)


# Document
s3BucketName = "chrisyou.sagemi.com"
documentName = "DETAIL.pdf"
if __name__ == "__main__":
file_name = sys.argv[1]
main(file_name)

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）