src/postorius/auth/decorators.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 1998-2021 by the Free Software Foundation, Inc.
   3 #
   4 # This file is part of Postorius.
   5 #
   6 # Postorius is free software: you can redistribute it and/or modify it under
   7 # the terms of the GNU General Public License as published by the Free
   8 # Software Foundation, either version 3 of the License, or (at your option)
   9 # any later version.
  10 #
  11 # Postorius is distributed in the hope that it will be useful, but WITHOUT
  12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13 # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  14 # more details.
  15 #
  16 # You should have received a copy of the GNU General Public License along with
  17 # Postorius.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 """Postorius view decorators."""
  20 from functools import wraps
  21
  22 from django.core.exceptions import PermissionDenied
  23
  24 from postorius.auth.utils import set_list_access_props
  25
  26
  27 def list_owner_required(fn):
  28     """Check if the logged in user is the list owner of the given list.
  29     Assumes that the request object is the first arg and that list_id
  30     is present in kwargs.
  31     """
  32     @wraps(fn)
  33     def wrapper(*args, **kwargs):
  34         user = args[0].user
  35         list_id = kwargs['list_id']
  36         if not user.is_authenticated:
  37             raise PermissionDenied
  38         if user.is_superuser:
  39             return fn(*args, **kwargs)
  40         set_list_access_props(user, list_id)
  41         if user.is_list_owner:
  42             return fn(*args, **kwargs)
  43         else:
  44             raise PermissionDenied
  45     return wrapper
  46
  47
  48 def list_moderator_required(fn):
  49     """Check if the logged in user is a moderator of the given list.
  50     Assumes that the request object is the first arg and that list_id
  51     is present in kwargs.
  52     """
  53     @wraps(fn)
  54     def wrapper(*args, **kwargs):
  55         user = args[0].user
  56         list_id = kwargs['list_id']
  57         if not user.is_authenticated:
  58             raise PermissionDenied
  59         if user.is_superuser:
  60             return fn(*args, **kwargs)
  61         set_list_access_props(user, list_id)
  62         if user.is_list_owner or user.is_list_moderator:
  63             return fn(*args, **kwargs)
  64         else:
  65             raise PermissionDenied
  66     return wrapper
  67
  68
  69 def superuser_required(fn):
  70     """Make sure that the logged in user is a superuser or otherwise raise
  71     PermissionDenied.
  72     Assumes the request object to be the first arg."""
  73     @wraps(fn)
  74     def wrapper(*args, **kwargs):
  75         user = args[0].user
  76         if not user.is_superuser:
  77             raise PermissionDenied
  78         return fn(*args, **kwargs)
  79     return wrapper