11

app/schemas/admin_card_key.py

@@ -14,6 +14,8 @@ class CardKeyOut(BaseModel):
     activated_at: datetime | None = None
     user_id: int | None = None
     username: str | None = None
+    oem_id: int | None = None
+    agent_id: int | None = None
     remark: str | None = None
 
 
@@ -22,6 +24,8 @@ class CardKeyCreate(BaseModel):
     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)
+    oem_id: int | None = Field(default=None, description="预分配给 OEM")
+    agent_id: int | None = Field(default=None, description="预分配给代理")
 
     @field_validator("remark")
     @classmethod
@@ -43,6 +47,8 @@ class CardKeyCreate(BaseModel):
     def serial_only_for_single(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 与代理只能指定其一")
         return self