28 lines
855 B
Python
28 lines
855 B
Python
import os
|
|
|
|
config_file = 'config/config.py'
|
|
|
|
with open(config_file, 'r') as f:
|
|
content = f.read()
|
|
|
|
# 修改数据库配置部分，使其支持环境变量
|
|
modified_content = content.replace(
|
|
''' # 数据库配置
|
|
MYSQL_HOST = '27.124.22.104'
|
|
MYSQL_USER = 'taibai'
|
|
MYSQL_PASSWORD = 'taibaishopping'
|
|
MYSQL_DB = 'online_shopping'
|
|
MYSQL_PORT = 3306''',
|
|
''' # 数据库配置
|
|
MYSQL_HOST = os.environ.get('MYSQL_HOST', '27.124.22.104')
|
|
MYSQL_USER = os.environ.get('MYSQL_USER', 'taibai')
|
|
MYSQL_PASSWORD = os.environ.get('MYSQL_PASSWORD', 'taibaishopping')
|
|
MYSQL_DB = os.environ.get('MYSQL_DB', 'online_shopping')
|
|
MYSQL_PORT = int(os.environ.get('MYSQL_PORT', '3306'))'''
|
|
)
|
|
|
|
with open(config_file, 'w') as f:
|
|
f.write(modified_content)
|
|
|
|
print(f"✅ 成功更新 {config_file} 以支持环境变量")
|