Сериализовать в 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 в этом случае) является просто строкой, поэтому он должен представлять проблему для сериализации. Что мне не хватает?
Ответы
Ответ 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=="}'