88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
from logging.handlers import RotatingFileHandler
|
|
import os
|
|
from flask import Flask, jsonify, request
|
|
from app.config import config
|
|
from app.services.remote_service import DatabricksQuery
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_cors import CORS
|
|
from flask_jwt_extended import JWTManager
|
|
import logging
|
|
|
|
db = SQLAlchemy()
|
|
jwt = JWTManager()
|
|
|
|
|
|
# 配置日志
|
|
def setup_logging(app):
|
|
# 创建日志目录
|
|
if not os.path.exists('logs'):
|
|
os.makedirs('logs')
|
|
|
|
# 配置文件处理器 - 按大小轮转
|
|
file_handler = RotatingFileHandler(
|
|
'logs/app.log',
|
|
maxBytes=500*1024*1024, # 100MB
|
|
backupCount=10
|
|
)
|
|
file_handler.setLevel(logging.INFO)
|
|
|
|
# 配置控制台处理器
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(logging.DEBUG)
|
|
|
|
# 设置日志格式
|
|
formatter = logging.Formatter(
|
|
'%(asctime)s [%(levelname)s] %(funcName)s:%(lineno)d - %(message)s'
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
console_handler.setFormatter(formatter)
|
|
|
|
# 添加到app.logger
|
|
app.logger.addHandler(file_handler)
|
|
app.logger.addHandler(console_handler)
|
|
app.logger.setLevel(logging.DEBUG)
|
|
|
|
# 禁止传播到root logger（避免重复日志）
|
|
app.logger.propagate = False
|
|
|
|
# 记录启动信息
|
|
# app.logger.info('Flask应用启动成功！')
|
|
|
|
|
|
def create_app():
|
|
app = Flask(__name__)
|
|
app.config["JWT_SECRET_KEY"] = "your-secret-key" # 替换为你的密钥
|
|
app.config["JWT_TOKEN_LOCATION"] = ["headers"] # 从请求头获取令牌
|
|
app.config["JWT_HEADER_NAME"] = "Authorization" # 请求头字段名
|
|
# 注意：这里可以将 JWT_HEADER_TYPE 设为空，避免强制要求 Bearer 前缀
|
|
app.config["JWT_HEADER_TYPE"] = "" # 默认为 'Bearer'，设为空则不强制前缀
|
|
# 加载配置
|
|
app.config.from_object(config)
|
|
db.init_app(app)
|
|
jwt.init_app(app)
|
|
|
|
# 配置日志
|
|
setup_logging(app)
|
|
|
|
from app.blueprints.auth.routes import auth_bp
|
|
from app.blueprints.login.routes import login_bp
|
|
from app.blueprints.incident_gen5.routes import incident_gen5_bp
|
|
from app.blueprints.oauth.routes import oauth_bp
|
|
from app.blueprints.incident_gen6.routes import incident_gen6_bp
|
|
from app.blueprints.accident.routes import accident_bp
|
|
from app.blueprints.file_uploads.routes import file_uploads_bp
|
|
|
|
domain_prefix='accidentportal/api'
|
|
|
|
app.register_blueprint(auth_bp, url_prefix=f"/{domain_prefix}/api/auth")
|
|
app.register_blueprint(login_bp, url_prefix=f"/{domain_prefix}/user")
|
|
app.register_blueprint(incident_gen5_bp, url_prefix=f"/{domain_prefix}/gen5")
|
|
app.register_blueprint(incident_gen6_bp, url_prefix=f"/{domain_prefix}/gen6")
|
|
app.register_blueprint(oauth_bp, url_prefix="")
|
|
app.register_blueprint(accident_bp, url_prefix=f"/{domain_prefix}/accident")
|
|
app.register_blueprint(file_uploads_bp, url_prefix=f"/{domain_prefix}/file_uploads")
|
|
|
|
CORS(app, supports_credentials=True)
|
|
|
|
return app
|