Передача списка словарей в функцию, которая принимает файл json - PullRequest
0 голосов
/ 30 апреля 2018

У меня есть список словарей, которые я хочу передать в функцию, которая принимает файл json.

Мой текущий подход заключается в том, чтобы (1) преобразовать список диктов в json с использованием json.dumps() и (2) передать его в качестве аргумента через StringIO()

Я получаю ошибку. AttributeError: 'str' object has no attribute 'decode'

Я не уверен, как решить эту проблему или даже если я использую правильный подход. Любая помощь приветствуется.

Редактировать:

Пример кода

import tdclient as td
import json
from io import StringIO

l = [{'a': 1}, {'b':2}, {'c':3}]
l_json = json.dumps(l)

with td.Client(TD_APIKEY) as con:
    con.import_file('test', 'temp', 'json', StringIO(l_json))

и ошибка:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-64-d81f043acd58> in <module>()
      7 
      8 with td.Client(TD_APIKEY) as con:
----> 9     con.import_file('test', 'temp', 'json', StringIO(l_json))

~/anaconda3/lib/python3.6/site-packages/tdclient/client.py in import_file(self, db_name, table_name, format, file, unique_id)
    605         Returns: float represents the elapsed time to import data
    606         """
--> 607         return self.api.import_file(db_name, table_name, format, file, unique_id=unique_id)
    608 
    609     def results(self):

~/anaconda3/lib/python3.6/site-packages/tdclient/import_api.py in import_file(self, db, table, format, file, unique_id, **kwargs)
     61         Returns: float represents the elapsed time to import data
     62         """
---> 63         with contextlib.closing(self._prepare_file(file, format, **kwargs)) as fp:
     64             size = os.fstat(fp.fileno()).st_size
     65             return self.import_data(db, table, "msgpack.gz", fp, size, unique_id=unique_id)

~/anaconda3/lib/python3.6/site-packages/tdclient/api.py in _prepare_file(self, file_like, fmt, **kwargs)
    426             packer = msgpack.Packer()
    427             with contextlib.closing(self._read_file(file_like, fmt, **kwargs)) as items:
--> 428                 for item in items:
    429                     try:
    430                         mp = packer.pack(item)

~/anaconda3/lib/python3.6/site-packages/tdclient/api.py in _read_json_file(self, file_like, **kwargs)
    471         # current impl doesn't torelate any JSON parse error
    472         for s in file_like:
--> 473             record = json.loads(s.decode("utf-8"))
    474             self._validate_record(record)
    475             yield record

AttributeError: 'str' object has no attribute 'decode'

1 Ответ

0 голосов
/ 30 апреля 2018

Import_file ожидает BytesIO, а не StringIO (см. Их test )

Попробуйте это

import tdclient as td
import json
from io import BytesIO

l = [{'a': 1}, {'b':2}, {'c':3}]
l_json = json.dumps(l)

with td.Client(2) as con:
    con.import_file('test', 'temp', 'json', BytesIO(bytes(l_json, 'utf-8')))

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

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...