Расшифровать массивы JSON в Frozensets - PullRequest
Новая
       36

Расшифровать массивы JSON в Frozensets

0 голосов
/ 15 мая 2019

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

import json
class FrozensetDecoder(json.JSONDecoder): 
    def default(self, obj): 
        print(obj) 
        if isinstance(obj, list): 
            return frozenset(obj) 
        return obj 
    array = list = default 


In [8]: json.loads('[1,[2],3]', cls=FrozensetDecoder)                                                                                                                                                                                                                                                  
Out[8]: [1, [2], 3]

Но я хочу 

frozenset({1, frozenset({2}), 3})

1 Ответ

1 голос
/ 15 мая 2019

Я не знаком с тем подходом, который вы используете, переопределяя array и list как default функцию.

Вот код, который делает то, что вы хотите: 

import json
from json import scanner


class MyDecoder(json.JSONDecoder):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        # set up an alternative function for parsing arrays
        self.previous_parse_array = self.parse_array
        self.parse_array = self.json_array_frozenset
        # ensure that the replaced function gets used
        self.scan_once = scanner.py_make_scanner(self)

    def json_array_frozenset(self, s_and_end, scan_once, **kwargs):
        # call the parse_array that would have been used previously
        values, end = self.previous_parse_array(s_and_end, scan_once, **kwargs)
        # return the same result, but turn the `values` into a frozenset
        return frozenset(values), end


data = json.loads('[1,[2],3]', cls=MyDecoder)
print(data)

Обратите внимание, что результатом будет frozenset({1, 3, frozenset({2})}), а не frozenset({1, frozenset({2}), 3}), но так как множества неупорядочены, это не должно иметь значения.

...