feat: base 优化get_object

2025-09-15 09:40:12 +08:00 · 2025-09-15 09:40:12 +08:00 · 67b92f0dd4
parent 06e86330bd
commit 67b92f0dd4
1 changed files with 2 additions and 2 deletions
--- 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:
            # 其他情况：普通查询