.

atlas/decorators.py

@@ -1,66 +1,102 @@
 from django.core.exceptions import PermissionDenied
 from .models import Case, Series
 
+
 def user_is_author_or_atlas_series_checker_or_atlas_marker(function):
     def wrap(request, *args, **kwargs):
-        series = Series.objects.get(pk=kwargs['pk'])
-        if request.user in series.get_author_objects() or request.user.groups.filter(name='atlas_checker').exists() or request.user.groups.filter(name='atlas_marker').exists() or request.user.is_superuser:
+        series = Series.objects.get(pk=kwargs["pk"])
+        if (
+            request.user in series.get_author_objects()
+            or request.user.groups.filter(name="atlas_editor").exists()
+            or request.user.groups.filter(name="atlas_marker").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
     return wrap
 
+
 def user_is_author_or_atlas_series_checker(function):
     def wrap(request, *args, **kwargs):
-        series = Series.objects.get(pk=kwargs['pk'])
-        if request.user in series.get_author_objects() or request.user.groups.filter(name='atlas_checker').exists() or request.user.is_superuser:
+        series = Series.objects.get(pk=kwargs["pk"])
+        if (
+            request.user in series.get_author_objects()
+            or request.user.groups.filter(name="atlas_editor").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
     return wrap
 
-def user_is_author_or_atlas_checker_or_atlas_marker(function):
+
+def user_is_author_or_atlas_editor_or_atlas_marker(function):
     def wrap(request, *args, **kwargs):
-        atlas = Case.objects.get(pk=kwargs['pk'])
-        if request.user in atlas.author.all() or request.user.groups.filter(name='atlas_checker').exists()  or request.user.groups.filter(name='atlas_marker').exists() or request.user.is_superuser:
+        atlas = Case.objects.get(pk=kwargs["pk"])
+        if (
+            request.user in atlas.author.all()
+            or request.user.groups.filter(name="atlas_editor").exists()
+            or request.user.groups.filter(name="atlas_marker").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
     return wrap
 
-def user_is_author_or_atlas_checker(function):
+
+def user_is_author_or_atlas_editor(function):
     def wrap(request, *args, **kwargs):
-        atlas = Case.objects.get(pk=kwargs['pk'])
-        if request.user in atlas.author.all() or request.user.groups.filter(name='atlas_checker').exists() or request.user.is_superuser:
+        atlas = Case.objects.get(pk=kwargs["pk"])
+        if (
+            request.user in atlas.author.all()
+            or request.user.groups.filter(name="atlas_editor").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
     return wrap
 
-def user_is_atlas_checker(function):
+
+def user_is_atlas_editor(function):
     def wrap(request, *args, **kwargs):
-        if request.user.groups.filter(name='atlas_checker').exists() or request.user.is_superuser:
+        if (
+            request.user.groups.filter(name="atlas_editor").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
     return wrap
 
+
 def user_is_atlas_marker(function):
     def wrap(request, *args, **kwargs):
-        if request.user.groups.filter(name='atlas_marker').exists() or request.user.is_superuser:
+        if (
+            request.user.groups.filter(name="atlas_marker").exists()
+            or request.user.is_superuser
+        ):
             return function(request, *args, **kwargs)
         else:
             raise PermissionDenied
+
     wrap.__doc__ = function.__doc__
     wrap.__name__ = function.__name__
-    return wrap
+    return wrap