guzhi/app/core/token_blacklist.py

14 lines
426 B
Python

from datetime import datetime
from typing import Optional
from app.models.token_blacklist import TokenBlacklist
async def add_to_blacklist(token: str, user_id: int, exp: Optional[datetime] = None, jti: Optional[str] = None) -> None:
await TokenBlacklist.create(token=token, user_id=user_id, exp=exp, jti=jti)
async def is_blacklisted(token: str) -> bool:
return await TokenBlacklist.filter(token=token).exists()