Python Flask swagger自动生成文档【插件存在版本兼容问题,部署会出问题!不推荐】

首先安装依赖:

pip install flask_siwadoc pydantic

封装swagger.py文件,代码如下:

from flask_siwadoc import SiwaDoc
siwa = SiwaDoc()

然后在主应用中(项目入口文件)加入以下代码:

from flask import Flask
from swagger import siwa #导入它
app = Flask(__name__)
siwa.init_app(app)

在user.py中加入以下代码:

from flask import Blueprint,request, session
from swagger import siwa #导入它
from pydantic import BaseModel, Field
user_bp = Blueprint('user', __name__, url_prefix='/user')
class loginModel(BaseModel):
    username:str=Field(...)
    password:str=Field(...)
@user_bp.route('/login', methods=['POST'])
@siwa.doc(body=loginModel,tags=["user"])
def user_login(body:loginModel):
    reqJSONData = request.get_json(silent=True)
    print(reqJSONData)
    if not reqJSONData: return r(code=401, msg='登录失败, 请求参数为空')
    username = reqJSONData.get('username')
    password = reqJSONData.get('password')

    if not all([username, password]):
        return r(code=401, msg='登录, 缺少请求参数')

    user = supabase.table('sys_user').select('id,username,name').eq('username',username).eq('password',password).execute().data
    # 4. 用户不存在, 直接返回
    if not user:
        return r(code=404, msg='用户名或密码错误')
    else:
        access_token = create_access_token(identity=user[0])
        session['user_info'] = user
        return r(msg='登录成功', data={"token":access_token})

你可能感兴趣的:(python,flask,python,flask,pip)