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()