Сохранение вывода на печать в формате dict или JSON

Question

У меня есть следующий код, который использует boto3 для AWS.

import boto3
from trp import Document

# Document
s3BucketName = "bucket"
documentName = "doc.png"

# Amazon Textract client
textract = boto3.client('textract')

# Call Amazon Textract
response = textract.analyze_document(
    Document={
        'S3Object': {
            'Bucket': s3BucketName,
            'Name': documentName
        }
    },
    FeatureTypes=["FORMS"])

#print(response)

doc = Document(response)

for page in doc.pages:
    # Print fields
    print("Fields:")
    for field in page.form.fields:
        print("Key: {}, Value: {}".format(field.key, field.value))

Я пытаюсь сохранить вывод этой функции как dict, JSON или CSV, но я еще не опытный программист на python.

Я пробовал это:

key_map = {}
filepath = 'output.txt'
with open(filepath) as fp:
    line = fp.readline()
    cnt = 1
    while line:
        for page in doc.pages:
            # Print fields
            print("Fields:")
            for field in page.form.fields:
                #print("Key: {}, Value: {}".format(field.key, field.value))
                key_map[str(field.key, field.value)] = cnt
                line = fp.readline()
                cnt +=1

Но я не думаю, что это решение работает. Любые советы о том, как сохранить выходные данные цикла for в формате JSON?

Answer 1

Если вы хотите использовать как вывод csv, вы можете использовать модуль csv как:

import csv

doc = Document(response)

with open('aws_doc.csv', mode='w') as aws_field_file:
    field_write = csv.writer(aws_field_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)

    for page in doc.pages:
        for field in page.form.fields:
            # This will write it as your <key>, <value>
            field_write.writerow([field.key, field.value])

Если вам нужны заголовки в файле, вы также можете использовать DictWriter, которыйоблегчил бы вам просто передать словарь: https://docs.python.org/3.4/library/csv.html#csv.DictWriter