feat(invite): 添加邀请码功能模块

2026-04-07 18:43:49 +08:00
parent b22673ce49
commit c121c1092f
4 changed files with 962 additions and 1 deletions
@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+import uuid
+from datetime import datetime
+
+from sqlalchemy import CheckConstraint, DateTime, ForeignKey, Integer, String
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import Mapped, mapped_column
+
+from core.db.base import Base, TimestampMixin
+from core.db.types import json_jsonb
+from schemas.enums import InviteCodeStatus
+
+__all__ = ["InviteCode", "InviteCodeStatus"]
+
+
+class InviteCode(TimestampMixin, Base):
+    __tablename__: str = "invite_codes"
+    __table_args__ = (
+        CheckConstraint(
+            "status IN ('active', 'disabled', 'expired')",
+            name="invite_codes_status_check",
+        ),
+        CheckConstraint("used_count >= 0", name="invite_codes_used_count_check"),
+        CheckConstraint(
+            "max_uses IS NULL OR max_uses >= 1",
+            name="invite_codes_max_uses_check",
+        ),
+        {"extend_existing": True},
+    )
+
+    id: Mapped[uuid.UUID] = mapped_column(
+        UUID(as_uuid=True),
+        primary_key=True,
+        default=uuid.uuid4,
+    )
+    code: Mapped[str] = mapped_column(
+        String(6),
+        nullable=False,
+        unique=True,
+        index=True,
+    )
+    owner_id: Mapped[uuid.UUID | None] = mapped_column(
+        UUID(as_uuid=True),
+        ForeignKey("profiles.id", ondelete="SET NULL"),
+        nullable=True,
+        index=True,
+    )
+    status: Mapped[str] = mapped_column(
+        String(20),
+        nullable=False,
+        default=InviteCodeStatus.ACTIVE.value,
+    )
+    used_count: Mapped[int] = mapped_column(
+        Integer,
+        nullable=False,
+        default=0,
+    )
+    max_uses: Mapped[int | None] = mapped_column(
+        Integer,
+        nullable=True,
+    )
+    expires_at: Mapped[datetime | None] = mapped_column(
+        DateTime(timezone=True),
+        nullable=True,
+    )
+    reward_config: Mapped[dict[str, object]] = mapped_column(
+        json_jsonb,
+        nullable=False,
+        server_default="{}",
+    )
@@ -1,11 +1,51 @@
 from __future__ import annotations

+from datetime import datetime
 from typing import ClassVar
+from uuid import UUID

-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
+
+from schemas.enums import InviteCodeStatus


 class InviteCodeRewardConfig(BaseModel):
    model_config: ClassVar[ConfigDict] = ConfigDict(extra="allow")

    pass
+
+
+class InviteCodeBase(BaseModel):
+    code: str = Field(
+        ...,
+        min_length=6,
+        max_length=6,
+        pattern=r"^[ABCDEFGHJKMNPQRSTUVWXYZ23456789]{6}$",
+    )
+    status: InviteCodeStatus = InviteCodeStatus.ACTIVE
+    used_count: int = Field(default=0, ge=0)
+    max_uses: int | None = Field(default=None, ge=1)
+    expires_at: datetime | None = None
+    reward_config: InviteCodeRewardConfig = Field(
+        default_factory=InviteCodeRewardConfig
+    )
+
+
+class InviteCodeCreate(InviteCodeBase):
+    owner_id: UUID | None = None
+
+
+class InviteCodeUpdate(BaseModel):
+    status: InviteCodeStatus | None = None
+    max_uses: int | None = Field(default=None, ge=1)
+    expires_at: datetime | None = None
+    reward_config: InviteCodeRewardConfig | None = None
+
+
+class InviteCodeRead(InviteCodeBase):
+    model_config: ClassVar[ConfigDict] = ConfigDict(from_attributes=True)
+
+    id: UUID
+    owner_id: UUID | None
+    created_at: datetime
+    updated_at: datetime