17 lines
992 B
Python
17 lines
992 B
Python
from datetime import datetime
|
|
from sqlalchemy import String, DateTime, ForeignKey
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
from sqlalchemy.sql import func
|
|
from .base import Base
|
|
|
|
class AuthToken(Base):
|
|
__tablename__ = "tokens"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
token_hash: Mapped[str] = mapped_column(String(512), index=True, nullable=False, unique=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), nullable=False)
|
|
expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
|
|
revoked: Mapped[bool] = mapped_column(default=False, nullable=False)
|
|
revoked_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
|
def is_valid(self) -> bool:
|
|
return not self.revoked and self.expires_at > datetime.utcnow()
|