79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
import os
|
|
import secrets
|
|
from dotenv import load_dotenv
|
|
|
|
# 加载环境变量
|
|
load_dotenv()
|
|
|
|
class BaseConfig:
|
|
"""基础配置类"""
|
|
# 应用设置
|
|
SECRET_KEY = os.environ.get('SECRET_KEY') or secrets.token_hex(16)
|
|
DEBUG = False
|
|
|
|
# 数据库设置
|
|
SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///email_system.db'
|
|
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
|
|
|
# SMTP服务器设置
|
|
SMTP_HOST = '0.0.0.0'
|
|
SMTP_PORT = 25
|
|
|
|
# 存储设置
|
|
MAIL_STORAGE_PATH = os.path.join(os.getcwd(), 'email_data')
|
|
|
|
# 日志设置
|
|
LOG_LEVEL = 'INFO'
|
|
LOG_FILE = os.path.join(os.getcwd(), 'logs', 'email_system.log')
|
|
|
|
# API设置
|
|
API_RATE_LIMIT = '200/hour'
|
|
API_TOKEN_EXPIRE = 3600 * 24 * 7 # 7天
|
|
|
|
# 邮件域名配置
|
|
MAIL_DOMAINS = os.environ.get('MAIL_DOMAINS', 'example.com,mail.example.com').split(',')
|
|
|
|
|
|
class DevelopmentConfig(BaseConfig):
|
|
"""开发环境配置"""
|
|
DEBUG = True
|
|
LOG_LEVEL = 'DEBUG'
|
|
|
|
# 使用内存数据库方便开发
|
|
# SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:'
|
|
|
|
|
|
class TestingConfig(BaseConfig):
|
|
"""测试环境配置"""
|
|
TESTING = True
|
|
DEBUG = True
|
|
|
|
# 使用内存数据库进行测试
|
|
SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:'
|
|
|
|
# 使用随机端口进行测试
|
|
SMTP_PORT = 2525
|
|
|
|
|
|
class ProductionConfig(BaseConfig):
|
|
"""生产环境配置"""
|
|
DEBUG = False
|
|
LOG_LEVEL = 'ERROR'
|
|
|
|
# SMTP服务器设置
|
|
SMTP_PORT = 25
|
|
|
|
# API设置
|
|
API_RATE_LIMIT = '100/hour'
|
|
|
|
|
|
# 配置映射
|
|
config_by_name = {
|
|
'development': DevelopmentConfig,
|
|
'testing': TestingConfig,
|
|
'production': ProductionConfig,
|
|
'default': DevelopmentConfig
|
|
}
|
|
|
|
# 根据环境变量选择配置
|
|
active_config = config_by_name.get(os.environ.get('FLASK_ENV', 'default'), DevelopmentConfig) |