pyjwt引发TypeError

时间:2019-09-27 13:20:11

标签: python-3.x unit-testing pyjwt

我正在按以下方式使用pyjwt:

def encode_auth_token(self, user_id):
    '''Generates the auth token.'''

    try:
        payload = {
            'exp': datetime.utcnow() + datetime.timedelta(
                days = current_app.config.get('TOKEN_EXPIRATION_DAYS'),
                seconds = current_app.config.get('TOKEN_EXPIRATION_SECONDS')
            ),
            'iat': datetime.datetime.utcnow(),
            'sub': user_id
        }
        return jwt.encode(
            payload,
            current_app.config.get('SECRET_KEY'),
            algorithm='HS256'
        )
    except Exception as e:
        return e

此问题是根据docs instance.encode()应该返回bytes,而根据另一个资源应该返回str。当我通过unitttests运行它时:

auth_token = user.encode_auth_token(user.id)
self.assertTrue(isinstance(auth_token, str))

我得到:AssertionError: False is not true,当我将str替换为bytes时,我得到了相同的错误。那么该方法应该返回什么类型?

1 个答案:

答案 0 :(得分:0)

它很好地返回字节数据。如果可以确认确实如此,则可以通过在令牌实例本身上调用解码方法来强制其返回字符串。

token = jwt.encode(payload,secret).decode('utf-8')
return token