From 6b6f20a032eb31d747366c8bc270ca81731eaf6a Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Fri, 4 Sep 2026 17:09:24 +0200 Subject: Fixed pre-commit issues --- db/session.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'db/session.py') diff --git a/db/session.py b/db/session.py index 82c0211..48ef3b6 100644 --- a/db/session.py +++ b/db/session.py @@ -1,21 +1,28 @@ from contextlib import contextmanager -from sqlalchemy.orm import sessionmaker, scoped_session -from sqlmodel import create_engine -from sqlmodel import Session +from sqlalchemy.orm import scoped_session, sessionmaker +from sqlmodel import Session, create_engine from conf import DATABASE_URL from utils.json import json_serializer - -engine = create_engine(DATABASE_URL, echo=True, future=True, json_serializer=json_serializer, pool_size=100, pool_recycle=3600) +engine = create_engine( + DATABASE_URL, + echo=True, + future=True, + json_serializer=json_serializer, + pool_size=100, + pool_recycle=3600, +) session_factory = sessionmaker(engine, class_=Session) ScopedSession = scoped_session(session_factory=session_factory) + def get_session(): with ScopedSession() as session: yield session + @contextmanager def get_session_context(autocommit=True): session = next(get_session()) -- cgit v1.3