Fix PyJWT (import jwt) code to work with latest version

aws · Jul 5, 2023 · 6f6afb5 · 6f6afb5
1 parent 579550a
commit 6f6afb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/samples/todo-app/code/chalicelib/auth.py b/docs/source/samples/todo-app/code/chalicelib/auth.py
@@ -12,7 +12,7 @@ def get_jwt_token(username, password, record, secret):
         record['hash'],
         password.encode('utf-8'),
         record['salt'].value,
-        record['rounds']
+        int(record['rounds'])
     )
     expected = record['hashed'].value
     if hmac.compare_digest(actual, expected):
@@ -25,7 +25,7 @@ def get_jwt_token(username, password, record, secret):
             'jti': unique_id,
             # NOTE: We can also add 'exp' if we want tokens to expire.
         }
-        return jwt.encode(payload, secret, algorithm='HS256').decode('utf-8')
+        return jwt.encode(payload, secret, algorithm='HS256')
     raise UnauthorizedError('Invalid password')