mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
33 lines
944 B
Python
33 lines
944 B
Python
"""Handle JSON Web Token Encoding & Decoding."""
|
|
|
|
# Standard Library
|
|
import datetime
|
|
|
|
# Third Party
|
|
import jwt
|
|
|
|
# Project
|
|
from hyperglass.exceptions import RestError
|
|
|
|
|
|
async def jwt_decode(payload: str, secret: str) -> str:
|
|
"""Decode & validate an encoded JSON Web Token (JWT)."""
|
|
try:
|
|
decoded = jwt.decode(payload, secret, algorithm="HS256")
|
|
decoded = decoded["payload"]
|
|
return decoded
|
|
except (KeyError, jwt.PyJWTError) as exp:
|
|
raise RestError(str(exp)) from None
|
|
|
|
|
|
async def jwt_encode(payload: str, secret: str, duration: int) -> str:
|
|
"""Encode a query to a JSON Web Token (JWT)."""
|
|
token = {
|
|
"payload": payload,
|
|
"nbf": datetime.datetime.utcnow(),
|
|
"iat": datetime.datetime.utcnow(),
|
|
"exp": datetime.datetime.utcnow() + datetime.timedelta(seconds=duration),
|
|
}
|
|
encoded = jwt.encode(token, secret, algorithm="HS256").decode("utf-8")
|
|
return encoded
|