refac/enh: db session sharing
This commit is contained in:
1 parent
d4de26bd05
commit
2041ab483e
20 files changed
+600
-562
No files matched your search
@@ -160,3 +160,13 @@ def get_session():
|
||||
|
||||
|
||||
get_db = contextmanager(get_session)
|
||||
|
||||
|
||||
@contextmanager
|
||||
def get_db_context(db: Optional[Session] = None):
|
||||
if db:
|
||||
yield db
|
||||
else:
|
||||
with get_db() as session:
|
||||
yield session
|
||||
|
||||
Reference in new issue
Block a user