https://dev.kronoz.win/r/marzneshin.git
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()