From 23b9db19a8c1d4814316604b6701882032a85c9a Mon Sep 17 00:00:00 2001
From: hzhang <hzhang@hangman-lab.top>
Date: Tue, 3 Dec 2024 14:35:25 +0000
Subject: [PATCH] fix docker config

---
 db/__init__.py  | 5 +++--
 env_provider.py | 7 ++++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/db/__init__.py b/db/__init__.py
index 232406c..b888265 100644
--- a/db/__init__.py
+++ b/db/__init__.py
@@ -2,13 +2,14 @@
 from contextlib import contextmanager
 from db.models import Base
 
-
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy import create_engine
 import env_provider
+from env_provider import DB_HOST, DB_PORT, DB_NAME, DB_USER, DB_PASSWORD
 
-engine = create_engine(env_provider.DB_CONNECT_STRING)
+engine = create_engine(f"mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}")
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
 @contextmanager
 def get_db():
     db = SessionLocal()
diff --git a/env_provider.py b/env_provider.py
index 1487a64..85e1f41 100644
--- a/env_provider.py
+++ b/env_provider.py
@@ -6,5 +6,10 @@ load_dotenv()
 
 CLIENT_ID = os.getenv("CLIENT_ID")
 CLIENT_SECRET = os.getenv("CLIENT_SECRET")
-DB_CONNECT_STRING = os.getenv("DB_CONNECT_STRING")
+
+DB_HOST = os.getenv("DB_HOST")
+DB_PORT = os.getenv("DB_PORT")
+DB_NAME = os.getenv("DB_NAME")
+DB_USER = os.getenv("DB_USER")
+DB_PASSWORD = os.getenv("DB_PASSWORD")
 SESSION_SECRET_KEY = os.getenv("SESSION_SECRET_KEY")
\ No newline at end of file