Python 获取 API 响应并添加到 JSON 密钥错误

Question

我有一个脚本，它从 JSON 文件中获取一个 ID，将其添加到 URL 中以用于 API 请求。 目的是让一个循环遍历 1000 个 id 并推断出一个 JSON 文件，其中包含所有信息。

当前代码调用第一个请求并创建并填充 JSON 文件，但在循环中运行时会抛出一个键错误。

import json
import requests

fname = "NewdataTest.json"

def write_json(data, fname):
    fname = "NewdataTest.json"
    with open(fname, "w") as f:
        json.dump(data, f, indent = 4)
    with open (fname) as json_file:
        data = json.load(json_file)
        temp = data[0]
            #print(newData) 
        y = newData
        data.append(y)

# Read test.json to get tmdb IDs 
tmdb_ids = []
with open('test.json', 'r') as json_fp:
    imdb_info = json.load(json_fp)

tmdb_ids = [movie_info['tmdb_id'] for movies_chunk in imdb_info for movie_index, movie_info in movies_chunk.items()]
# Add IDs to API call URL
for tmdb_id in tmdb_ids:
    print("https://api.themoviedb.org/3/movie/" + str(tmdb_id) + "?api_key=****")
    # Send API Call
    response_API = requests.get("https://api.themoviedb.org/3/movie/" + str(tmdb_id) + "?api_key=****")
    # Check API Call Status
    print(response_API.status_code)
    write_json((response_API.json()), "NewdataTest.json")   
            

错误是在这一行“temp = data[0]”我试过打印数据键，没有。 在这一点上，我不知道我在哪里，因为我已经破解了它，它几乎不像一段有凝聚力的代码。 我的目标是制作一个简单的 function 从 JSON 获取数据，一个生成 API 调用 URL，一个将结果写入新的 JSON。

API 响应 JSON 的示例：

{
    "adult": false,
    "backdrop_path": "/e1cC9muSRtAHVtF5GJtKAfATYIT.jpg",
    "belongs_to_collection": null,
    "budget": 0,
    "genres": [
        {
            "id": 10749,
            "name": "Romance"
        },
        {
            "id": 35,
            "name": "Comedy"
        }
    ],
    "homepage": "",
    "id": 1063242,
    "imdb_id": "tt24640474",
    "original_language": "fr",
    "original_title": "Disconnect: The Wedding Planner",
    "overview": "After falling victim to a scam, a desperate man races the clock as he attempts to plan a luxurious destination wedding for an important investor.",
    "popularity": 34.201,
    "poster_path": "/tGmCxGkVMOqig2TrbXAsE9dOVvX.jpg",
    "production_companies": [],
    "production_countries": [
        {
            "iso_3166_1": "KE",
            "name": "Kenya"
        },
        {
            "iso_3166_1": "NG",
            "name": "Nigeria"
        }
    ],
    "release_date": "2023-01-13",
    "revenue": 0,
    "runtime": 107,
    "spoken_languages": [
        {
            "english_name": "English",
            "iso_639_1": "en",
            "name": "English"
        },
        {
            "english_name": "Afrikaans",
            "iso_639_1": "af",
            "name": "Afrikaans"
        }
    ],
    "status": "Released",
    "tagline": "",
    "title": "Disconnect: The Wedding Planner",
    "video": false,
    "vote_average": 5.8,
    "vote_count": 3
}

Answer 1

您可以将 API 调用的所有结果存储到一个列表中，然后将此列表以 Json 格式保存到一个文件中。 例如：

#...

all_data = []

for tmdb_id in tmdb_ids:
    print("https://api.themoviedb.org/3/movie/" + str(tmdb_id) + "?api_key=****")
    # Send API Call
    response_API = requests.get("https://api.themoviedb.org/3/movie/" + str(tmdb_id) + "?api_key=****")

    # Check API Call Status
    print(response_API.status_code)

    if response_API.status_code == 200:
        # store the Json data in a list:
        all_data.append(response_API.json())

# write the list to file
with open('output.json', 'w') as f_out:
    json.dump(all_data, f_out, indent=4)

这将生成output.json ，所有响应均采用 Json 格式。