From fa70e3f0aeb48eff20441d3487c7a1c661114024 Mon Sep 17 00:00:00 2001
From: sophon <oscar@easyprompt8.com>
Date: Sun, 10 Aug 2025 19:19:00 +0800
Subject: [PATCH] modify scripts

---
 src/sqlalchemy/config.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/sqlalchemy/config.py b/src/sqlalchemy/config.py
index ad31e00..b776662 100644
--- a/src/sqlalchemy/config.py
+++ b/src/sqlalchemy/config.py
@@ -9,4 +9,9 @@ db_config = {
     'database': 'stockdb'
 }
 
+# 读取环境变量，来修改 db_config 中host的值 
+if 'DB_HOST' in os.environ:
+    db_config['host'] = os.environ['DB_HOST']
+
+
 global_db_url = f"mysql+pymysql://{db_config['user']}:{db_config['password']}@{db_config['host']}:3306/{db_config['database']}?charset=utf8mb4"
\ No newline at end of file