blob: 4d83649d356913570240bb4a50414256d7a94656 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from contextlib import contextmanager
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, scoped_session
from sqlmodel import Session
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():
yield from get_session()
|
