innopolis-fwd-project-backend

1
from sqlalchemy import create_engine
2
from sqlalchemy.ext.declarative import declarative_base
3
from sqlalchemy.orm import sessionmaker
4

5
SQLALCHEMY_DATABASE_URL = "sqlite:///./data.db"
6
# SQLALCHEMY_DATABASE_URL = "postgresql://user:password@postgresserver/db"
7

8
engine = create_engine(
9
    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
10
)
11
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
12

13
Base = declarative_base()
14

15

16
def get_db():
17
    db = SessionLocal()
18
    try:
19
        yield db
20
    finally:
21
        db.close()
22