refac: get_db_context
This commit is contained in:
@@ -164,7 +164,7 @@ get_db = contextmanager(get_session)
|
|||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def get_db_context(db: Optional[Session] = None):
|
def get_db_context(db: Optional[Session] = None):
|
||||||
if db:
|
if isinstance(db, Session):
|
||||||
yield db
|
yield db
|
||||||
else:
|
else:
|
||||||
with get_db() as session:
|
with get_db() as session:
|
||||||
|
|||||||
Reference in New Issue
Block a user