config.py

import os
from datetime import timedelta
from dotenv import load_dotenv

load_dotenv()

class Config:
    # 微信支付配置（可选，未填写则启用模拟支付网关）
    WXPAY_MCH_ID = os.getenv('WXPAY_MCH_ID')
    WXPAY_PRIVATE_KEY = os.getenv('WXPAY_PRIVATE_KEY')
    WXPAY_CERT_SERIAL_NO = os.getenv('WXPAY_CERT_SERIAL_NO')
    WXPAY_APP_ID = os.getenv('WXPAY_APP_ID')
    WXPAY_APIV3_KEY = os.getenv('WXPAY_APIV3_KEY')
    WXPAY_NOTIFY_URL = os.getenv('WXPAY_NOTIFY_URL')
    # 基础配置
    SECRET_KEY = os.getenv('SECRET_KEY', 'dev-secret-key')
    
    # 数据库配置
    DB_HOST = os.getenv('DB_HOST', 'localhost')
    DB_PORT = os.getenv('DB_PORT', '3306')
    DB_USER = os.getenv('DB_USER', 'root')
    DB_PASSWORD = os.getenv('DB_PASSWORD', 'password')
    DB_NAME = os.getenv('DB_NAME', 'happy_community')
    
    SQLALCHEMY_DATABASE_URI = f'mysql+pymysql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}'
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    
    # JWT配置
    JWT_SECRET_KEY = os.getenv('JWT_SECRET_KEY', 'jwt-secret-key')
    JWT_ACCESS_TOKEN_EXPIRES = timedelta(hours=24)
    JWT_TOKEN_LOCATION = ['headers']
    
    # 文件上传配置
    UPLOAD_FOLDER = os.getenv('UPLOAD_FOLDER', 'uploads')
    MAX_CONTENT_LENGTH = 16 * 1024 * 1024  # 16MB
    
    # 子目录配置
    AVATAR_UPLOAD_FOLDER = os.path.join(UPLOAD_FOLDER, 'avatars')
    REPAIR_UPLOAD_FOLDER = os.path.join(UPLOAD_FOLDER, 'repairs')
    COMMUNITY_UPLOAD_FOLDER = os.path.join(UPLOAD_FOLDER, 'community')
    ANNOUNCEMENT_UPLOAD_FOLDER = os.path.join(UPLOAD_FOLDER, 'announcements')
    
    # Redis配置（用于缓存和WebSocket）
    REDIS_URL = os.getenv('REDIS_URL', 'redis://localhost:6379/0')

    # 支付宝配置
    ALIPAY_APP_ID = os.getenv('ALIPAY_APP_ID')
    APP_PRIVATE_KEY = os.getenv('APP_PRIVATE_KEY')
    ALIPAY_PUBLIC_KEY = os.getenv('ALIPAY_PUBLIC_KEY')
    SIGN_TYPE = os.getenv('SIGN_TYPE', 'RSA2')

    # Celery配置（用于异步任务）
    CELERY_BROKER_URL = REDIS_URL
    CELERY_RESULT_BACKEND = REDIS_URL

    # 服务监听端口（从环境变量读取，默认 5000）
    PORT = int(os.getenv('PORT', '5000'))

    # 支付模拟开关：如果为 True，则使用模拟支付网关而不初始化第三方 SDK
    MOCK_PAYMENTS = os.getenv('MOCK_PAYMENTS', 'false').lower() in ('1', 'true', 'yes')

class DevelopmentConfig(Config):
    DEBUG = True

class ProductionConfig(Config):
    DEBUG = False

config = {
    'development': DevelopmentConfig,
    'production': ProductionConfig,
    'default': DevelopmentConfig
}