94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
import os
|
|
from flask import Flask
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_jwt_extended import JWTManager
|
|
from flask_pymongo import PyMongo
|
|
from .config import Config,config_map
|
|
from flask_cors import CORS
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
|
|
db = SQLAlchemy()
|
|
mongo = PyMongo()
|
|
jwt = JWTManager()
|
|
|
|
# def create_app():
|
|
# app = Flask(__name__)
|
|
# app.config.from_object(Config)
|
|
|
|
# db.init_app(app)
|
|
# mongo.init_app(app)
|
|
# # jwt.init_app(app)
|
|
|
|
# # 注册蓝图
|
|
# from .blueprints.auth import auth_bp
|
|
# from .blueprints.bag_data.routes import data_bp
|
|
|
|
|
|
# app.register_blueprint(auth_bp, url_prefix='/api/auth')
|
|
# app.register_blueprint(data_bp, url_prefix='/api/data')
|
|
|
|
# return app
|
|
|
|
# 加载配置
|
|
def create_app():
|
|
app = Flask(__name__)
|
|
# app.config.from_object(Config)
|
|
config_name = os.environ.get('FLASK_ENV', 'default')
|
|
app.config.from_object(config_map[config_name])
|
|
|
|
db.init_app(app)
|
|
mongo.init_app(app)
|
|
jwt.init_app(app)
|
|
setup_logging(app) # 配置日志
|
|
# 注册蓝图
|
|
from .blueprints.auth.routes import auth_bp
|
|
from .blueprints.bag_data.routes import data_bp
|
|
from app.blueprints.data_display.routes import check_data
|
|
from app.blueprints.data_factory.routes import data_factory_bp
|
|
from app.blueprints.label_data.rotutes import label_bp
|
|
from app.blueprints.remote_data.rotutes import remote_bp
|
|
from app.blueprints.vlm.rotutes import vlm_bp
|
|
|
|
app.register_blueprint(auth_bp, url_prefix='/api/auth')
|
|
app.register_blueprint(data_bp, url_prefix='/api/data')
|
|
app.register_blueprint(check_data, url_prefix='/api/inspect')
|
|
app.register_blueprint(data_factory_bp, url_prefix='/api/factory')
|
|
app.register_blueprint(label_bp, url_prefix='/api/label')
|
|
app.register_blueprint(remote_bp, url_prefix='/api/remote')
|
|
app.register_blueprint(vlm_bp, url_prefix='/api/vlm')
|
|
|
|
CORS(app, supports_credentials=True)
|
|
|
|
return app
|
|
|
|
# 配置日志
|
|
def setup_logging(app):
|
|
# 确保日志目录存在
|
|
os.makedirs(app.config['LOG_DIR'], exist_ok=True)
|
|
|
|
# 创建文件处理器
|
|
file_handler = RotatingFileHandler(
|
|
os.path.join(app.config['LOG_DIR'], 'app.log'),
|
|
maxBytes=app.config['LOG_FILE_MAX_BYTES'],
|
|
backupCount=app.config['LOG_FILE_BACKUP_COUNT']
|
|
)
|
|
|
|
# 设置日志级别
|
|
log_level = getattr(logging, app.config['LOG_LEVEL'])
|
|
file_handler.setLevel(log_level)
|
|
app.logger.setLevel(log_level)
|
|
|
|
# 设置日志格式
|
|
formatter = logging.Formatter(
|
|
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
file_handler.setFormatter(formatter)
|
|
|
|
# 添加处理器
|
|
if not app.logger.handlers:
|
|
app.logger.addHandler(file_handler)
|
|
|
|
# 记录启动信息
|
|
app.logger.info(f'Application started with config') |