Decorators

Decorators are a powerful feature in Python that allow you to modify the behavior of functions or methods. They are often used to add functionality to functions without modifying their code.

In this lesson, you’ll learn what decorators are, how to use them, and how to create your own decorators.

A decorator is a function that takes another function as input and returns a new function with added functionality. It’s often used with the @decorator_name syntax.

# Basic example of a decorator
def decorator(func):
    def wrapper():
        print("Before the function call")
        func()
        print("After the function call")
    return wrapper

@decorator
def say_hello():
    print("Hello!")

say_hello()

When you use a decorator, Python replaces the original function with the wrapper function returned by the decorator.

# How decorators work
def decorator(func):
    def wrapper():
        print("Wrapper called")
        func()
    return wrapper

@decorator
def greet():
    print("Greetings!")

print(greet.__name__)  # Outputs: wrapper
greet()

Decorators can work with functions that take arguments by using *args and **kwargs in the wrapper function:

# Decorator for functions with arguments
def decorator(func):
    def wrapper(*args, **kwargs):
        print(f"Arguments: {args}, {kwargs}")
        return func(*args, **kwargs)
    return wrapper

@decorator
def add(a, b):
    return a + b

print(add(3, 5))

You can use multiple decorators on a single function. They are applied in the order they appear, from top to bottom:

# Chaining multiple decorators
def decorator1(func):
    def wrapper():
        print("Decorator 1")
        func()
    return wrapper

def decorator2(func):
    def wrapper():
        print("Decorator 2")
        func()
    return wrapper

@decorator1
@decorator2
def greet():
    print("Hello!")

greet()

Parameterized decorators take arguments and allow for more flexible functionality:

# Parameterized decorator
def repeat(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                func(*args, **kwargs)
        return wrapper
    return decorator

@repeat(3)
def say_hello():
    print("Hello!")

say_hello()

Decorators are commonly used for:

• Logging function calls and their arguments.
• Measuring execution time of functions.
• Access control (e.g., checking user permissions).
• Caching results for performance improvements.

Python provides some built-in decorators, including:

• @staticmethod: Defines a static method.
• @classmethod: Defines a class method.
• @property: Defines a property method.

Here are some mistakes to avoid:

• Not preserving the original function’s name and docstring (use functools.wraps).
• Using decorators unnecessarily for simple tasks.
• Forgetting to handle arguments in the wrapper function.

In this lesson, you learned:

• What decorators are and how they work in Python.
• How to create and use basic, parameterized, and chained decorators.
• Common use cases and built-in decorators in Python.

Great job! Let’s move to the next topic.