# import os | |
# from sqlalchemy import create_engine | |
# from sqlalchemy.ext.declarative import declarative_base | |
# from sqlalchemy.orm import sessionmaker | |
# from sqlalchemy.exc import SQLAlchemyError | |
# # MySQL ๋ฐ์ดํฐ๋ฒ ์ด์ค ์ค์ | |
# DATABASE_URL = os.getenv("DATABASE_URL", "mysql+mysqlconnector://root:[email protected]:3306/chathess") | |
# # SQLAlchemy ์์ง ์์ฑ (์์ธ ์ฒ๋ฆฌ ์ถ๊ฐ) | |
# try: | |
# engine = create_engine(DATABASE_URL) | |
# print("Database engine created successfully.") | |
# except SQLAlchemyError as e: | |
# print("Failed to create database engine.") | |
# print("Error:", e) | |
# engine = None | |
# # ์ธ์ ์์ฑ (์์ง์ด None์ด๋ฉด ์ธ์ ์ด๊ธฐํ ์ ํจ) | |
# if engine: | |
# SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) | |
# else: | |
# SessionLocal = None | |
# # Base ํด๋์ค ์์ฑ | |
# Base = declarative_base() | |
# # ๋ฐ์ดํฐ๋ฒ ์ด์ค ์ธ์ ์์กด์ฑ | |
# def get_db(): | |
# if not SessionLocal: | |
# print("Database session is not available.") | |
# raise RuntimeError("Database is not initialized.") | |
# db = SessionLocal() | |
# try: | |
# yield db | |
# finally: | |
# db.close() | |