fastapi_homework/database.py at main · koheon2/fastapi_homework

24 lines (19 loc) · 620 Bytes

from sqlalchemy import create_engine

from sqlalchemy.orm import sessionmaker, declarative_base

# SQLite 데이터베이스 URL

SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"

# 데이터베이스 엔진 생성

engine = create_engine(

SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False} # SQLite는 멀티스레드 사용 시 설정 필요

)

# 세션 로컬 클래스 생성

SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 기본 클래스 선언

Base = declarative_base()

# 의존성 주입

def get_db():

db = SessionLocal()

try:

yield db

finally:

db.close()

Provide feedback