flask 带时效令牌验证(flask 92)

用户5760343

发布于 2019-08-20 18:06:03

5220

发布于 2019-08-20 18:06:03

文章被收录于专栏：sktj

from itsdangerous import BadSignature, SignatureExpired from itsdangerous import TimedJSONWebSignatureSerializer as Serializer

def generate_token(user, operation, expire_in=None, **kwargs): s = Serializer(current_app.config['SECRET_KEY'], expire_in)

data = {'id': user.id, 'operation': operation}
data.update(**kwargs)
return s.dumps(data)

def validate_token(user, token, operation, new_password=None): s = Serializer(current_app.config['SECRET_KEY'])

try:
    data = s.loads(token)
except (SignatureExpired, BadSignature):
    return False

if operation != data.get('operation') or user.id != data.get('id'):
    return False

if operation == Operations.CONFIRM:
    user.confirmed = True
elif operation == Operations.RESET_PASSWORD:
    user.set_password(new_password)
elif operation == Operations.CHANGE_EMAIL:
    new_email = data.get('new_email')
    if new_email is None:
        return False
    if User.query.filter_by(email=new_email).first() is not None:
        return False
    user.email = new_email
else:
    return False

db.session.commit()
return True

token = generate_token(user=user, operation='confirm') send_confirm_email(user=user, token=token)

if validate_token(user=current_user, token=token, operation=Operations.CONFIRM):
    flash('Account confirmed.', 'success')
    return redirect(url_for('main.index'))

默认过期事件1个小时

Serializer

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019.08.13 ，如有侵权请联系 cloudcommunity@tencent.com 删除

云推荐引擎

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

云推荐引擎

登录后参与评论

0 条评论

热度