blob: 48ef3b636cef210fdd2c170c626f60bdcf6a45c3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from contextlib import contextmanager
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,
)
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())
yield session
if autocommit:
session.commit()
|
