Commit 43b683b2 authored by Tanner Prestegard's avatar Tanner Prestegard Committed by GraceDB

Add decorator for LV-EM only access

parent 8cee1ae4
...@@ -25,3 +25,26 @@ def internal_user_required(function=None, raise_exception=True, **kwargs): ...@@ -25,3 +25,26 @@ def internal_user_required(function=None, raise_exception=True, **kwargs):
if function: if function:
return actual_decorator(function) return actual_decorator(function)
return actual_decorator return actual_decorator
def lvem_observers_only(function=None, login_url=None, superuser_allowed=False,
"""Allow access only to non-LVC LV-EM observers"""
def check_groups(user):
in_lvem_obs = user.groups.filter(
in_lvc = user.groups.filter(name=settings.LVC_GROUP).exists()
if ((in_lvem_obs and not in_lvc) or
(superuser_allowed and user.is_superuser)):
return True
if raise_exception:
raise PermissionDenied
return False
actual_decorator = user_passes_test(check_groups, login_url=login_url)
if function:
return actual_decorator(function)
return actual_decorator
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment