[python flask token 校验 装饰器]#flask

login_rerquired.py

from flask import request
from app.libs.token_auth import verify_auth_token


def login_required(fun):

    def wrapper(*args, **kwargs):
        headers = request.headers
        auth = headers.get("Authorization")
        verify_result = verify_auth_token(auth.split(" ")[1])
        return fun(*args, **kwargs)
    return wrapper
    

# 使用
from flask_restful import Resource
from app.models.server import Server
from app.libs.error_code import Success
from app.libs.login_required import login_required


class HandleGetIp(Resource):

    @login_required
    def get(self):
        servers = Server.query.order_by("id").all()
        return {"code": "hello", "data": [Server.serialize(record) for record in servers]}