29 lines
951 B
Python
29 lines
951 B
Python
from datetime import UTC, datetime
|
|
|
|
from sqlalchemy import TEXT, DateTime, Enum, ForeignKey
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from db.base import Base
|
|
from schemas.enums import LinkCodeStatus
|
|
|
|
|
|
class LinkCode(Base):
|
|
__tablename__ = "link_codes"
|
|
|
|
code: Mapped[str] = mapped_column(TEXT, nullable=False, unique=True, primary_key=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
|
|
status: Mapped[LinkCodeStatus] = mapped_column(
|
|
Enum(LinkCodeStatus, name="linkcodestatus"), nullable=False, default=LinkCodeStatus.ACTIVE
|
|
)
|
|
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True),
|
|
nullable=False,
|
|
default=lambda: datetime.now(UTC),
|
|
)
|
|
expires_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True),
|
|
nullable=False,
|
|
)
|
|
used_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=True)
|