неструктурирующий комплексный класс с каттрами - PullRequest
Новая
       24

неструктурирующий комплексный класс с каттрами

1 голос
/ 12 марта 2019

У меня проблемы с использованием attrs / cattrs для преобразования пользовательского класса данных в формат JSON. Мой класс имеет вид: 

@attr.s(auto_attribs=True)
class B:
    w: Set[datetime] = attr.ib()
    x: Set[str] = attr.ib()
    y: Set['A'] = attr.ib()
    z: Set[Tuple[datetime, str]] = attr.ib(factory=set)


@attr.s(auto_attribs=True, cmp=False)
class A:
    a: str = attr.ib()
    b: FrozenSet[Team] = attr.ib()
    c: FrozenSet[Tuple[datetime, str]] = attr.ib(factory=frozenset)
    d: Dict[Tuple[str, str], float] = attr.ib(factory=dict)

Проблема, с которой я сталкиваюсь, заключается в том, что когда я пытаюсь преобразовать в dict через cattrs.unstructure и наоборот через cattrs.structure, cattrs показывает ошибку, сообщающую мне, что я должен использовать ловушку: 

# Create instance of b
b_instance = B(...)
...
data = cattr.unstructure(b_instance)
print()
print(data)
print()
restored = cattr.structure(data, B)
assert b_instance == restored

ValueError: Unsupported type: <class 'datetime.datetime'>. Register a structure hook for it.

Я зарегистрировал ловушку для преобразования даты и времени в str с заданным форматом: 

import cattr
from datetime import datetime

time_format = '%Y-%m-%d %H:%M:%S'
cattr.register_unstructure_hook(datetime, lambda dt: dt.strftime(time_format))
cattr.register_structure_hook(datetime, lambda s, _: datetime.strptime(s, time_format))

но тогда я получаю следующую ошибку TypeError: strptime() argument 1 must be str, not datetime.datetime.

Мне нужно иметь возможность преобразовать экземпляр объекта в JSON и обратно, используя дату и время в формате %Y-%m-%d %H:%M:%S, как мне этого добиться?

1 Ответ

0 голосов
/ 19 марта 2019

Проблема с типом набора.Если он существует в типе, то cattr.unstructure не может его сериализовать.замените его на кортеж и все будет работать: 

from typing import Tuple, Set
from datetime import datetime

import cattr
import attr
import pytest


TIME_FORMAT = "%Y-%m-%d %H:%M:%S"
COMPLEX_TYPE = Tuple[Tuple[str, datetime]]
COMPLEX_TYPE_BAD = Set[Tuple[str, datetime]]

cattr.register_unstructure_hook(
    datetime, lambda dt: datetime.strftime(dt, format=TIME_FORMAT)
)
cattr.register_structure_hook(
    datetime, lambda dt_str, _: datetime.strptime(dt_str, TIME_FORMAT)
)


def test_cattrs_io_for_a():
    @attr.s(auto_attribs=True)
    class A:
        ds: datetime = attr.ib(factory=datetime.utcnow)

    a = A()
    a_ser = cattr.unstructure(a)
    a_des = cattr.structure(a_ser, A)
    assert isinstance(a_des, A)


def test_cattrs_io_for_b():
    @attr.s(auto_attribs=True)
    class B:
        ds: COMPLEX_TYPE = attr.ib(factory=set)

    b = B((("a", datetime.utcnow()), ("b", datetime.utcnow())))
    b_ser = cattr.unstructure(b)
    b_des = cattr.structure(b_ser, B)
    assert isinstance(b_des, B)


def test_cattrs_io_for_b_bad():
    @attr.s(auto_attribs=True)
    class B:
        ds: COMPLEX_TYPE_BAD = attr.ib(factory=set)

    b = B({("a", datetime.utcnow()), ("b", datetime.utcnow())})
    b_ser = cattr.unstructure(b)
    with pytest.raises(TypeError):
        b_des = cattr.structure(b_ser, B)
        assert isinstance(b_des, B)

Это похоже на проблему в cattrs https://github.com/Tinche/cattrs/issues/48

Будет исправлено в следующем выпуске cattrs.Или вы можете построить его из master ветви

...