intellif-aihub 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

aihub/models/task_center.py

@@ -1,117 +1,117 @@
-from __future__ import annotations
-
-import json
-from enum import Enum
-from typing import Optional, List
-
-from pydantic import BaseModel, Field, field_serializer, field_validator
-
-
-class TaskCenterPriorityEnum(Enum):
-    """
-    任务优先级枚举
-    """
-
-    low = "low"
-    medium = "medium"
-    high = "high"
-
-
-class LabelProjectTypeEnum(Enum):
-    """
-    任务类型枚举
-    """
-
-    IMAGE_CLASSIFICATION = 1
-    OBJECT_DETECTION = 2
-    SEMANTIC_SEGMENTATION = 3
-    TEXT_CLASSIFICATION = 4
-    TEXT_SEGMENTATION = 5
-    TEXT_RECOGNITION = 6
-    IMAGE_CAPTIONING = 7
-    IMAGE_STYLE_TRANSFER = 8
-    IMAGE_COLORIZATION = 9
-    IMAGE_RESTORATION = 10
-
-
-class CreateTaskOtherInfo(BaseModel):
-    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
-    dataset_id: int = Field(alias="dataset_id")
-    dataset_version_id: int = Field(alias="dataset_version_id")
-    doc_id: int = Field(alias="doc_id")
-    doc_type: str = Field(alias="doc_type", default="doc_center")
-
-
-class ProjectInfo(BaseModel):
-    label_project_id: int = Field(alias="label_project_id")
-    label_project_name: str = Field(alias="label_project_name")
-
-
-class TaskDetailOtherInfo(BaseModel):
-    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
-    dataset_id: int = Field(alias="dataset_id")
-    dataset_version_id: int = Field(alias="dataset_version_id")
-    doc_id: int = Field(alias="doc_id")
-    doc_type: str = Field(alias="doc_type", default="doc_center")
-    label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects")
-
-
-class CreateTaskReq(BaseModel):
-    name: str
-    description: Optional[str] = None
-    task_priority: Optional[str] = None
-    type: Optional[str] = None
-    receiver_id: Optional[int] = None
-    project_id: Optional[int] = None
-    other_info: CreateTaskOtherInfo = Field(alias="other_info")
-    estimated_delivery_at: Optional[int] = None
-
-    @field_serializer("other_info")
-    def serialize_other_info(self, value: CreateTaskOtherInfo) -> str:
-        """将 other_info 序列化为 JSON 字符串"""
-        return value.model_dump_json()
-
-
-class CreateTaskResp(BaseModel):
-    id: int = Field(alias="id")
-
-
-class LabelTaskDetail(BaseModel):
-    """任务详情"""
-
-    name: str
-    description: Optional[str] = Field(alias="description")
-    task_priority: Optional[str] = Field(alias="task_priority")
-    type: Optional[str] = Field(alias="type")
-    receiver_id: Optional[int] = Field(alias="receiver_id")
-    project_id: Optional[int] = None
-    other_info: TaskDetailOtherInfo = Field(alias="other_info")
-    estimated_delivery_at: Optional[int] = None
-
-    @field_serializer("other_info")
-    def serialize_other_info(self, value: TaskDetailOtherInfo) -> str:
-        """将 other_info 序列化为 JSON 字符串"""
-        return value.model_dump_json()
-
-    @field_validator("other_info", mode="before")
-    @classmethod
-    def parse_other_info(cls, value):
-        """将字符串解析为 TaskDetailOtherInfo 对象"""
-        if isinstance(value, str):
-            try:
-                # 解析 JSON 字符串为字典
-                data = json.loads(value)
-                # 创建 TaskDetailOtherInfo 对象
-                return TaskDetailOtherInfo(**data)
-            except (json.JSONDecodeError, TypeError, ValueError) as e:
-                raise ValueError(f"无法解析 other_info 字符串: {e}")
-        elif isinstance(value, dict):
-            # 如果传入的是字典，直接创建对象
-            return TaskDetailOtherInfo(**value)
-        elif isinstance(value, TaskDetailOtherInfo):
-            # 如果已经是对象，直接返回
-            return value
-        else:
-            raise ValueError(
-                f"other_info 必须是字符串、字典或 TaskDetailOtherInfo 对象，得到: {type(value)}"
-            )
+from __future__ import annotations
+
+import json
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, Field, field_serializer, field_validator
+
+
+class TaskCenterPriorityEnum(Enum):
+    """
+    任务优先级枚举
+    """
+
+    low = "low"
+    medium = "medium"
+    high = "high"
+
+
+class LabelProjectTypeEnum(Enum):
+    """
+    任务类型枚举
+     1 - 目标检测 2 - 语义分割 3 - 图片分类 4 - 实例分割 5 - 视频标注 6 - 人类偏好文本标注 7- 敏感预料文本标注 8 - 文本标注 9 - 关键点标注
+    """
+
+    OBJECT_DETECTION = 1
+    SEGMENTATION = 2
+    IMAGE_CLASSIFICATION = 3
+    INSTANCE_SEGMENTATION = 4
+    VIDEO_LABELING = 5
+    HUMAN_PREFERENCE_TEXT_LABELING = 6
+    SENSITIVE_TEXT_LABELING = 7
+    TEXT_LABELING = 8
+    KEYPOINT_LABELING = 9
+
+
+class CreateTaskOtherInfo(BaseModel):
+    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
+    dataset_id: int = Field(alias="dataset_id")
+    dataset_version_id: int = Field(alias="dataset_version_id")
+    doc_id: int = Field(alias="doc_id")
+    doc_type: str = Field(alias="doc_type", default="doc_center")
+
+
+class ProjectInfo(BaseModel):
+    label_project_id: int = Field(alias="label_project_id")
+    label_project_name: str = Field(alias="label_project_name")
+
+
+class TaskDetailOtherInfo(BaseModel):
+    label_project_type: LabelProjectTypeEnum = LabelProjectTypeEnum.IMAGE_CLASSIFICATION
+    dataset_id: int = Field(alias="dataset_id")
+    dataset_version_id: int = Field(alias="dataset_version_id")
+    doc_id: int = Field(alias="doc_id")
+    doc_type: str = Field(alias="doc_type", default="doc_center")
+    label_projects: Optional[List[ProjectInfo]] = Field(alias="label_projects")
+
+
+class CreateTaskReq(BaseModel):
+    name: str
+    description: Optional[str] = None
+    task_priority: Optional[str] = None
+    type: Optional[str] = None
+    receiver_id: Optional[int] = None
+    project_id: Optional[int] = None
+    other_info: CreateTaskOtherInfo = Field(alias="other_info")
+    estimated_delivery_at: Optional[int] = None
+
+    @field_serializer("other_info")
+    def serialize_other_info(self, value: CreateTaskOtherInfo) -> str:
+        """将 other_info 序列化为 JSON 字符串"""
+        return value.model_dump_json()
+
+
+class CreateTaskResp(BaseModel):
+    id: int = Field(alias="id")
+
+
+class LabelTaskDetail(BaseModel):
+    """任务详情"""
+
+    name: str
+    description: Optional[str] = Field(alias="description")
+    task_priority: Optional[str] = Field(alias="task_priority")
+    type: Optional[str] = Field(alias="type")
+    receiver_id: Optional[int] = Field(alias="receiver_id")
+    project_id: Optional[int] = None
+    other_info: TaskDetailOtherInfo = Field(alias="other_info")
+    estimated_delivery_at: Optional[int] = None
+
+    @field_serializer("other_info")
+    def serialize_other_info(self, value: TaskDetailOtherInfo) -> str:
+        """将 other_info 序列化为 JSON 字符串"""
+        return value.model_dump_json()
+
+    @field_validator("other_info", mode="before")
+    @classmethod
+    def parse_other_info(cls, value):
+        """将字符串解析为 TaskDetailOtherInfo 对象"""
+        if isinstance(value, str):
+            try:
+                # 解析 JSON 字符串为字典
+                data = json.loads(value)
+                # 创建 TaskDetailOtherInfo 对象
+                return TaskDetailOtherInfo(**data)
+            except (json.JSONDecodeError, TypeError, ValueError) as e:
+                raise ValueError(f"无法解析 other_info 字符串: {e}")
+        elif isinstance(value, dict):
+            # 如果传入的是字典，直接创建对象
+            return TaskDetailOtherInfo(**value)
+        elif isinstance(value, TaskDetailOtherInfo):
+            # 如果已经是对象，直接返回
+            return value
+        else:
+            raise ValueError(
+                f"other_info 必须是字符串、字典或 TaskDetailOtherInfo 对象，得到: {type(value)}"
+            )

aihub/models/user_system.py

@@ -0,0 +1,262 @@
+# !/usr/bin/env python
+# -*-coding:utf-8 -*-
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+# ======================================================================
+# COMMON
+# ======================================================================
+
+class Role(BaseModel):
+    id: int
+    name: str
+    role_type: int = Field(alias="role_type")
+    menu_ids: Optional[List[int]] = Field(None, alias="menu_ids")
+
+
+class Menu(BaseModel):
+    id: int
+    name: str
+    parent: int
+    auth: str
+
+
+class TreeMenu(BaseModel):
+    id: int
+    name: str
+    parent: int
+    auth: str
+    children: Optional[List["TreeMenu"]] = None
+    roles: Optional[List[Role]] = None
+
+
+class TagBrief(BaseModel):
+    id: int
+    name: str
+
+
+# ======================================================================
+# ------------------------------- AUTH ---------------------------------
+# ======================================================================
+
+class LoginRequest(BaseModel):
+    username: str = Field(alias="username")
+    password: str = Field(alias="password")
+
+
+class LoginResponse(BaseModel):
+    id: int = Field(alias="id")
+    token: str = Field(alias="token")
+
+
+class SignupRequest(BaseModel):
+    username: str = Field(alias="username")
+    password: str = Field(alias="password")
+    nickname: str = Field(alias="nickname")
+    email: str = Field(alias="email")
+    role_ids: List[int] = Field(alias="role_ids")
+
+
+class SignupResponse(BaseModel):
+    id: int = Field(alias="id")
+
+
+# ======================================================================
+# ------------------------------- MENU ---------------------------------
+# ======================================================================
+class ListMenusRequest(BaseModel):
+    need_roles: Optional[bool] = Field(None, alias="need_roles")
+
+
+class ListMenusResponse(BaseModel):
+    menus: List[TreeMenu] = Field(None, alias="menus")
+
+
+class CreateMenuRequest(BaseModel):
+    name: str = Field(alias="name")
+    parent: int = Field(alias="parent")
+    auth: str = Field(alias="auth")
+    role_ids: Optional[List[int]] = Field(None, alias="role_ids")
+
+
+class CreateMenuResponse(BaseModel):
+    id: int = Field(alias="id")
+
+
+class UpdateMenuRequest(BaseModel):
+    name: Optional[str] = Field(None, alias="name")
+    parent: Optional[int] = Field(None, alias="parent")
+    auth: str = Field(alias="auth")
+    role_ids: Optional[List[int]] = Field(None, alias="role_ids")
+
+
+class GetMenuRolesResponse(BaseModel):
+    role_ids: List[int] = Field(alias="role_ids")
+
+
+class SetMenuRolesRequest(BaseModel):
+    role_ids: List[int] = Field(alias="role_ids")
+
+
+class SearchMenusRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    name: Optional[str] = None
+    parent_ids: Optional[List[int]] = Field(None, alias="parent_ids")
+    auth: Optional[str] = None
+    menu_ids: Optional[List[int]] = Field(None, alias="menu_ids")
+
+
+class SearchMenusResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[Menu]
+
+
+# ======================================================================
+# ------------------------------- ROLE ---------------------------------
+# ======================================================================
+
+
+class CreateRoleRequest(BaseModel):
+    id: Optional[int] = Field(None, alias="id")
+    name: str
+    role_type: int = Field(alias="role_type")
+    menu_ids: Optional[List[int]] = Field(None, alias="menu_ids")
+
+
+class CreateRoleResponse(BaseModel):
+    id: int
+
+
+class UpdateRoleRequest(BaseModel):
+    name: Optional[str] = None
+    role_type: Optional[int] = Field(None, alias="role_type")
+    menu_ids: Optional[List[int]] = Field(None, alias="menu_ids")
+
+
+class GetRoleMenusResponse(BaseModel):
+    menu_ids: List[int] = Field(alias="menu_ids")
+
+
+class SetRoleMenusRequest(BaseModel):
+    menu_ids: List[int] = Field(alias="menu_ids")
+
+
+class ListRolesRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    role_type: Optional[int] = Field(None, alias="role_type")
+
+
+class ListRolesResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[Role]
+
+
+class SearchRolesRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    name: Optional[str] = None
+    role_ids: Optional[List[int]] = Field(None, alias="role_ids")
+    menu_ids: Optional[List[int]] = Field(None, alias="menu_ids")
+
+
+class SearchRolesResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[Role]
+
+
+# ======================================================================
+# ------------------------------- USER ---------------------------------
+# ======================================================================
+
+class User(BaseModel):
+    id: int
+    username: str
+    nickname: str
+    email: str
+    roles: Optional[List[Role]] = Field(None, alias="roles")
+    status: int
+    tags: Optional[List[TagBrief]] = Field(None, alias="tags")
+    created_at: int = Field(alias="created_at")
+
+
+class ListUsersRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    search_key: Optional[str] = Field(None, alias="search_key")
+
+
+class ListUsersResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[User]
+
+
+class CreateUserRequest(BaseModel):
+    id: int
+    username: str
+    password: str
+    nickname: str
+    email: str
+    role_ids: Optional[List[int]] = Field(None, alias="role_ids")
+    created_at: Optional[int] = Field(None, alias="created_at")
+    updated_at: Optional[int] = Field(None, alias="updated_at")
+    status: Optional[int] = None
+    tag_ids: Optional[List[int]] = Field(None, alias="tag_ids")
+
+
+class CreateUserResponse(BaseModel):
+    id: int
+
+
+class UpdateUserRequest(BaseModel):
+    username: Optional[str] = None
+    nickname: Optional[str] = None
+    email: Optional[str] = None
+    password: Optional[str] = None
+    role_ids: Optional[List[int]] = Field(default_factory=list, alias="role_ids")
+    status: Optional[int] = None
+    tag_ids: Optional[List[int]] = Field(default_factory=list, alias="tag_ids")
+
+
+class SetUserRolesRequest(BaseModel):
+    role_ids: List[int] = Field(alias="role_ids")
+
+
+class GetUserMenusResponse(BaseModel):
+    menus: List[TreeMenu]
+
+
+class SearchUsersRequest(BaseModel):
+    page_size: int = Field(20, alias="page_size")
+    page_num: int = Field(1, alias="page_num")
+    username: Optional[str] = None
+    nickname: Optional[str] = None
+    email: Optional[str] = None
+    user_ids: Optional[List[int]] = Field(None, alias="user_ids")
+    role_ids: Optional[List[int]] = Field(None, alias="role_ids")
+    role_names: Optional[List[str]] = Field(None, alias="role_names")
+    status: Optional[int] = None
+
+
+class SearchUsersResponse(BaseModel):
+    total: int
+    page_size: int = Field(alias="page_size")
+    page_num: int = Field(alias="page_num")
+    data: List[User]
+
+
+# 此行放在文件末尾，否则序列化报错
+TreeMenu.update_forward_refs()