def prm_loader(claims): token_id = tokenstr.jti2id(claims['jti']) token_data = tables.Token.query.filter_by(id=token_id).first() if token_data: assert token_data.typ == 'prm' return TokenUser(claims, permissions=token_data.permissions, token_data=token_data)
def usr_loader(claims): token_id = tokenstr.jti2id(claims['jti']) token_data = tables.Token.query.filter_by(id=token_id).first() if token_data and not token_data.disabled: assert token_data.typ == 'usr' return TokenUser(claims, permissions=token_data.permissions, token_data=token_data, authenticated_email=token_data.user)
def test_jti2id(): eq_(tokenstr.jti2id('t12'), 12) assert_raises(TypeError, lambda: tokenstr.jti2id('xx')) assert_raises(TypeError, lambda: tokenstr.jti2id(None))