init

backend/system/views/menu.py

@@ -0,0 +1,131 @@
+from django_filters.rest_framework import DjangoFilterBackend
+from rest_framework import viewsets, serializers, status
+from rest_framework.decorators import action
+from rest_framework.filters import SearchFilter, OrderingFilter
+from rest_framework.response import Response
+
+from system.models import Menu, MenuMeta
+from utils.custom_model_viewSet import CustomModelViewSet
+
+
+class MenuMetaSerializer(serializers.ModelSerializer):
+    """菜单元数据序列化器"""
+    class Meta:
+        model = MenuMeta
+        fields = '__all__'
+
+class MenuSerializer(serializers.ModelSerializer):
+    """菜单序列化器"""
+    parent = serializers.CharField(source='pid.name', read_only=True)
+    meta = MenuMetaSerializer()
+    children = serializers.SerializerMethodField()
+    status_text = serializers.SerializerMethodField()
+    type_text = serializers.SerializerMethodField()
+
+    class Meta:
+        model = Menu
+        fields = '__all__'
+        read_only_fields = ['id', 'create_time', 'update_time']
+
+    def get_children(self, obj):
+        """获取子菜单"""
+        children = obj.children.all()
+        if children:
+            return MenuSerializer(children, many=True).data
+        return []
+
+    def get_status_text(self, obj):
+        """获取状态文本"""
+        return obj.get_status_display()
+
+    def get_type_text(self, obj):
+        """获取菜单类型文本"""
+        return obj.get_type_display()
+
+    def create(self, validated_data):
+        """创建菜单及关联的元数据"""
+        meta_data = validated_data.pop('meta')
+        meta = MenuMeta.objects.create(**meta_data)
+        menu = Menu.objects.create(meta=meta, **validated_data)
+        return menu
+
+    def update(self, instance, validated_data):
+        """更新菜单及关联的元数据"""
+        meta_data = validated_data.pop('meta', {})
+        meta_serializer = self.fields['meta']
+        meta_serializer.update(instance.meta, meta_data)
+        return super().update(instance, validated_data)
+
+
+
+class MenuMetaViewSet(viewsets.ModelViewSet):
+    """菜单元数据视图集"""
+    queryset = MenuMeta.objects.all()
+    serializer_class = MenuMetaSerializer
+
+
+class MenuViewSet(CustomModelViewSet):
+    """菜单管理视图集"""
+    queryset = Menu.objects.filter(pid__isnull=True).order_by('id', 'status')
+    serializer_class = MenuSerializer
+    filter_backends = [DjangoFilterBackend, SearchFilter, OrderingFilter]
+    filterset_fields = ['status', 'type', 'pid', 'name']
+    search_fields = ['name', 'path', 'auth_code']
+    ordering_fields = ['meta__order', 'create_time']
+
+    @action(detail=False, methods=['get'])
+    def tree(self, request):
+        """获取菜单树形结构"""
+        queryset = self.get_queryset().filter(pid__isnull=True)
+        serializer = self.get_serializer(queryset, many=True)
+        return Response(serializer.data)
+
+    @action(detail=False, methods=['get'], url_path='name-exists')
+    def name_exists(self, request):
+        return self._build_response()
+
+    @action(detail=False, methods=['get'], url_path='name-search')
+    def name_search(self, request):
+        name = request.GET.get('name')
+        pk = request.GET.get('id', None)
+        queryset = Menu.objects.all()
+        if pk:
+            queryset = queryset.exclude(pk=pk)
+        if name:
+            queryset = queryset.filter(name=name)
+        has_menu_name = queryset.exists()
+        print(has_menu_name, 'has_menu_name')
+        return self._build_response(data=has_menu_name)
+
+    @action(detail=False, methods=['get'], url_path='path-exists')
+    def path_exists(self, request):
+        return self._build_response()
+
+
+    def update(self, request, *args, **kwargs):
+        partial = kwargs.pop('partial', False)
+        pk = kwargs['pk']
+        instance = Menu.objects.get(pk=pk)
+        serializer = self.get_serializer(instance, data=request.data, partial=partial)
+        serializer.is_valid(raise_exception=True)
+        self.perform_update(serializer)
+
+        if getattr(instance, '_prefetched_objects_cache', None):
+            # If 'prefetch_related' has been applied to a queryset, we need to
+            # forcibly invalidate the prefetch cache on the instance.
+            instance._prefetched_objects_cache = {}
+        headers = self.get_success_headers(serializer.data)
+        return self._build_response(
+            data=serializer.data,
+            message="ok",
+            status=status.HTTP_200_OK,
+        )
+
+    def destroy(self, request, *args, **kwargs):
+        pk = kwargs['pk']
+        instance = Menu.objects.get(pk=pk)
+        self.perform_destroy(instance)
+        return self._build_response(
+            message="ok",
+            status=status.HTTP_200_OK,
+        )