From 870b1fabe590e7bfb90255930bc427ee82f84188 Mon Sep 17 00:00:00 2001 From: Arthur Date: Fri, 13 Oct 2023 09:21:22 -0700 Subject: [PATCH] 12336 review changes --- netbox/netbox/api/viewsets/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/netbox/netbox/api/viewsets/__init__.py b/netbox/netbox/api/viewsets/__init__.py index 7a12c060d..c6794bb61 100644 --- a/netbox/netbox/api/viewsets/__init__.py +++ b/netbox/netbox/api/viewsets/__init__.py @@ -161,20 +161,20 @@ class NetBoxModelViewSet( return super().perform_destroy(instance) -class MPTTLockedMixin(GenericViewSet): +class MPTTLockedMixin: """ Puts pglock on objects that derive from MPTTModel for parallel API calling. Note: If adding this to a view, must add the model name to ADVISORY_LOCK_KEYS """ def create(self, request, *args, **kwargs): - with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.verbose_name.lower().replace(" ", "")]): + with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.model_name]): return super().create(request, *args, **kwargs) def update(self, request, *args, **kwargs): - with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.verbose_name.lower().replace(" ", "")]): + with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.model_name]): return super().update(request, *args, **kwargs) def destroy(self, request, *args, **kwargs): - with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.verbose_name.lower().replace(" ", "")]): + with advisory_lock(ADVISORY_LOCK_KEYS[self.queryset.model._meta.model_name]): return super().destroy(request, *args, **kwargs)