| # 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:root@10.10.10.180: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() | |