ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 03:24:20 +00:00
Code Activity

🔧 Improve and rename SQLAlchemyPDA

Browse source
This commit is contained in:
Steffo 2021-04-13 04:30:42 +02:00
parent c6ac960f2b
commit 98846f5c56
Signed by: steffo
GPG key ID: 6965406171929D01
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
royalnet/engineer/pda.py
View file

@ -254,10 +254,10 @@ class ConversationListPDA(PDA, metaclass=abc.ABCMeta):
return full
class WithDatabaseSession(ConversationListPDA, metaclass=abc.ABCMeta):
class SQLAlchemyPDA(ConversationListPDA, metaclass=abc.ABCMeta):
"""
A :class:`.ConversationListPDA` with database support provided by :mod:`sqlalchemy`\\ by extending
:meth:`._conversation_kwargs` with the ``_session`` kwarg.
Extends :class:`.ConversationListPDA` with database support provided by :mod:`sqlalchemy`\\ by adding the
``_session`` kwarg to the :meth:`._conversation_kwargs`.
"""
def __init__(self, engine: sqlalchemy.engine.Engine, conversation_kwargs: dict[str, t.Any]):
@ -271,15 +271,14 @@ class WithDatabaseSession(ConversationListPDA, metaclass=abc.ABCMeta):
with self.Session(future=True) as session:
yield {
"_pda": self,
"_conv": conv,
"_session": session,
**self.conversation_kwargs,
**super()._conversation_kwargs(conv=conv),
}
__all__ = (
"PDA",
"ConversationListPDA",
"WithDatabaseSession",
"SQLAlchemyPDA",
)