decorators.py 639 B

12345678910111213141516171819
  1. from django.contrib.auth import REDIRECT_FIELD_NAME
  2. from django.contrib.auth.decorators import user_passes_test
  3. def staff_member_required(
  4. view_func=None, redirect_field_name=REDIRECT_FIELD_NAME, login_url="admin:login"
  5. ):
  6. """
  7. Decorator for views that checks that the user is logged in and is a staff
  8. member, redirecting to the login page if necessary.
  9. """
  10. actual_decorator = user_passes_test(
  11. lambda u: u.is_active and u.is_staff,
  12. login_url=login_url,
  13. redirect_field_name=redirect_field_name,
  14. )
  15. if view_func:
  16. return actual_decorator(view_func)
  17. return actual_decorator