|
|
import os
|
|
|
from apiflask import APIFlask
|
|
|
from .extensions import init_exts
|
|
|
from .routes import init_routes
|
|
|
from ..config import get_config
|
|
|
from .events import init_event_handlers
|
|
|
|
|
|
|
|
|
def create_app(config_name=None):
|
|
|
"""
|
|
|
应用工厂函数
|
|
|
|
|
|
Args:
|
|
|
config_name: 配置名称 ('dev', 'test', 'prod')
|
|
|
如果为 None，则从环境变量 FLASK_ENV 读取
|
|
|
|
|
|
Returns:
|
|
|
Flask 应用实例
|
|
|
|
|
|
docs_ui: The UI of API documentation, one of `swagger-ui` (default), `redoc`,
|
|
|
`elements`, `rapidoc`, and `rapipdf`.
|
|
|
"""
|
|
|
app = APIFlask(
|
|
|
__name__.split(".")[0],
|
|
|
title="iTi-Flask",
|
|
|
version="1.0.0",
|
|
|
json_errors=True,
|
|
|
docs_ui="elements",
|
|
|
)
|
|
|
|
|
|
# 加载配置
|
|
|
config_obj = get_config(config_name)
|
|
|
app.config.from_object(config_obj)
|
|
|
|
|
|
# 确保必要的目录存在
|
|
|
_ensure_directories(app)
|
|
|
|
|
|
# 初始化扩展
|
|
|
init_exts(app)
|
|
|
|
|
|
# 初始化事件处理器
|
|
|
init_event_handlers(app)
|
|
|
|
|
|
# 初始化路由
|
|
|
init_routes(app)
|
|
|
|
|
|
# 打印当前环境信息
|
|
|
env = config_name or os.getenv("FLASK_ENV", "dev")
|
|
|
print(f"🚀 应用启动 - 环境: {env}")
|
|
|
print(f"📊 数据库: {app.config.get('SQLALCHEMY_DATABASE_URI')}")
|
|
|
|
|
|
return app
|
|
|
|
|
|
|
|
|
def _ensure_directories(app):
|
|
|
"""确保必要的目录存在"""
|
|
|
# 数据库目录（SQLite）
|
|
|
db_uri = app.config.get("SQLALCHEMY_DATABASE_URI", "")
|
|
|
if "sqlite:///" in db_uri and not db_uri.endswith(":memory:"):
|
|
|
db_path = db_uri.replace("sqlite:///", "")
|
|
|
db_dir = os.path.dirname(db_path)
|
|
|
if db_dir and not os.path.exists(db_dir):
|
|
|
os.makedirs(db_dir, exist_ok=True)
|
|
|
|
|
|
file_storage_config = app.config.get("FILE_STORAGE", {})
|
|
|
local_config = file_storage_config.get("LOCAL", {})
|
|
|
local_path = local_config.get("base_path")
|
|
|
if local_path:
|
|
|
os.makedirs(local_path, exist_ok=True) |