from attr import define, has, fields
from cattr.gen import make_dict_unstructure_fn, make_dict_structure_fn, override
from cattr.converters import GenConverter


def strip_underscore_unstructure(cls):
    return make_dict_unstructure_fn(
        cls,
        converter,
        **{a.name: override(rename=a.name[1:]) for a in fields(cls) if a.name[0] == "_"}
    )


def strip_underscore_structure(cls):
    return make_dict_structure_fn(
        cls,
        converter,
        _cattrs_forbid_extra_keys=True,
        **{a.name: override(rename=a.name[1:]) for a in fields(cls) if a.name[0] == "_"}
    )


converter = GenConverter(forbid_extra_keys=True)
converter.register_structure_hook_factory(has, strip_underscore_structure)
converter.register_unstructure_hook_factory(has, strip_underscore_unstructure)


@define
class A:
    b: int
    _c: str

    @property
    def c(self):
        return self._c


a = A(1, "abc")


print(converter.unstructure(a))  # {'b': 1, 'c': 'abc'}


# Traceback (most recent call last):
#   File "/Users/clupo/Github/ufoLib2/test_cattrs.py", line 43, in <module>
#     converter.structure(converter.unstructure(a), A)
#   File "/Users/clupo/Github/cattrs/src/cattr/converters.py", line 300, in structure
#     return self._structure_func.dispatch(cl)(obj, cl)
#   File "<cattrs generated structure __main__.A>", line 8, in structure_A
# Exception: Extra fields in constructor for A: c
converter.structure(converter.unstructure(a), A)