Tablib не может преобразовать ответ JSON API в файл XLSX и выдает ошибку KeyError:0

Я пытаюсь преобразовать ответ JSON, полученный после вызова API, в файл Excel. Мне не нужно добавлять конкретные заголовки или получать только определенные данные, мне просто нужно все, что возвращает этот вызов.

Я нашел tablib библиотека

Мне удалось заставить его работать, когда я получал данные из GitLab, но после изменения полезной нагрузки на Tfs я получаю сообщение об ошибке, которое я не уверен, в чем проблема или как ее решить.

Это мой код:

import requests
import urllib3
import json
from requests.packages.urllib3.exceptions import InsecureRequestWarning
import tablib
import datetime
import time
import os

class gitlab():

    def get_closed():

        url = "https://IP:443/DefaultCollection/_apis/projects"

        payload = {}

        querystring = {"api-version":"4.1"}

        headers = {
            'Content-Type': "application/json-patch+json",
            'Authorization': "KEY" 
        }

        requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

        response = requests.request(
            "GET", url, headers=headers, data=payload, params=querystring,  verify=False)

        return json.loads(response.text)


if __name__ == "__main__":

    list_b = gitlab.get_closed() 

    print(list_b)
    data = tablib.Dataset()

    data.json = json.dumps(list_b)

    data_export = data.export('xlsx')

    filename = os.path.dirname(os.path.realpath(__file__)) +'/closed_' + str(datetime.date.today()) + '.xlsx'

    with open(filename, 'wb') as f: 
        f.write(data_export)  

    f.close()

Выполнение с python script где версия питона 3.7.2 показывает следующую ошибку:

{'count': 1, 'value': [{'id': 'ID', 'name': 'TFS', 'url': 'https://TFS/DefaultCollection/_apis/projects/PROJ', 'state': 'wellFormed', 'revision': 00, 'visibility': 'private'}]}

Traceback (most recent call last):  File ".\gitlab.py", line 94, in <module>    data.json = json.dumps(list_with_bugs, indent=4, ensure_ascii=False)  File "C:\Users\marialena\AppData\Local\Programs\Python\Python37\lib\site-packages\tablib\formats\_json.py", line 39, in import_set
    dset.dict = json.loads(in_stream)
  File "C:\Users\marialena\AppData\Local\Programs\Python\Python37\lib\site-packages\tablib\core.py", line 381, in _set_dict
    if isinstance(pickle[0], list):
KeyError: 0

Вы также можете увидеть ответ API в выходных данных. Почему tablib не удается преобразовать в Excel?

1 ответ

Решение

Dataset.json ожидает получения сериализованного списка. Код в вопросе передает сериализованный dictи это является причиной ошибки.

Глядя на данные, кажется, значение value ключ в словаре - это то, что требуется, поэтому передайте это в набор данных.

>>> import json
>>> import tablib
>>> d = {'count': 1, 'value': [{'id': 'ID', 'name': 'TFS', 'url': 'https://TFS/DefaultCollection/_apis/projects/PROJ', 'state': 'wellFormed', 'revision': 00, 'visibility': 'private'}]}
>>> ds = tablib.Dataset()
>>> ds.json = json.dumps(d['value'])
>>> with open('test.xlsx', 'wb') as f:
...     f.write(ds.export('xlsx'))

Я вижу, что вы подняли ошибку на трекере проблем проекта. Хотя такое поведение, возможно, и не является ошибкой, оно, безусловно, будет улучшением, если код выдаст более значимое сообщение об ошибке.

Другие вопросы по тегам