Flask扩展实现HTTP令牌token

2020-01-16 15:20:57 浏览数 (1)

Token认证

在restful设计中,用户认证模式通常使用json web token,而不会使用传统的HTTP Basic认证(传入账号密码) token认证模式如下:在请求header中加入token

Flask中的实现

flask扩展flask-httpauth提供了该认证方法

代码语言:javascript复制
$ pip install flask-httpauth

首先实例化

代码语言:javascript复制
auth = HTTPTokenAuth(scheme='JWT')

上面的scheme就是token前面的字符串,可以自定义,这里我使用JWT 源代码如下:

它提供了一个装饰器,里面可以实现自己的token检验方法

代码语言:javascript复制
@auth.verify_token
def verify_token(token):
    pass
#这里需要自定义

Flask中的itsdangerous提供了令牌生成和校验的方法

代码语言:javascript复制
from itsdangerous import TimedJSONWebSignatureSerializer, BadSignature,
    SignatureExpired

组合起来:

代码语言:javascript复制
@auth.verify_token
def verify_token(token):
    s = TimedJSONWebSignatureSerializer(
        current_app.config['SECRET_KEY']
    )
    try:
        data = s.loads(token)
        print(data)
    except BadSignature:
        raise AuthFailed(msg='token不正确')
    except SignatureExpired:
        raise AuthFailed(msg='token过期')
    uid = data['uid']
    return True

通过验证后返回True代表token认证成功! 如果有需要登录的视图,就可以使用如下装饰器:

代码语言:javascript复制
@api.route('/login_require/', methods=['POST'])
@auth.login_required
def func():
    pass

只有Token认证成功才能进入该视图

0 人点赞