mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
import re
|
|
|
|
import uuid_utils.compat as uuid
|
|
from django.core import validators
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from tools.models import Tool, ToolScope
|
|
|
|
|
|
class ToolModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Tool
|
|
fields = ['id', 'name', 'icon', 'desc', 'code', 'input_field_list', 'init_field_list', 'init_params',
|
|
'scope', 'is_active', 'user_id', 'template_id',
|
|
'create_time', 'update_time']
|
|
|
|
|
|
class ToolInputField(serializers.Serializer):
|
|
name = serializers.CharField(required=True, label=_('variable name'))
|
|
is_required = serializers.BooleanField(required=True, label=_('required'))
|
|
type = serializers.CharField(required=True, label=_('type'), validators=[
|
|
validators.RegexValidator(regex=re.compile("^string|int|dict|array|float$"),
|
|
message=_('fields only support string|int|dict|array|float'), code=500)
|
|
])
|
|
source = serializers.CharField(required=True, label=_('source'), validators=[
|
|
validators.RegexValidator(regex=re.compile("^custom|reference$"),
|
|
message=_('The field only supports custom|reference'), code=500)
|
|
])
|
|
|
|
|
|
class ToolCreateRequest(serializers.Serializer):
|
|
name = serializers.CharField(required=True, label=_('tool name'))
|
|
|
|
desc = serializers.CharField(required=False, allow_null=True, allow_blank=True,
|
|
label=_('tool description'))
|
|
|
|
code = serializers.CharField(required=True, label=_('tool content'))
|
|
|
|
input_field_list = serializers.ListField(child=ToolInputField(), required=True, label=_('input field list'))
|
|
|
|
init_field_list = serializers.ListField(required=False, default=list, label=_('init field list'))
|
|
|
|
is_active = serializers.BooleanField(required=False, label=_('Is active'))
|
|
|
|
|
|
class ToolSerializer(serializers.Serializer):
|
|
class Create(serializers.Serializer):
|
|
user_id = serializers.UUIDField(required=True, label=_('user id'))
|
|
|
|
def insert(self, instance, with_valid=True):
|
|
if with_valid:
|
|
self.is_valid(raise_exception=True)
|
|
ToolCreateRequest(data=instance).is_valid(raise_exception=True)
|
|
tool = Tool(id=uuid.uuid7(),
|
|
name=instance.get('name'),
|
|
desc=instance.get('desc'),
|
|
code=instance.get('code'),
|
|
user_id=self.data.get('user_id'),
|
|
input_field_list=instance.get('input_field_list'),
|
|
init_field_list=instance.get('init_field_list'),
|
|
scope=ToolScope.WORKSPACE,
|
|
is_active=False)
|
|
tool.save()
|
|
return ToolModelSerializer(tool).data
|