How to create a custom decorator in Django?

by

Played around with the various links above and couldn’t get them working and then came across this really simple one which I adapted. http://code.activestate.com/recipes/498217-custom-django-login_required-decorator/

from functools import wraps
from django.http import HttpResponseRedirect

def authors_only(function):
  @wraps(function)
  def wrap(request, *args, **kwargs):

        profile = request.user.get_profile()
        if profile.usertype == 'Author':
             return function(request, *args, **kwargs)
        else:
            return HttpResponseRedirect("https://stackoverflow.com/")

  return wrap

Using @wraps is better than manually overriding like doing wrap.__doc__ = fn.__doc__. Amongst other things, it ensures your wrapper function gets the same name as the wrapped function.

See https://docs.python.org/2/library/functools.html

More Related Contents:

  1. How to make function decorators and chain them together in Python?
  2. Python decorator? – can someone please explain this? [duplicate]
  3. How to make function decorators and chain them together?
  4. How does the @property decorator work in Python?
  5. How to call function that takes an argument in a Django template?
  6. Decorator execution order
  7. How to inject variable into scope with a decorator?
  8. Python Class Based Decorator with parameters that can decorate a method or a function
  9. How do I make function decorators and chain them together?
  10. How to do a conditional decorator in python?
  11. Decorator error: NoneType object is not callable
  12. How to skip a pytest using an external fixture?
  13. How are lambdas useful? [closed]
  14. Multiple ModelAdmins/views for same model in Django admin
  15. Python decorators in classes
  16. How can I pass information (a variable) out of a function? How can I use that from outside, e.g. in other functions?
  17. How do I pass variables across functions?
  18. Django Model Field Default Based Off Another Field in Same Model
  19. Django orm get latest for each group
  20. int([x[, base]]). Square brackets in functions in Python documentation? [duplicate]
  21. Using a Django custom model method property in order_by()
  22. Split models.py into several files
  23. Unexpected NoReverseMatch error when using include() in urls patterns
  24. Why is “None” printed after my function’s output?
  25. Can I combine two decorators into a single one in Python?
  26. Is it possible to not return anything from a function in python? (What happens if the function reaches the end without `return`ing?)
  27. Django JSONField filtering
  28. What is the best way to access stored procedures in Django’s ORM
  29. How to assign a local file to the FileField in Django?
  30. Writable nested serializer in django-rest-framework?

Leave a Comment