diff --git a/amanuensis/db/__init__.py b/amanuensis/db/__init__.py index 09192ae..ecec332 100644 --- a/amanuensis/db/__init__.py +++ b/amanuensis/db/__init__.py @@ -12,4 +12,20 @@ from .models import ( ArticleContentRuleType, ArticleContentRule, Post, -) \ No newline at end of file +) + +__all__ = [ + 'DbContext', + 'User', + 'Lexicon', + 'Membership', + 'Character', + 'ArticleState', + 'Article', + 'IndexType', + 'ArticleIndex', + 'ArticleIndexRule', + 'ArticleContentRuleType', + 'ArticleContentRule', + 'Post', +] \ No newline at end of file diff --git a/amanuensis/db/database.py b/amanuensis/db/database.py index 22647a6..988a6ba 100644 --- a/amanuensis/db/database.py +++ b/amanuensis/db/database.py @@ -33,5 +33,10 @@ class DbContext(): # Create a thread-safe session factory self.session = scoped_session(sessionmaker(bind=self.engine)) + def __call__(self, *args, **kwargs): + """Provides shortcut access to session.execute.""" + return self.session.execute(*args, **kwargs) + def create_all(self): + """Initializes the database schema.""" ModelBase.metadata.create_all(self.engine)