# db.py

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, declarative_base
import urllib.parse

DB_USER = "root"
DB_PASSWORD = urllib.parse.quote_plus("wjdals2413@")
DB_HOST = "localhost"
DB_PORT = "3306"
DB_NAME = "gradproject"

DATABASE_URL = f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}?charset=utf8mb4"
# DATABASE_URL = "mysql://root:oIvlzNQSswVbBwNiTwFsLKdmAbFlLlDx@turntable.proxy.rlwy.net:44568/railway"
engine = create_engine(DATABASE_URL, echo=True)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

Base = declarative_base()


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()