custom action name support in default perms

mesemus · Jan 22, 2018 · 0c7375d · 0c7375d
1 parent d012163
commit 0c7375d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/rest_delegated_permissions/permissions.py b/rest_delegated_permissions/permissions.py
@@ -159,15 +159,9 @@ def has_object_permission(self, request, view, obj):
                self.object_permissions.has_object_permission(request, view, obj)
 
     def get_queryset_filters(self, rest_permissions, qs, user, action):
-        operation = {
-            'retrieve': 'view',
-            'view': 'view',
-            'update': 'change',
-            'change': 'change',
-            'delete': 'delete',
-            'destroy': 'delete',
-            'partial_update': 'change'
-        }[action]
+        known_operations = {'retrieve': 'view', 'view': 'view', 'update': 'change', 'change': 'change', 'delete': 'delete',
+                   'destroy': 'delete', 'partial_update': 'change'}
+        operation = known_operations.get(action, action)
         ct = ContentType.objects.get_for_model(qs.model)
         perm = '%s_%s' % (operation, ct.model)
 

diff --git a/setup.py b/setup.py
@@ -34,7 +34,7 @@ class InvoiceViewSet(ModelViewSet):
 
 setup(
     name='django-rest-delegated-permissions',
-    version='0.5.6',
+    version='0.5.6.1',
     packages=[
         'rest_delegated_permissions',
     ],