2026-07-21 16:11:23 +08:00
|
|
|
import os
|
2026-07-23 13:18:39 +08:00
|
|
|
|
2026-07-21 16:11:23 +08:00
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
from sqlalchemy import create_engine
|
|
|
|
|
from sqlalchemy.engine import URL
|
|
|
|
|
from sqlalchemy.orm import DeclarativeBase
|
|
|
|
|
|
|
|
|
|
load_dotenv()
|
|
|
|
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
database_url = URL.create(
|
|
|
|
|
drivername = "mysql+pymysql",
|
|
|
|
|
username = os.getenv("MYSQL_USER"),
|
|
|
|
|
password = os.getenv("MYSQL_PASSWORD"),
|
|
|
|
|
host = os.getenv("MYSQL_HOST"),
|
|
|
|
|
port = int(os.getenv("MYSQL_PORT")),
|
|
|
|
|
database = os.getenv("MYSQL_DATABASE"),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
engine = create_engine(database_url, pool_pre_ping=True)
|
