From 2eb29e655b10f1596ff37ae1cedc6564334e07b7 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 15 Sep 2025 09:40:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20base=20=E4=BC=98=E5=8C=96get=5Fobject?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/utils/viewsets.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/apps/utils/viewsets.py b/apps/utils/viewsets.py
index b824df3..6ce2052 100755
--- a/apps/utils/viewsets.py
+++ b/apps/utils/viewsets.py
@@ -109,10 +109,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:
             # 其他情况：普通查询