Введение в декораторы

Декораторы — мощная возможность Python, позволяющая изменять поведение функций или методов. Их часто используют для добавления функциональности функциям без изменения их кода.

В этом уроке Вы узнаете, что такое декораторы, как их использовать и как создавать собственные декораторы.

Что такое decorator?

decorator — это функция, которая принимает другую функцию в качестве входных данных и возвращает новую функцию с дополнительной функциональностью. Такой decorator часто используют с синтаксисом @decorator_name.

# 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()

Как работают декораторы

При использовании decorator Python заменяет исходную функцию функцией wrapper, которую возвращает 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()

Использование декораторов с аргументами

Декораторы могут работать с функциями, принимающими аргументы, если в функции wrapper использовать *args и **kwargs:

# 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))

Цепочка нескольких декораторов

Вы можете использовать несколько декораторов для одной функции. Они применяются в том порядке, в котором расположены, — сверху вниз:

# 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 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()

Распространённые случаи использования декораторов

Декораторы часто используют для:

• Журналирования вызовов функций и их аргументов.
• Измерения времени выполнения функций.
• Управления доступом, например проверки разрешений пользователя.
• Кэширования результатов для повышения производительности.

Встроенные декораторы в Python

Python предоставляет несколько встроенных декораторов, в том числе:

• @staticmethod: определяет статический метод.
• @classmethod: определяет метод класса.
• @property: определяет метод-свойство.

Распространённые ошибки при работе с декораторами

Вот некоторые ошибки, которых следует избегать:

• Несохранение имени и строки документации исходной функции (используйте functools.wraps).
• Использование декораторов без необходимости для простых задач.
• Забывание об обработке аргументов в функции wrapper.

Что мы узнали

В этом уроке вы узнали:

• Что такое декораторы и как они работают в Python.
• Как создавать и использовать обычные, параметризованные декораторы и цепочки декораторов.
• О распространённых случаях использования и встроенных декораторах в Python.

Отлично! Перейдём к следующей теме.