22 lines
844 B
Python
Executable File
22 lines
844 B
Python
Executable File
import enum
|
|
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Enum
|
|
from sqlalchemy.sql import func
|
|
from app.db.base import Base
|
|
|
|
class TokenType(str, enum.Enum):
|
|
email_verify = "email_verify"
|
|
password_reset = "password_reset"
|
|
|
|
class VerificationToken(Base):
|
|
__tablename__ = "verification_tokens"
|
|
__table_args__ = {"schema": "data"}
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("data.users.id", ondelete="CASCADE"), nullable=False)
|
|
|
|
token_hash = Column(String(64), unique=True, index=True, nullable=False)
|
|
token_type = Column(Enum(TokenType, name="tokentype", schema="data"), nullable=False)
|
|
|
|
expires_at = Column(DateTime(timezone=True), nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False)
|