83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
|
"""
|
|||
|
|
基础配置类
|
|||
|
|
包含所有环境通用的配置项
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
from datetime import timedelta
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
"""
|
|||
|
|
基础配置类
|
|||
|
|
所有环境配置都继承此类
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# Flask基础配置
|
|||
|
|
SECRET_KEY = os.environ.get('SECRET_KEY')
|
|||
|
|
if not SECRET_KEY:
|
|||
|
|
raise ValueError("SECRET_KEY 环境变量未设置，请在.env文件中配置")
|
|||
|
|
|
|||
|
|
# 数据库配置
|
|||
|
|
SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///app.db'
|
|||
|
|
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
|||
|
|
SQLALCHEMY_ENGINE_OPTIONS = {
|
|||
|
|
'pool_pre_ping': True, # 连接前检查连接是否有效
|
|||
|
|
'pool_recycle': 300, # 连接回收时间（秒）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 跨域配置
|
|||
|
|
CORS_ORIGINS = os.environ.get('CORS_ORIGINS', '*').split(',')
|
|||
|
|
|
|||
|
|
# 日志配置
|
|||
|
|
LOG_LEVEL = os.environ.get('LOG_LEVEL', 'INFO')
|
|||
|
|
LOG_FILE = os.environ.get('LOG_FILE', 'logs/app.log')
|
|||
|
|
|
|||
|
|
# 缓存配置
|
|||
|
|
CACHE_TYPE = os.environ.get('CACHE_TYPE', 'simple')
|
|||
|
|
CACHE_DEFAULT_TIMEOUT = int(os.environ.get('CACHE_DEFAULT_TIMEOUT', 300))
|
|||
|
|
|
|||
|
|
# Redis缓存配置（当CACHE_TYPE=redis时使用）
|
|||
|
|
REDIS_URL = os.environ.get('REDIS_URL', 'redis://localhost:6379/0')
|
|||
|
|
|
|||
|
|
# 会话配置
|
|||
|
|
PERMANENT_SESSION_LIFETIME = timedelta(
|
|||
|
|
hours=int(os.environ.get('SESSION_LIFETIME_HOURS', 24))
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 文件上传配置
|
|||
|
|
MAX_CONTENT_LENGTH = int(os.environ.get('MAX_CONTENT_LENGTH', 16 * 1024 * 1024)) # 16MB
|
|||
|
|
UPLOAD_FOLDER = os.environ.get('UPLOAD_FOLDER', 'uploads')
|
|||
|
|
|
|||
|
|
# 安全配置
|
|||
|
|
WTF_CSRF_ENABLED = os.environ.get('WTF_CSRF_ENABLED', 'True').lower() == 'true'
|
|||
|
|
WTF_CSRF_TIME_LIMIT = int(os.environ.get('WTF_CSRF_TIME_LIMIT', 3600))
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def init_app(app):
|
|||
|
|
"""
|
|||
|
|
初始化应用配置
|
|||
|
|
创建必要的目录和配置日志
|
|||
|
|
"""
|
|||
|
|
# 创建必要的目录
|
|||
|
|
os.makedirs('logs', exist_ok=True)
|
|||
|
|
os.makedirs('uploads', exist_ok=True)
|
|||
|
|
|
|||
|
|
# 配置日志
|
|||
|
|
import logging
|
|||
|
|
from logging.handlers import RotatingFileHandler
|
|||
|
|
|
|||
|
|
if not app.debug and not app.testing:
|
|||
|
|
file_handler = RotatingFileHandler(
|
|||
|
|
Config.LOG_FILE,
|
|||
|
|
maxBytes=10240000, # 10MB
|
|||
|
|
backupCount=10
|
|||
|
|
)
|
|||
|
|
file_handler.setFormatter(logging.Formatter(
|
|||
|
|
'%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'
|
|||
|
|
))
|
|||
|
|
file_handler.setLevel(getattr(logging, Config.LOG_LEVEL))
|
|||
|
|
app.logger.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
app.logger.setLevel(getattr(logging, Config.LOG_LEVEL))
|
|||
|
|
app.logger.info('应用启动')
|
|||
|
|
|