diff --git a/backend/dvadmin/system/views/dept.py b/backend/dvadmin/system/views/dept.py index c4681fe..8d06c30 100644 --- a/backend/dvadmin/system/views/dept.py +++ b/backend/dvadmin/system/views/dept.py @@ -69,7 +69,8 @@ class DeptCreateUpdateSerializer(CustomModelSerializer): value = validated_data.get('parent', None) if value is None: validated_data['parent'] = self.request.user.dept - last_sort = Dept.objects.filter(parent=self.request.user.dept).order_by('-sort').first().sort + dept_obj = Dept.objects.filter(parent=self.request.user.dept).order_by('-sort').first() + last_sort = dept_obj.sort if dept_obj else 0 validated_data['sort'] = last_sort + 1 instance = super().create(validated_data) instance.dept_belong_id = instance.id diff --git a/backend/dvadmin/system/views/menu.py b/backend/dvadmin/system/views/menu.py index 633fa72..6899646 100644 --- a/backend/dvadmin/system/views/menu.py +++ b/backend/dvadmin/system/views/menu.py @@ -50,7 +50,8 @@ class MenuCreateSerializer(CustomModelSerializer): name = serializers.CharField(required=False) def create(self, validated_data): - last_sort = Menu.objects.filter(parent_id=validated_data.get('parent', None)).order_by('-sort').first().sort + menu_obj = Menu.objects.filter(parent_id=validated_data.get('parent', None)).order_by('-sort').first() + last_sort = menu_obj.sort if menu_obj else 0 validated_data['sort'] = last_sort + 1 return super().create(validated_data)