78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from flask import Flask
|
||
from flask_sqlalchemy import SQLAlchemy
|
||
from flask_migrate import Migrate
|
||
from flask_login import LoginManager
|
||
from flask_session import Session
|
||
from flask_wtf.csrf import CSRFProtect
|
||
from config import config
|
||
from flask_bootstrap import Bootstrap
|
||
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
import os
|
||
from datetime import datetime, timedelta
|
||
import glob
|
||
|
||
# 初始化扩展
|
||
db = SQLAlchemy()
|
||
migrate = Migrate()
|
||
login_manager = LoginManager()
|
||
sess = Session()
|
||
csrf = CSRFProtect()
|
||
bootstrap = Bootstrap()
|
||
login_manager.login_view = 'auth.login'
|
||
|
||
def create_app(config_name='default'):
|
||
"""应用工厂函数，创建并配置Flask应用实例"""
|
||
app = Flask(__name__)
|
||
|
||
# 加载配置
|
||
app.config.from_object(config[config_name])
|
||
config[config_name].init_app(app)
|
||
# 配置代理转发
|
||
from werkzeug.middleware.proxy_fix import ProxyFix
|
||
app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1, x_proto=1, x_host=1)
|
||
|
||
# 初始化扩展
|
||
db.init_app(app)
|
||
migrate.init_app(app, db)
|
||
login_manager.init_app(app)
|
||
sess.init_app(app)
|
||
csrf.init_app(app)
|
||
bootstrap.init_app(app)
|
||
|
||
# 配置日志
|
||
if not os.path.exists('logs'):
|
||
os.mkdir('logs')
|
||
|
||
# 清理过期日志
|
||
log_retention_days = int(os.environ.get('LOG_RETENTION_DAYS', 7))
|
||
cutoff_time = datetime.now() - timedelta(days=log_retention_days)
|
||
for log_file in glob.glob('logs/*.log*'):
|
||
if os.path.getmtime(log_file) < cutoff_time.timestamp():
|
||
os.remove(log_file)
|
||
|
||
file_handler = RotatingFileHandler('logs/pt_blacklist.log', maxBytes=10240000, backupCount=10)
|
||
file_handler.setFormatter(logging.Formatter(
|
||
'%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'
|
||
))
|
||
file_handler.setLevel(logging.INFO)
|
||
app.logger.addHandler(file_handler)
|
||
app.logger.setLevel(logging.INFO)
|
||
app.logger.info('PT黑名单系统启动')
|
||
|
||
# 注册自定义过滤器
|
||
from .filters import translate_status, translate_reason, translate_reasons_list, to_beijing_time
|
||
app.jinja_env.filters['translate_status'] = translate_status
|
||
app.jinja_env.filters['translate_reason'] = translate_reason
|
||
app.jinja_env.filters['translate_reasons_list'] = translate_reasons_list
|
||
app.jinja_env.filters['to_beijing_time'] = to_beijing_time
|
||
|
||
# 注册蓝图
|
||
from .routes import main as main_blueprint
|
||
app.register_blueprint(main_blueprint)
|
||
|
||
from .auth import auth as auth_blueprint
|
||
app.register_blueprint(auth_blueprint, url_prefix='/auth')
|
||
|
||
return app
|