Merge pull request #4 from soneji/patch-1

Fix to add support for PyJWT>=2.0.0a1 and keep support for PyJWT<=1.7.1
Welltory · Feb 25, 2021 · 23c431d · 23c431d
2 parents 93119b2 + 8723801
commit 23c431d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/jwt_auth.py b/src/jwt_auth.py
@@ -34,4 +34,8 @@ def make_jwt_payload(api_key: str, token_exp_delta: int) -> dict:
 
 def generate_jwt_token(payload: dict, api_secret: str) -> str:
     encoded = jwt.encode(payload, api_secret, algorithm='HS256')
-    return encoded.decode()
+    if isinstance(encoded, bytes):
+        # For PyJWT <= 1.7.1
+        return encoded.decode()
+    # For PyJWT >= 2.0.0a1
+    return encoded