gitscope
← marzneshin / v0.5.0 / app/db/base.py
https://dev.kronoz.win/r/marzneshin.git
app/db/base.py · Python
Blame Raw
 1from sqlalchemy import create_engine
 2from sqlalchemy.orm import declarative_base
 3from sqlalchemy.orm import sessionmaker
 4
 5from app.config.env import SQLALCHEMY_DATABASE_URL
 6
 7IS_SQLITE = SQLALCHEMY_DATABASE_URL.startswith("sqlite")
 8
 9if IS_SQLITE:
10    engine = create_engine(
11        SQLALCHEMY_DATABASE_URL,
12        connect_args={"check_same_thread": False},
13        # echo=True
14    )
15else:
16    engine = create_engine(
17        SQLALCHEMY_DATABASE_URL,
18        pool_size=10,
19        max_overflow=30,
20        pool_recycle=3600,
21        pool_timeout=10,
22    )
23
24SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
25Base = declarative_base()