Декораторы с параметрами в Python

Одной из сложных тем для новичков в Python является использование декораторов с параметрами. Например, когда необходимо передать в декоратор определенное значение.

Пример проблемы

Возможно, есть функция, которую нужно декорировать. Декоратор должен уметь работать с определенным параметром. Пример кода:

@my_decorator(5)
def my_function():
    pass

В этом случае, требуется передать значение 5 в декоратор my_decorator. Однако, здесь возникает проблема: как передать этот параметр в декоратор?

Решение проблемы

Для решения этой проблемы можно использовать еще одну функцию, которая обернет основной декоратор. Эта функция будет принимать параметры декоратора и возвращать собственно декоратор.

Вот как это выглядит на примере:

def my_decorator(arg):
    def actual_decorator(function):
        def wrapper():
            # здесь можно использовать arg
            result = function()  # вызов оригинальной функции
            return result
        return wrapper
    return actual_decorator

Теперь my_decorator принимает аргумент и возвращает настоящий декоратор, который, в свою очередь, может быть применен к функции.

С этим подходом, можно использовать декораторы с любыми параметрами. Это делает декораторы в Python невероятно мощным и гибким инструментом.

Тест на профориентацию

За 10 минут узнайте, как ваш опыт пригодиться в IT индустрии

Подробнее