Merge branch 'v2' into pr@v2@feat_resource_permission

Author: shaohuzhang1

apps/common/constants/exception_code_constants.py

@@ -0,0 +1,43 @@
+# coding=utf-8
+"""
+    @project: qabot
+    @Author：虎
+    @file： exception_code_constants.py
+    @date：2023/9/4 14:09
+    @desc: 异常常量类
+"""
+from enum import Enum
+
+from common.exception.app_exception import AppApiException
+from django.utils.translation import gettext_lazy as _
+
+
+class ExceptionCodeConstantsValue:
+    def __init__(self, code, message):
+        self.code = code
+        self.message = message
+
+    def get_message(self):
+        return self.message
+
+    def get_code(self):
+        return self.code
+
+    def to_app_api_exception(self):
+        return AppApiException(code=self.code, message=self.message)
+
+
+class ExceptionCodeConstants(Enum):
+    INCORRECT_USERNAME_AND_PASSWORD = ExceptionCodeConstantsValue(1000, _('The username or password is incorrect'))
+    NOT_AUTHENTICATION = ExceptionCodeConstantsValue(1001, _('Please log in first and bring the user Token'))
+    EMAIL_SEND_ERROR = ExceptionCodeConstantsValue(1002, _('Email sending failed'))
+    EMAIL_FORMAT_ERROR = ExceptionCodeConstantsValue(1003, _('Email format error'))
+    EMAIL_IS_EXIST = ExceptionCodeConstantsValue(1004, _('The email has been registered, please log in directly'))
+    EMAIL_IS_NOT_EXIST = ExceptionCodeConstantsValue(1005, _('The email is not registered, please register first'))
+    CODE_ERROR = ExceptionCodeConstantsValue(1005,
+                                             _('The verification code is incorrect or the verification code has expired'))
+    USERNAME_IS_EXIST = ExceptionCodeConstantsValue(1006, _('The username has been registered, please log in directly'))
+    USERNAME_ERROR = ExceptionCodeConstantsValue(1006,
+                                                 _('The username cannot be empty and must be between 6 and 20 characters long.'))
+    PASSWORD_NOT_EQ_RE_PASSWORD = ExceptionCodeConstantsValue(1007,
+                                                              _('Password and confirmation password are inconsistent'))

apps/common/constants/permission_constants.py

@@ -153,6 +153,8 @@ class PermissionConstants(Enum):
     """
     USER_READ = Permission(group=Group.USER, operate=Operate.READ, role_list=[RoleConstants.ADMIN,
                                                                               RoleConstants.USER])
+    USER_CREATE = Permission(group=Group.USER, operate=Operate.CREATE,
+                             role_list=[RoleConstants.ADMIN, 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])
 

apps/common/models/db_model_manage.py

@@ -0,0 +1,35 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： db_model_manage.py
+    @date：2024/7/22 17:00
+    @desc:
+"""
+from importlib import import_module
+from django.conf import settings
+
+
+def new_instance_by_class_path(class_path: str):
+    parts = class_path.rpartition('.')
+    package_path = parts[0]
+    class_name = parts[2]
+    module = import_module(package_path)
+    HandlerClass = getattr(module, class_name)
+    return HandlerClass()
+
+
+class DBModelManage:
+    model_dict = {}
+
+    @staticmethod
+    def get_model(model_name):
+        return DBModelManage.model_dict.get(model_name)
+
+    @staticmethod
+    def init():
+        handles = [new_instance_by_class_path(class_path) for class_path in
+                   (settings.MODEL_HANDLES if hasattr(settings, 'MODEL_HANDLES') else [])]
+        for h in handles:
+            model_dict = h.get_model_dict()
+            DBModelManage.model_dict = {**DBModelManage.model_dict, **model_dict}

apps/common/models/handle/base_handle.py

@@ -0,0 +1,15 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： base_handle.py
+    @date：2024/7/22 17:02
+    @desc:
+"""
+from abc import ABC, abstractmethod
+
+
+class IBaseModelHandle(ABC):
+    @abstractmethod
+    def get_model_dict(self):
+        pass

apps/common/models/handle/impl/default_base_model_handle.py

@@ -0,0 +1,14 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： default_base_model_handle.py
+    @date：2024/7/22 17:06
+    @desc:
+"""
+from common.models.handle.base_handle import IBaseModelHandle
+
+
+class DefaultBaseModelHandle(IBaseModelHandle):
+    def get_model_dict(self):
+        return {}

apps/common/utils/common.py

@@ -17,10 +17,12 @@
 from typing import List, Dict
 
 from django.core.files.uploadedfile import InMemoryUploadedFile
+from django.db.models import QuerySet
 from django.utils.translation import gettext as _
 from pydub import AudioSegment
 
 from ..exception.app_exception import AppApiException
+from ..models.db_model_manage import DBModelManage
 
 
 def password_encrypt(row_password):
@@ -211,3 +213,23 @@ def query_params_to_single_dict(query_params: Dict):
         filter(lambda item: item is not None, [({key: value} if value is not None and len(value) > 0 else None) for
                                                key, value in
                                                query_params.items()])), {})
+
+
+def valid_license(model=None, count=None, message=None):
+    def inner(func):
+        def run(*args, **kwargs):
+            xpack_cache = DBModelManage.get_model('xpack_cache')
+            is_license_valid = xpack_cache.get('XPACK_LICENSE_IS_VALID', False) if xpack_cache is not None else False
+            record_count = QuerySet(model).count()
+
+            if not is_license_valid and record_count >= count:
+                error_message = message or _(
+                    'Limit {count} exceeded, please contact us (https://fit2cloud.com/).').format(
+                    count=count)
+                raise AppApiException(400, error_message)
+
+            return func(*args, **kwargs)
+
+        return run
+
+    return inner

apps/knowledge/api/knowledge.py

@@ -55,50 +55,6 @@ def get_response():
         return KnowledgeCreateResponse
 
 
-class KnowledgeLarkCreateAPI(APIMixin):
-    @staticmethod
-    def get_parameters():
-        return [
-            OpenApiParameter(
-                name="workspace_id",
-                description="工作空间id",
-                type=OpenApiTypes.STR,
-                location='path',
-                required=True,
-            )
-        ]
-
-    @staticmethod
-    def get_request():
-        return KnowledgeBaseCreateRequest
-
-    @staticmethod
-    def get_response():
-        return KnowledgeCreateResponse
-
-
-class KnowledgeYuqueCreateAPI(APIMixin):
-    @staticmethod
-    def get_parameters():
-        return [
-            OpenApiParameter(
-                name="workspace_id",
-                description="工作空间id",
-                type=OpenApiTypes.STR,
-                location='path',
-                required=True,
-            )
-        ]
-
-    @staticmethod
-    def get_request():
-        return KnowledgeBaseCreateRequest
-
-    @staticmethod
-    def get_response():
-        return KnowledgeCreateResponse
-
-
 class KnowledgeTreeReadAPI(APIMixin):
     @staticmethod
     def get_parameters():

apps/knowledge/urls.py

@@ -7,6 +7,4 @@
     path('workspace/<str:workspace_id>/knowledge', views.KnowledgeView.as_view()),
     path('workspace/<str:workspace_id>/knowledge/base', views.KnowledgeBaseView.as_view()),
     path('workspace/<str:workspace_id>/knowledge/web', views.KnowledgeWebView.as_view()),
-    path('workspace/<str:workspace_id>/knowledge/lark', views.KnowledgeLarkView.as_view()),
-    path('workspace/<str:workspace_id>/knowledge/yuque', views.KnowledgeYuqueView.as_view()),
 ]

apps/knowledge/views/knowledge.py

@@ -7,8 +7,7 @@
 from common.auth.authentication import has_permissions
 from common.constants.permission_constants import PermissionConstants
 from common.result import result
-from knowledge.api.knowledge import KnowledgeBaseCreateAPI, KnowledgeLarkCreateAPI, \
-    KnowledgeWebCreateAPI, KnowledgeYuqueCreateAPI, KnowledgeTreeReadAPI
+from knowledge.api.knowledge import KnowledgeBaseCreateAPI, KnowledgeWebCreateAPI, KnowledgeTreeReadAPI
 from knowledge.serializers.knowledge import KnowledgeSerializer, KnowledgeTreeSerializer
 
 
@@ -66,41 +65,3 @@ def post(self, request: Request, workspace_id: str):
         return result.success(KnowledgeSerializer.Create(
             data={'user_id': request.user.id, 'workspace_id': workspace_id}
         ).insert(request.data))
-
-
-class KnowledgeLarkView(APIView):
-    authentication_classes = [TokenAuth]
-
-    @extend_schema(
-        methods=['POST'],
-        description=_('Create lark knowledge'),
-        operation_id=_('Create lark knowledge'),
-        parameters=KnowledgeLarkCreateAPI.get_parameters(),
-        request=KnowledgeLarkCreateAPI.get_request(),
-        responses=KnowledgeLarkCreateAPI.get_response(),
-        tags=[_('Knowledge Base')]
-    )
-    @has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
-    def post(self, request: Request, workspace_id: str):
-        return result.success(KnowledgeSerializer.Create(
-            data={'user_id': request.user.id, 'workspace_id': workspace_id}
-        ).insert(request.data))
-
-
-class KnowledgeYuqueView(APIView):
-    authentication_classes = [TokenAuth]
-
-    @extend_schema(
-        methods=['POST'],
-        description=_('Create yuque knowledge'),
-        operation_id=_('Create yuque knowledge'),
-        parameters=KnowledgeYuqueCreateAPI.get_parameters(),
-        request=KnowledgeYuqueCreateAPI.get_request(),
-        responses=KnowledgeYuqueCreateAPI.get_response(),
-        tags=[_('Knowledge Base')]
-    )
-    @has_permissions(PermissionConstants.KNOWLEDGE_CREATE.get_workspace_permission())
-    def post(self, request: Request, workspace_id: str):
-        return result.success(KnowledgeSerializer.Create(
-            data={'user_id': request.user.id, 'workspace_id': workspace_id}
-        ).insert(request.data))

apps/modules/serializers/module.py

@@ -25,10 +25,10 @@ def get_module_type(source):
         return None
 
 
-MODULE_DEPTH = 3  # Module 不能超过3层
+MODULE_DEPTH = 2  # Module 不能超过3层
 
 
-def check_depth(source, parent_id):
+def check_depth(source, parent_id, current_depth=0):
     # Module 不能超过3层
     Module = get_module_type(source)
 
@@ -46,10 +46,28 @@ def check_depth(source, parent_id):
             current_parent_id = parent_node.parent_id
 
         # 验证层级深度
-        if depth > MODULE_DEPTH:
+        if depth + current_depth > MODULE_DEPTH:
             raise serializers.ValidationError(_('Module depth cannot exceed 3 levels'))
 
 
+def get_max_depth(current_node):
+    if not current_node:
+        return 0
+
+    # 获取所有后代节点
+    descendants = current_node.get_descendants()
+
+    if not descendants.exists():
+        return 0
+
+    # 获取最大深度
+    max_level = descendants.order_by('-level').first().level
+    current_level = current_node.level
+    max_depth = max_level - current_level
+
+    return max_depth
+
+
 class ModuleSerializer(serializers.Serializer):
     id = serializers.CharField(required=True, label=_('module id'))
     name = serializers.CharField(required=True, label=_('module name'))
@@ -110,7 +128,8 @@ def edit(self, instance):
             parent_id = instance.get('parent_id')
             if parent_id is not None and current_id != 'root':
                 # Module 不能超过3层
-                check_depth(self.data.get('source'), parent_id)
+                current_depth = get_max_depth(current_node)
+                check_depth(self.data.get('source'), parent_id, current_depth)
                 parent = Module.objects.get(id=parent_id)
                 current_node.move_to(parent)