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
 
 

app/schemas/admin_oem_branding.py

@@ -0,0 +1,51 @@
+from datetime import datetime
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator
+
+
+class OemBrandingOut(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    user_id: int
+    software_name: str = ""
+    logo_path: str | None = None
+    wechat: str | None = None
+    qq: str | None = None
+    wechat_qrcode: str | None = None
+    qq_qrcode: str | None = None
+    description: str | None = None
+    created_at: datetime
+    updated_at: datetime
+
+
+class OemBrandingUpdate(BaseModel):
+    """整表保存默认 OEM 品牌配置（允许置空图片与联系方式）。"""
+
+    software_name: str = Field(default="", max_length=128)
+    logo_path: str | None = None
+    wechat: str | None = Field(default=None, max_length=128)
+    qq: str | None = Field(default=None, max_length=128)
+    wechat_qrcode: str | None = None
+    qq_qrcode: str | None = None
+
+    @field_validator("software_name")
+    @classmethod
+    def strip_software_name(cls, value: str) -> str:
+        return value.strip()
+
+    @field_validator("wechat", "qq")
+    @classmethod
+    def strip_optional_text(cls, value: str | None) -> str | None:
+        if value is None:
+            return None
+        stripped = value.strip()
+        return stripped or None
+
+    @field_validator("logo_path", "wechat_qrcode", "qq_qrcode")
+    @classmethod
+    def normalize_image_value(cls, value: str | None) -> str | None:
+        if value is None:
+            return None
+        stripped = value.strip()
+        return stripped or None

app/schemas/admin_user.py

@@ -10,6 +10,8 @@ class AdminUserOut(BaseModel):
     username: str
     phone: str | None = None
     role_id: int
+    oem_id: int | None = None
+    agent_id: int | None = None
     vip_end_time: datetime | None = None
     created_at: datetime
     updated_at: datetime
@@ -20,6 +22,8 @@ class AdminUserCreate(BaseModel):
     password: str = Field(min_length=6, max_length=128)
     phone: str | None = Field(default=None, max_length=64)
     role_id: int = Field(default=0, ge=0, le=3)
+    oem_id: int | None = None
+    agent_id: int | None = None
     vip_end_time: datetime | None = None
 
     @field_validator("username")
@@ -40,7 +44,9 @@ class AdminUserUpdate(BaseModel):
     username: str | None = Field(default=None, min_length=1, max_length=64)
     password: str | None = Field(default=None, min_length=6, max_length=128)
     phone: str | None = Field(default=None, max_length=64)
-    role_id: int | None = Field(default=None, ge=0, le=2)
+    role_id: int | None = Field(default=None, ge=0, le=3)
+    oem_id: int | None = None
+    agent_id: int | None = None
     vip_end_time: datetime | None = None
     clear_vip_end_time: bool = False
 

app/schemas/oem_public.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel, ConfigDict
+
+
+class OemPublicBrandingOut(BaseModel):
+    """对外公开的品牌信息（无需登录）。"""
+
+    model_config = ConfigDict(from_attributes=True)
+
+    oem_id: int
+    software_name: str = ""
+    logo_url: str | None = None
+    wechat: str | None = None