1

app/schemas/admin_card_key.py

@@ -3,13 +3,19 @@ from typing import Literal
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
 
+from app.core.card_key_types import CARD_TYPE_DURATION, CARD_TYPE_POINTS
+
+CardKeyType = Literal["时长", "点数"]
+
 
 class CardKeyOut(BaseModel):
     model_config = ConfigDict(from_attributes=True)
 
     id: int
     serial_number: str
+    card_type: CardKeyType
     duration_days: int
+    points_amount: float | None = None
     created_at: datetime
     activated_at: datetime | None = None
     user_id: int | None = None
@@ -22,7 +28,13 @@ class CardKeyOut(BaseModel):
 
 
 class CardKeyCreate(BaseModel):
-    duration_days: int = Field(ge=1, le=3650, description="有效天数")
+    card_type: CardKeyType = Field(default=CARD_TYPE_DURATION, description="卡密类型")
+    duration_days: int | None = Field(
+        default=None, ge=1, le=3650, description="时长卡有效天数"
+    )
+    points_amount: float | None = Field(
+        default=None, gt=0, description="点数卡点数面值"
+    )
     remark: str | None = Field(default=None, max_length=255)
     count: int = Field(default=1, ge=1, le=100, description="批量生成数量")
     serial_number: str | None = Field(default=None, max_length=64)
@@ -46,16 +58,24 @@ class CardKeyCreate(BaseModel):
         return stripped or None
 
     @model_validator(mode="after")
-    def serial_only_for_single(self) -> "CardKeyCreate":
+    def validate_type_and_value(self) -> "CardKeyCreate":
         if self.serial_number and self.count != 1:
             raise ValueError("指定序列号时仅可生成 1 张卡密")
         if self.oem_id is not None and self.agent_id is not None:
             raise ValueError("OEM 与代理只能指定其一")
+        if self.card_type == CARD_TYPE_DURATION:
+            if self.duration_days is None:
+                raise ValueError("时长卡密须填写有效天数")
+        elif self.card_type == CARD_TYPE_POINTS:
+            if self.points_amount is None:
+                raise ValueError("点数卡密须填写点数")
         return self
 
 
 class CardKeyUpdate(BaseModel):
+    card_type: CardKeyType | None = None
     duration_days: int | None = Field(default=None, ge=1, le=3650)
+    points_amount: float | None = Field(default=None, gt=0)
     remark: str | None = Field(default=None, max_length=255)
 
     @field_validator("remark")
@@ -66,6 +86,14 @@ class CardKeyUpdate(BaseModel):
         stripped = value.strip()
         return stripped or None
 
+    @model_validator(mode="after")
+    def validate_type_and_value(self) -> "CardKeyUpdate":
+        if self.card_type == CARD_TYPE_DURATION and self.points_amount is not None:
+            raise ValueError("时长卡密不能设置点数")
+        if self.card_type == CARD_TYPE_POINTS and self.duration_days is not None:
+            raise ValueError("点数卡密不能设置有效天数")
+        return self
+
 
 class CardKeyBatchCreateResult(BaseModel):
     items: list[CardKeyOut]