How do I create class-based decorators in Python?

How do I create class-based decorators in Python?

To create a class-based decorator in Python, you define a class with a __call__ method. This allows instances of the class to be used as decorators.

Can you show me a simple example?

Sure! Here's a basic example:

class MyDecorator:
    def __call__(self, func):
        def wrapper(*args, **kwargs):
            print('Before calling the function')
            result = func(*args, **kwargs)
            print('After calling the function')
            return result
        return wrapper

@MyDecorator()
def say_hello():
    print('Hello!')

say_hello()

What does the wrapper function do?

The wrapper function adds behavior before and after the original function is called, allowing you to extend its functionality.

Can I pass arguments to the decorator?

Yes, you can modify the __init__ method of the class to accept arguments and use them in the wrapper function.