From 94d24a9530d62c1b17e880a46b4b6c4fea980722 Mon Sep 17 00:00:00 2001
From: Nan Zhou <nelzomal@gmail.com>
Date: Thu, 30 May 2024 15:35:05 +0800
Subject: [PATCH] feat: improve db config

---
 config/db_config.py | 15 ++++++++++++---
 requirements.txt    |  1 +
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/config/db_config.py b/config/db_config.py
index f7d9748..4a35ea2 100644
--- a/config/db_config.py
+++ b/config/db_config.py
@@ -1,12 +1,21 @@
 import os
+import dotenv
+
+dotenv.load_dotenv()
 
 # redis config
 REDIS_DB_HOST = "127.0.0.1"  # your redis host
 REDIS_DB_PWD = os.getenv("REDIS_DB_PWD", "123456")  # your redis password
 
 # mysql config
-RELATION_DB_PWD = os.getenv("RELATION_DB_PWD", "123456")  # your relation db password
-RELATION_DB_URL = f"mysql://root:{RELATION_DB_PWD}@localhost:3306/media_crawler"
+RELATION_DB_PWD = os.getenv("RELATION_DB_PWD", "123456")
+RELATION_DB_USER = os.getenv("RELATION_DB_USER", "root")
+RELATION_DB_HOST = os.getenv("RELATION_DB_HOST", "localhost")
+RELATION_DB_PORT = os.getenv("RELATION_DB_PORT", "3306")
+RELATION_DB_NAME = os.getenv("RELATION_DB_NAME", "media_crawler")
+
+
+RELATION_DB_URL = f"mysql://{RELATION_DB_USER}:{RELATION_DB_PWD}@{RELATION_DB_HOST}:{RELATION_DB_PORT}/{RELATION_DB_NAME}"
 
 # sqlite3 config
-# RELATION_DB_URL = f"sqlite://data/media_crawler.sqlite"
\ No newline at end of file
+# RELATION_DB_URL = f"sqlite://data/media_crawler.sqlite"
diff --git a/requirements.txt b/requirements.txt
index 83a5cda..8e7ec3f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,3 +10,4 @@ pydantic==2.5.2
 aiofiles~=23.2.1
 fastapi==0.110.2
 uvicorn==0.29.0
+python-dotenv==1.0.1