import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    # 数据库配置
    DB_HOST = os.getenv('DB_HOST', 'localhost')
    DB_PORT = int(os.getenv('DB_PORT', 3306))
    DB_USER = os.getenv('DB_USERNAME', 'root')
    DB_PASSWORD = os.getenv('DB_PASSWORD', 'dfwy2025')
    DB_NAME = os.getenv('DB_DATABASE', 'drug_data')
    DB_TABLE = 'mt_drug_middle'
    DB_SHOP_TABLE = 'mt_shop_info_middle'
    DB_PDD_TABLE = 'pdd_drug_middle'
    DB_PDD_SHOP_TABLE = 'pdd_shop_info_middle'
    DB_RH_TABLE = 'rh_mt_drug'
    DB_RH_SHOP_TABLE = 'rh_mt_shop_info'

    DB_AUTO_DRUG_TABLE = 'mt_auto_drug_test'


    LOCAL_DB_HOST = os.getenv('LOCAL_DB_HOST', 'localhost')
    LOCAL_DB_PORT = int(os.getenv('LOCAL_DB_PORT', 3306))
    LOCAL_DB_USER = os.getenv('LOCAL_DB_USERNAME', 'root')
    LOCAL_DB_PASSWORD = os.getenv('LOCAL_DB_PASSWORD', 'dfwy2025')
    LOCAL_DB_NAME = os.getenv('LOCAL_DB_DATABASE', 'drug_data')
    
    # 应用配置
    PACKAGE_NAME = 'com.sankuai.meituan'    
    # 连接池配置
    DB_POOL_SIZE = int(os.getenv('DB_POOL_SIZE', 5))
    DB_POOL_TIMEOUT = int(os.getenv('DB_POOL_TIMEOUT', 30))