fix: 修复部门和菜单新增父级下的第一个子级菜单时sort找不到的bug

This commit is contained in:
ahhui
2023-08-07 14:50:34 +08:00
parent 89ba768932
commit f67869d428
2 changed files with 4 additions and 2 deletions

View File

@@ -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

View File

@@ -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)