feat: 新增flask中间件
This commit is contained in:
parent
59db9888f2
commit
f45dab6b07
8
main.py
8
main.py
|
|
@ -4,9 +4,11 @@ import os
|
|||
import fastapi_plugins
|
||||
from fastapi import FastAPI, Request
|
||||
from fastapi.exceptions import HTTPException, RequestValidationError
|
||||
from fastapi.middleware.wsgi import WSGIMiddleware
|
||||
from fastapi_sqlalchemy import DBSessionMiddleware
|
||||
|
||||
from config import init_config
|
||||
from src.middleware.flask import flask_app
|
||||
from src.utils.exception import (http_exception_handler,
|
||||
request_validation_error_handler)
|
||||
|
||||
|
|
@ -41,9 +43,15 @@ def create_app():
|
|||
app.add_exception_handler(HTTPException, http_exception_handler)
|
||||
app.add_exception_handler(RequestValidationError, request_validation_error_handler)
|
||||
|
||||
# 可以在这里挂载Flask的应用,复用之前项目的相关代码
|
||||
app.mount("/v1", WSGIMiddleware(flask_app))
|
||||
|
||||
# 在这里添加API route
|
||||
from src.api import example
|
||||
app.include_router(example.router, tags=["API示例"], prefix="/example")
|
||||
|
||||
|
||||
|
||||
return app
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,9 @@
|
|||
from flask import Flask, escape, request
|
||||
|
||||
flask_app = Flask(__name__)
|
||||
|
||||
|
||||
@flask_app.route("/")
|
||||
def index():
|
||||
name = request.args.get("name", "World")
|
||||
return f"Hello, {escape(name)} from Flask!"
|
||||
Loading…
Reference in New Issue