diff --git a/royalnet/alchemist/__init__.py b/royalnet/alchemist/__init__.py index 6ec7f7c3..1050b2d3 100644 --- a/royalnet/alchemist/__init__.py +++ b/royalnet/alchemist/__init__.py @@ -1,5 +1,6 @@ from royalnet.typing import * import sqlalchemy as sa +import sqlalchemy.orm as saorm __all__ = ( @@ -14,6 +15,7 @@ class Alchemist: engine_args: Iterable[Any], engine_kwargs: Mapping[str, Any]): self.engine: sa.engine.Engine = sa.create_engine(*engine_args, **engine_kwargs) + self.Session: Union[saorm.sessionmaker, Type[saorm.Session]] = saorm.sessionmaker(bind=self.engine) def add_metadata(self, metadata: sa.MetaData): """Bind a MetaData object to the engine, and create all tables linked with it."""