feat: implement Tool model and related API for Tool management

Author: liuruibin

apps/common/constants/permission_constants.py

@@ -109,6 +109,8 @@ class PermissionConstants(Enum):
                                                                               RoleConstants.USER])
     USER_EDIT = Permission(group=Group.USER, operate=Operate.EDIT, role_list=[RoleConstants.ADMIN])
     USER_DELETE = Permission(group=Group.USER, operate=Operate.DELETE, role_list=[RoleConstants.ADMIN])
+    TOOL_CREATE = Permission(group=Group.USER, operate=Operate.CREATE, role_list=[RoleConstants.ADMIN,
+                                                                                  RoleConstants.USER])
 
     def get_workspace_application_permission(self):
         return lambda r, kwargs: Permission(group=self.value.group, operate=self.value.operate,

apps/maxkb/settings/base.py

@@ -39,6 +39,7 @@
     'drf_spectacular',
     'drf_spectacular_sidecar',
     'users.apps.UsersConfig',
+    'tools.apps.ToolConfig',
     'common',
     'system_manage'
 ]

apps/maxkb/urls.py

@@ -30,6 +30,7 @@
 SpectacularRedocView.authentication_classes = [AnonymousAuthentication]
 urlpatterns = [
     path("api/", include("users.urls")),
+    path("api/", include("tools.urls"))
 ]
 urlpatterns += [
     path('schema/', SpectacularAPIView.as_view(), name='schema'),  # schema的配置文件的路由，下面两个ui也是根据这个配置文件来生成的

apps/tools/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

apps/tools/admin.py

@@ -0,0 +1 @@
+# coding=utf-8

apps/tools/api/__init__.py

@@ -0,0 +1,18 @@
+# coding=utf-8
+
+from django.utils.translation import gettext_lazy as _
+from drf_yasg import openapi
+from rest_framework import serializers
+
+
+class PyLintApi(serializers.Serializer):
+    @staticmethod
+    def get_request_body_api():
+        return openapi.Schema(
+            type=openapi.TYPE_OBJECT,
+            required=['code'],
+            properties={
+                'code': openapi.Schema(type=openapi.TYPE_STRING, title=_('tool content'),
+                                       description=_('tool content'))
+            }
+        )

apps/tools/api/py_lint.py

@@ -0,0 +1,20 @@
+# coding=utf-8
+
+from common.mixins.api_mixin import APIMixin
+from common.result import ResultSerializer
+from tools.serializers.tool import ToolModelSerializer, ToolCreateRequest
+
+
+class ToolCreateResponse(ResultSerializer):
+    def get_data(self):
+        return ToolModelSerializer()
+
+
+class ToolCreateAPI(APIMixin):
+    @staticmethod
+    def get_request():
+        return ToolCreateRequest
+
+    @staticmethod
+    def get_response():
+        return ToolCreateResponse

apps/tools/api/tool.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ToolConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'tools'

apps/tools/apps.py

@@ -0,0 +1,42 @@
+# Generated by Django 5.2 on 2025-04-16 09:49
+
+import django.contrib.postgres.fields
+import django.db.models.deletion
+import uuid_utils.compat
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('users', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Tool',
+            fields=[
+                ('id', models.UUIDField(default=uuid_utils.compat.uuid7, editable=False, primary_key=True, serialize=False, verbose_name='主键id')),
+                ('name', models.CharField(max_length=64, verbose_name='函数名称')),
+                ('desc', models.CharField(max_length=128, verbose_name='描述')),
+                ('code', models.CharField(max_length=102400, verbose_name='python代码')),
+                ('input_field_list', django.contrib.postgres.fields.ArrayField(base_field=models.JSONField(default=dict, verbose_name='输入字段'), default=list, size=None, verbose_name='输入字段列表')),
+                ('init_field_list', models.JSONField(default=list, verbose_name='启动字段列表')),
+                ('icon', models.CharField(default='/ui/favicon.ico', max_length=256, verbose_name='函数库icon')),
+                ('is_active', models.BooleanField(default=True)),
+                ('permission_type', models.CharField(choices=[('SHARED', '共享'), ('PRIVATE', '私有')], default='PRIVATE', max_length=20, verbose_name='权限类型')),
+                ('tool_type', models.CharField(choices=[('INTERNAL', '内置'), ('PUBLIC', '公开')], default='PUBLIC', max_length=20, verbose_name='函数类型')),
+                ('template_id', models.UUIDField(default=None, null=True, verbose_name='模版id')),
+                ('module_id', models.UUIDField(default='root', null=True, verbose_name='模块id')),
+                ('init_params', models.CharField(max_length=102400, null=True, verbose_name='初始化参数')),
+                ('create_time', models.DateTimeField(auto_now_add=True, null=True, verbose_name='创建时间')),
+                ('update_time', models.DateTimeField(auto_now=True, null=True, verbose_name='修改时间')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='users.user', verbose_name='用户id')),
+            ],
+            options={
+                'db_table': 'tool',
+            },
+        ),
+    ]