Подтвердить что ты не робот

Сериализовать в JSON кодированные base64 данные

Я пишу script для автоматизации генерации данных для демонстрации, и мне нужно сериализовать в JSON некоторые данные. Часть этих данных - это изображение, поэтому я закодировал его в base64, но когда я пытаюсь запустить мой script, я получаю:

Traceback (most recent call last):
  File "lazyAutomationScript.py", line 113, in <module>
    json.dump(out_dict, outfile)
  File "/usr/lib/python3.4/json/__init__.py", line 178, in dump
    for chunk in iterable:
  File "/usr/lib/python3.4/json/encoder.py", line 422, in _iterencode
    yield from _iterencode_dict(o, _current_indent_level)
  File "/usr/lib/python3.4/json/encoder.py", line 396, in _iterencode_dict
    yield from chunks
  File "/usr/lib/python3.4/json/encoder.py", line 396, in _iterencode_dict
    yield from chunks
  File "/usr/lib/python3.4/json/encoder.py", line 429, in _iterencode
    o = _default(o)
  File "/usr/lib/python3.4/json/encoder.py", line 173, in default
    raise TypeError(repr(o) + " is not JSON serializable")
  TypeError: b'iVBORw0KGgoAAAANSUhEUgAADWcAABRACAYAAABf7ZytAAAABGdB...
     ...
   BF2jhLaJNmRwAAAAAElFTkSuQmCC' is not JSON serializable

Насколько я знаю, base64-encoded-whatever (изображение PNG в этом случае) является просто строкой, поэтому он должен представлять проблему для сериализации. Что мне не хватает?

4b9b3361

Ответ 1

Вы должны быть осторожны с типами данных.

Если вы читаете двоичное изображение, вы получаете байты. Если вы кодируете эти байты в base64, вы снова получите... байты! (см. документацию по b64encode)

json не может обрабатывать необработанные байты, поэтому вы получаете ошибку.

Я только что написал несколько примеров с комментариями, надеюсь, это поможет:

from base64 import b64encode
from json import dumps

ENCODING = 'utf-8'
IMAGE_NAME = 'spam.jpg'
JSON_NAME = 'output.json'

# first: reading the binary stuff
# note the 'rb' flag
# result: bytes
with open(IMAGE_NAME, 'rb') as open_file:
    byte_content = open_file.read()

# second: base64 encode read data
# result: bytes (again)
base64_bytes = b64encode(byte_content)

# third: decode these bytes to text
# result: string (in utf-8)
base64_string = base64_bytes.decode(ENCODING)

# optional: doing stuff with the data
# result here: some dict
raw_data = {IMAGE_NAME: base64_string}

# now: encoding the data to json
# result: string
json_data = dumps(raw_data, indent=2)

# finally: writing the json string to disk
# note the 'w' flag, no 'b' needed as we deal with text here
with open(JSON_NAME, 'w') as another_open_file:
    another_open_file.write(json_data)

Ответ 2

Альтернативным решением будет кодирование вещей на лету с помощью специального кодировщика:

import json
from base64 import b64encode

class Base64Encoder(json.JSONEncoder):
    # pylint: disable=method-hidden
    def default(self, o):
        if isinstance(o, bytes):
            return b64encode(o).decode()
        return json.JSONEncoder.default(self, o)

Определившись с этим, вы можете сделать:

m = {'key': b'\x9c\x13\xff\x00'}
json.dumps(m, cls=Base64Encoder)

Он будет производить:

'{"key": "nBP/AA=="}'