diff --git a/src/orm2pydantic/__init__.py b/src/orm2pydantic/__init__.py index a9c8e19..23aa0d7 100644 --- a/src/orm2pydantic/__init__.py +++ b/src/orm2pydantic/__init__.py @@ -1 +1 @@ -from .sqla import from_sqla +from .sqla import sqla2pydantic diff --git a/src/orm2pydantic/sqla.py b/src/orm2pydantic/sqla.py index a608c92..7053087 100644 --- a/src/orm2pydantic/sqla.py +++ b/src/orm2pydantic/sqla.py @@ -34,7 +34,7 @@ from .utils import resolve_dotted_path __all__ = [ 'field_from_column', 'field_from_relationship', - 'from_sqla' + 'sqla2pydantic' ] @@ -104,9 +104,9 @@ def field_from_relationship(rel_prop: RelationshipProperty) -> FieldDef: return list[rel_prop.mapper.class_.__name__], Field(default=None) -def from_sqla(db_model: Type[DeclarativeMeta], config: Type[BaseConfig] = OrmConfig, exclude: Container[str] = (), - incl_relationships: bool = True, add_fields: dict[str, FieldDef] = None, - add_local_ns: dict[str, BaseModel] = None): +def sqla2pydantic(db_model: Type[DeclarativeMeta], config: Type[BaseConfig] = OrmConfig, exclude: Container[str] = (), + incl_relationships: bool = True, add_fields: dict[str, FieldDef] = None, + add_local_ns: dict[str, BaseModel] = None): """ Takes an SQLAlchemy ORM model class and returns a matching Pydantic model class. diff --git a/tests/informal.py b/tests/informal.py index dabdd14..f740e9e 100644 --- a/tests/informal.py +++ b/tests/informal.py @@ -5,7 +5,7 @@ from sqlalchemy.sql.functions import now as db_now from sqlalchemy.sql.schema import Column, ForeignKey as FKey from sqlalchemy.sql.sqltypes import Integer, String, TIMESTAMP, Unicode -from orm2pydantic.sqla import from_sqla +from orm2pydantic.sqla import sqla2pydantic ORMBase = declarative_base() @@ -65,10 +65,10 @@ def main_test() -> None: engine = create_engine("sqlite://") AbstractBase.metadata.create_all(engine) - _PydanticStateProvince = from_sqla(StateProvince, exclude=['cities']) - _PydanticCity = from_sqla(City, exclude=['streets']) - _PydanticStreet = from_sqla(Street, exclude=['addresses']) - _PydanticAddress = from_sqla(Address) + _PydanticStateProvince = sqla2pydantic(StateProvince, exclude=['cities']) + _PydanticCity = sqla2pydantic(City, exclude=['streets']) + _PydanticStreet = sqla2pydantic(Street, exclude=['addresses']) + _PydanticAddress = sqla2pydantic(Address) with Session(engine) as session: bavaria = StateProvince(country="de", name="Bavaria")