diff --git a/apps/utils/viewsets.py b/apps/utils/viewsets.py index 6fa5fbb7..9947b4a0 100755 --- a/apps/utils/viewsets.py +++ b/apps/utils/viewsets.py @@ -112,10 +112,10 @@ class CustomGenericViewSet(MyLoggingMixin, GenericViewSet): try: obj = queryset.get(**filter_kwargs) l_obj = queryset.model.objects.select_for_update().get(pk=obj.pk) - self.check_object_permissions(self.request, obj) + self.check_object_permissions(self.request, l_obj) return l_obj - except ObjectDoesNotExist: + except queryset.model.DoesNotExist: raise Http404 else: # 其他情况:普通查询