from contextlib import contextmanager

from sqlalchemy.orm import sessionmaker, scoped_session
from sqlmodel import create_engine
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(autocommit=True):
    session = next(get_session())

    yield session

    if autocommit:
        session.commit()