modify scripts

2025-08-10 19:19:00 +08:00
parent d9ba319269
commit fa70e3f0ae
1 changed files with 5 additions and 0 deletions
--- 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"