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 = 'retrieve_scrape_tmp_data'
    DB_PDD_SHOP_TABLE = 'pdd_shop_info_middle'

    DB_PDD_AUTO_TABLE = 'pdd_auto_drug_test'
    
    # 应用配置
    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))

    access_key_id = os.getenv('OSS_ACCESS_KEY_ID')
    access_key_secret = os.getenv('OSS_ACCESS_KEY_SECRET')
    endpoint = os.getenv('OSS_ENDPOINT')
    bucket_name = os.getenv('OSS_BUCKET_NAME')
    oss_prefix = os.getenv('OSS_PREFIX')

    # 本地截图配置
    LOCAL_SCREENSHOT_DIR = "local_screenshots"  # 本地截图保存目录
    LOCAL_SCREENSHOT_NAME = None  # 自动生成文件名，无需手动指定
    LOCAL_CROPPED_DIR = "./local_cropped_screenshots"  # 裁剪后图片保存目录


    # 图片压缩配置
    IMAGE_COMPRESS_ENABLE = True  # 是否开启图片压缩（True=开启，False=关闭）
    IMAGE_COMPRESS_QUALITY = 30  # jpg/jpeg格式压缩质量（1-95，数值越大画质越好，文件越大，推荐80-90）
    IMAGE_COMPRESS_PNG_LEVEL = 9  # png格式压缩级别（0-9，数值越大压缩率越高，速度越慢，推荐5-7）


    #百度OCR配置
    APP_KEY = os.getenv('APP_KEY')
    APP_SECRET = os.getenv('APP_SECRET')
    token_url = os.getenv('token_url')



    API_KEY = os.getenv('API_KEY')
    SECRET_KEY = os.getenv('SECRET_KEY')