From 5ece330457a602e279b777b0f5d6e8bc522308ac Mon Sep 17 00:00:00 2001 From: caoqianming Date: Fri, 12 Sep 2025 14:47:50 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20base=20=E4=BC=98=E5=8C=96get=5Fobject?= =?UTF-8?q?=E4=BA=8B=E5=8A=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/utils/viewsets.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/apps/utils/viewsets.py b/apps/utils/viewsets.py index 41588d26..6fa5fbb7 100755 --- a/apps/utils/viewsets.py +++ b/apps/utils/viewsets.py @@ -20,6 +20,7 @@ from drf_yasg.utils import swagger_auto_schema import json from django.db import connection from django.core.exceptions import ObjectDoesNotExist +from django.db.utils import NotSupportedError class CustomGenericViewSet(MyLoggingMixin, GenericViewSet): """ @@ -109,9 +110,11 @@ class CustomGenericViewSet(MyLoggingMixin, GenericViewSet): filter_kwargs = {self.lookup_field: self.kwargs[lookup_url_kwarg]} try: - obj = queryset.select_for_update().get(**filter_kwargs) + obj = queryset.get(**filter_kwargs) + l_obj = queryset.model.objects.select_for_update().get(pk=obj.pk) self.check_object_permissions(self.request, obj) - return obj + return l_obj + except ObjectDoesNotExist: raise Http404 else: