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