问Micropython: json文件中的字节数组

EN

import math, ujson, utime


class JSON(object):
    @staticmethod
    def convert(data:dict, convert_keys=None) -> dict:
        if isinstance(convert_keys, (tuple, list)):
            for key in convert_keys:
                if isinstance(data[key], (bytes, bytearray)):
                    data[key] = int.from_bytes(data[key], 'big')
                elif isinstance(data[key], int):
                    data[key] = data[key].to_bytes(1 if not data[key]else int(math.log(data[key], 256)) + 1, 'big')
        return data
    
    @staticmethod
    def save(filename:str, data:dict, convert_keys=None) -> None:
        #dump doesn't seem to like working directly with open
        with open(filename, 'w') as doc:
            ujson.dump(JSON.convert(data, convert_keys), doc) 

    @staticmethod
    def open(filename:str, convert_keys=None) -> dict:
        return JSON.convert(ujson.load(open(filename, 'r')), convert_keys)
    

#example with both styles of bytes for the sake of being thorough
json_data = dict(address=bytearray(b'\xFF\xEE\xDD\xCC'), data=b'\x00\x01\02\x03', date=utime.mktime(utime.localtime()))

keys = ['address', 'data'] #list of keys to convert to int/bytes

JSON.save('test.json', json_data, keys)

json_data = JSON.open('test.json', keys)

print(json_data) #{'date': 1621035727, 'data': b'\x00\x01\x02\x03', 'address': b'\xff\xee\xdd\xcc'}