From 20272c4b4ae9be8ff7bf440017b98795e9d661da Mon Sep 17 00:00:00 2001 From: Daniil Fajnberg Date: Wed, 3 Aug 2022 11:29:54 +0200 Subject: [PATCH] Minor changes --- src/orm2pydantic/sqla.py | 2 +- tests/informal.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/orm2pydantic/sqla.py b/src/orm2pydantic/sqla.py index ec7a7a8..06d96d8 100644 --- a/src/orm2pydantic/sqla.py +++ b/src/orm2pydantic/sqla.py @@ -54,7 +54,7 @@ def from_sqla(db_model: Type[DeclarativeMeta], incl_many_to_one: bool = True, in config: Type[BaseConfig] = OrmConfig, exclude: Container[str] = (), add_fields: dict[str, FieldDef] = None): assert isinstance(db_model, DeclarativeMeta) - assert not (incl_one_to_many and incl_many_to_one) + assert not (incl_one_to_many and incl_many_to_one), "Pydantic is unable to handle the circular relationship" fields = {} for attr in inspect(db_model).attrs: if attr.key in exclude: diff --git a/tests/informal.py b/tests/informal.py index 14d2beb..641762d 100644 --- a/tests/informal.py +++ b/tests/informal.py @@ -9,7 +9,6 @@ from orm2pydantic.sqla import from_sqla ORMBase = declarative_base() -engine = create_engine("sqlite://") def default_factory() -> str: return '1' @@ -65,11 +64,12 @@ class Address(AbstractBase): def main_test() -> None: + engine = create_engine("sqlite://") AbstractBase.metadata.create_all(engine) - from_sqla(StateProvince) - from_sqla(City) - from_sqla(Street) + _PydanticStateProvince = from_sqla(StateProvince) + _PydanticCity = from_sqla(City) + _PydanticStreet = from_sqla(Street) _PydanticAddress = from_sqla(Address) with Session(engine) as session: