python3,使用泛型集合序列化/反序列化类

时间:2021-01-30 14:28:42

标签: json python-3.x generics

我正在尝试使用泛型集合定义类,将其序列化/反序列化到/从 json

import json
from typing import NamedTuple
from typing import Sequence

class ColumnDef(NamedTuple):
    name: str


class RunConfig(NamedTuple):
    columns: Sequence[ColumnDef]


def read_run_config(json_config: str) -> RunConfig:
    loaded_json = json.loads(json_config)
    return RunConfig(**loaded_json)

并测试:

import unittest
from model.RunConfig import read_run_config
from model.RunConfig import ColumnDef


class TestReadRunConfig(unittest.TestCase):

    def test_read_run_config(self):
        run_config_json = """
        {
           
            "columns": [
                    {
                        "name": "id"
                    }        
                ]
        }
        """
        run_config = read_run_config(run_config_json)
        print(f"{run_config}")
        self.assertTrue(len(run_config.columns) == 1)
        self.assertEqual(run_config.columns[0], ColumnDef(name="id"))


if __name__ == '__main__':
    unittest.main()

输出:

RunConfig(columns=[{'name': 'id'}])

失败:

AssertionError: {'name': 'id'} != ColumnDef(name='id')

为什么json lib将ColumnDef反序列化为Dictionary,而不是类实例?

1 个答案:

答案 0 :(得分:0)

使其发挥作用的正确类组合:

from typing import NamedTuple
from typing import List
from typing_json import loads


class RunConfig(NamedTuple):
    # Sequence didn't work, have no idea why
    columns: List[ColumnDef]

# typing_json.loads works correctly
def read_run_config(json_config: str) -> RunConfig:
    return loads(json_config, RunConfig)
相关问题