Всем привет! В этой статье я расскажу об инструменте, разработанном мной, который изменяет работу декораторов в Python и делает их более «Питоничными».
Я не буду рассказывать про области применения декораторов. Есть множество статей на эту тему.
Для начала, давайте вспомним: что же такое декораторы в Пайтон.
Если совсем просто, то это удобный способ передать одну функцию в другую и получить третью. В этом определении нет ни одного слова правды, но мы вернёмся к этому позже.
Давайте разбираться!
Как работают декораторы
def decorator_function(wrapped_func):
def wrapper():
print('Входим в функцию-обёртку')
print('Оборачиваемая функция: ', wrapped_func)
print('Выполняем обёрнутую функцию...')
wrapped_func()
print('Выходим из обёртки')
return wrapper
Так выглядит функция-декоратор. Как вы можете увидеть, она принимает в качестве аргумента другую функцию. Затем с этой функцией что-то делают внутри вложенной функции-обёртки и возвращают из декоратора уже обёртку вместо исходной функции.
Здесь декоратор получает функцию hello_world, и подменяет её своей вложенной функцией wrapper.
Вывод:
Входим в функцию-обёртку
Оборачиваемая функция: <function hello_world at 0x0201B2F8>
Выполняем обёрнутую функцию...
Hello world!
Выходим из обёртки
Важно помнить!
Декоратор исполняется только один раз: при объявлении оборачиваемой функции. При дальнейшем вызове функции исполняется только вложенная функция wrapper.
Мы это увидим, если добавим две строчки в наш декоратор:
def decorator_function(wrapped_func):
print('Входим в декоратор')
def wrapper():
...
print('Выходим из декоратора')
return wrapper
Здесь аргумент decorator_text передаётся при декорировании в строке №11 и попадает в функцию fictive, строка №1. Таким образом, появился ещё один уровень вложенности только для того, чтобы принять аргументы декоратора.
Вывод:
Hello, world!
Пойдём дальше. А что, если декоратор может быть, в одних случаях с аргументами, в других - без аргументов? Поехали!
Как Вам код? Вспомним, мантру Питонистов из начала статьи:
Декораторы - это удобный способ передать...
Ничего, на помощь придёт DecoratorHelper! Но, перед этим, ещё пара слов о декораторах.
Мифы декораторов
Декораторы удобны. Думаю, с этим мы уже разобрались.
В декораторы нужно передавать функции. Передавать можно не только функции, но и любые callable объекты. Это такие объекты, у которых определён дандер метод (магический метод) __call__. Этот метод отвечает за операции, которые будут произведены при вызове объекта (когда вы ставите скобочки после имени объекта: object()). Вместо функции может быть метод или класс.
Декораторы - это функции. И опять: это может быть любой callable объект.
Декоратор возвращает функцию. Декоратор может возвращать что угодно. Стоит лишь помнить, что если декоратор возвращает не callable объект, то вызывать его не получится.
DecoratorHelper: решение проблем
Устанавливаем модуль:
pip install DecoratorHelper
Импортируем и используем как декоратор:
from DecoratorHelper import DecoratorHelper
@DecoratorHelper
def hello_world(text):
print(text)
hello_world('Hello, world!')
Что это даёт?
Вы больше не думаете над тем, будут ли аргументы у декоратора. DecoratorHelper думает об этом вместо Вас.
Вы получаете удобный, Питоничный доступ ко всем аргументам, самой функции, к тому, что будет происходить до и после выполнения функции.
В итоге Вы получаете вместо функции объект, который имеет следующие атрибуты:
self.function - оборачиваемая функция
self.decorator_args - аргументы декоратора. Кортеж позиционных аргументов, последний элемент которого - словарь с именованными аргументами.
self.function_args - аргументы функции. Кортеж позиционных аргументов, последний элемент которого - словарь с именованными аргументами.
self.pre_function - то, что будет происходить перед выполнением функции (так можно превратить функцию в коллбэк).
self.post_function - то, что будет происходить после выполнения функции (так можно добавить функции в коллбэк).
Как использовать?
Перепишем приведённый ранее код декоратора, который может принимать/не принимать аргументы:
