import secrets import uuid from datetime import datetime, timedelta, timezone from sqlalchemy import DateTime, ForeignKey, Text from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.orm import Mapped, mapped_column from ..database import Base TOKEN_TTL_HOURS = 24 class EmailVerificationToken(Base): __tablename__ = "email_verification_tokens" id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4) user_id: Mapped[uuid.UUID] = mapped_column( UUID(as_uuid=True), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, ) token: Mapped[str] = mapped_column(Text, nullable=False, unique=True) expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False) used_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True) @staticmethod def make(user_id: uuid.UUID) -> "EmailVerificationToken": return EmailVerificationToken( user_id=user_id, token=secrets.token_urlsafe(32), expires_at=datetime.now(timezone.utc) + timedelta(hours=TOKEN_TTL_HOURS), )