core/auth/token.py

from datetime import datetime

import jwt

from settings import JWT_ALGORITHM, JWT_SECRET_KEY
from auth.validations import PayLoad, User


class Token:
    @staticmethod
    def encode(user: User, exp: datetime, device: str = "pc") -> str:
        payload = {"user_id": user.id, "device": device, "exp": exp, "iat": datetime.utcnow()}
        return jwt.encode(payload, JWT_SECRET_KEY, JWT_ALGORITHM)

    @staticmethod
    def decode(token: str, verify_exp: bool = True) -> PayLoad:
        payload = jwt.decode(
            token,
            key=JWT_SECRET_KEY,
            options={"verify_exp": verify_exp},
            algorithms=[JWT_ALGORITHM],
        )
        return PayLoad(**payload)