python - 使用 yield 流式传输到 JSON 文件

Question

我有一个读取文件的函数，该文件读取 block 并在每次迭代后返回以下内容。

def foo_now():
    for file in all_files:
        key_id, final_text = readChunk(file)
        yield {
            key_id : {
                'desc': final_text
            }
        }

之前我将所有这些返回的字典存储到一个字典中，然后它们将整个字典转储到一个 JSON 文件中，使用:

def foo_earlier():
    temp_dict = dict()
    for file in all_files:
        key_id, final_text = readChunk(file)
        temp_dict.update({key_id : {'desc': final_text}})
    return temp_dict

def saveJSON(filename, data):
    with open(filename, 'wb') as outfile:
        json.dump(data, outfile)
    print "Data saved in " + filename

saveJSON("file.json", foo_earlier())

但现在我想在使用 foo_now() 从文件中读取每个字典时将其逐一转储。我应该怎么做？

Answer 1

for item in foo_now():
    with open("my_out.json","ab") as f:
        json.dump(item,f)

或

with open("my_out.json","ab") as f:
    for item in foo_now():
        json.dump(item,f)