Class Import

apps/tools/serializers/tool.py:904–1144

                    os.remove(file_name)

    class Import(serializers.Serializer):
        file = UploadedFileField(required=True, label=_("file"))
        user_id = serializers.UUIDField(required=True, label=_("User ID"))
        workspace_id = serializers.CharField(required=True, label=_("workspace id"))
        folder_id = serializers.CharField(required=False, allow_null=True, label=_("folder id"))

        @staticmethod
        def to_tool_workflow(work_flow, update_tool_map):
            for node in work_flow.get("nodes", []):
                hand_node(node, update_tool_map)
                if node.get("type") == "loop-node":
                    for n in node.get("properties", {}).get("node_data", {}).get("loop_body", {}).get("nodes", []):
                        hand_node(n, update_tool_map)
            return work_flow

        @staticmethod
        def to_tool(tool, workspace_id, user_id, folder_id):
            # 如果是技能类型的工具，需要将code保存为文件
            code = tool.get("code")
            if tool.get("tool_type") == ToolType.SKILL:
                skill_file_id = uuid.uuid7()
                skill_file = File(
                    id=skill_file_id,
                    file_name=f"{tool.get('name')}.zip",
                    source_type=FileSourceType.TOOL,
                    source_id=tool.get("id"),
                    meta={},
                )
                skill_file.save(base64.b64decode(code))
                tool["code"] = skill_file_id
            return Tool(
                id=tool.get("id"),
                user_id=user_id,
                name=tool.get("name"),
                code=tool.get("code"),
                template_id=tool.get("template_id"),
                input_field_list=tool.get("input_field_list"),
                init_field_list=tool.get("init_field_list"),
                is_active=False
                if (len((tool.get("init_field_list") or [])) > 0 or tool.get("tool_type") == ToolType.WORKFLOW)
                else tool.get("is_active"),
                tool_type=tool.get("tool_type", "CUSTOM") or "CUSTOM",
                scope=ToolScope.SHARED if workspace_id == "None" else ToolScope.WORKSPACE,
                folder_id=folder_id if folder_id else "default" if workspace_id == "None" else workspace_id,
                workspace_id=workspace_id,
            )

        def import_workflow_tools(self, tool, workspace_id, user_id, folder_id, new_child_policy):
            """

            @param tool:                  工具对象
            @param workspace_id:          工作空间id
            @param user_id:               用户id
            @param folder_id:             文件夹id
            @param new_child_policy:      子工具创建策略
                                          0: 不创建
                                          1: 对比创建: 如果存在就不创建 不存在则创建
                                          2: 全部创建