pzx-web-api/app/api/v1/auth.py

from apiflask import APIBlueprint

from app import rpc
from app.api.v1.schema.auth import WechatLoginIn, WechatLoginOut, RefreshTokenIn, RefreshTokenOut

auth = APIBlueprint('auth', __name__)


@auth.post('/wechat_login')
@auth.input(WechatLoginIn)
@auth.output(WechatLoginOut)
def wechat_login(json_data):
    try:
        result = rpc.admin.wechat_login(json_data['code'])
    except Exception as e:
        # TODO 返回 apiflask 的 json 格式的 error
        raise e
    return result


@auth.get('/refresh_token')
@auth.input(RefreshTokenIn, location='query')
@auth.output(RefreshTokenOut)
def refresh_token(query_data):
    try:
        result = rpc.admin.refresh_token(query_data['refresh_token'])
    except Exception as e:
        # TODO 返回 apiflask 的 json 格式的 error
        raise e
    return result