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} 以支持环境变量")