How to do a conditional decorator in python?

by

Decorators are simply callables that return a replacement, optionally the same function, a wrapper, or something completely different. As such, you could create a conditional decorator:

def conditional_decorator(dec, condition):
    def decorator(func):
        if not condition:
            # Return the function unchanged, not decorated.
            return func
        return dec(func)
    return decorator

Now you can use it like this:

@conditional_decorator(timeit, doing_performance_analysis)
def foo():
    time.sleep(2)

The decorator could also be a class:

class conditional_decorator(object):
    def __init__(self, dec, condition):
        self.decorator = dec
        self.condition = condition

    def __call__(self, func):
        if not self.condition:
            # Return the function unchanged, not decorated.
            return func
        return self.decorator(func)

Here the __call__ method plays the same role as the returned decorator() nested function in the first example, and the closed-over dec and condition parameters here are stored as arguments on the instance until the decorator is applied.

More Related Contents:

  1. How to make function decorators and chain them together?
  2. How does the @property decorator work in Python?
  3. Decorator execution order
  4. How to inject variable into scope with a decorator?
  5. How do I make function decorators and chain them together?
  6. How to make function decorators and chain them together in Python?
  7. Python decorator? – can someone please explain this? [duplicate]
  8. How to create a custom decorator in Django?
  9. Decorator error: NoneType object is not callable
  10. How to skip a pytest using an external fixture?
  11. Creating a singleton in Python
  12. Conditional Replace Pandas
  13. How to get method parameter names?
  14. Calling class staticmethod within the class body?
  15. How do I pass extra arguments to a Python decorator?
  16. How can I decorate an instance method with a decorator class?
  17. Attaching a decorator to all functions within a class
  18. Conditional with statement in Python
  19. Introspection to get decorator names on a method?
  20. What are some common uses for Python decorators? [closed]
  21. What is the best way to do automatic attribute assignment in Python, and is it a good idea?
  22. How do I assign values based on multiple conditions for existing columns?
  23. Python decorator as a staticmethod
  24. Python functools.wraps equivalent for classes
  25. How to auto register a class when it’s defined
  26. Assignment Condition in Python While Loop
  27. raise statement on a conditional expression
  28. Auto-register class methods using decorator
  29. Python pandas – new column’s value if the item is in the list
  30. Python decorator, self is mixed up [duplicate]

Leave a Comment