🎨 Как создать свой декоратор python? Изучаем простые шаги

Для создания собственного декоратора в Python следуйте этим шагам:

1. Определите функцию декоратора, которая будет принимать функцию в качестве аргумента.
2. Внутри декоратора определите новую функцию, которая будет выполняться до и после вызова оригинальной функции.
3. Возвращайте новую функцию из декоратора.
4. Примените декоратор к целевой функции, используя символ "@" и название декоратора перед определением функции.

def my_decorator(func):
    def wrapper():
        # Дополнительные действия до вызова функции
        print("Выполняется код до вызова функции")
        func()
        # Дополнительные действия после вызова функции
        print("Выполняется код после вызова функции")
    return wrapper

@my_decorator
def my_function():
    print("Оригинальная функция")

my_function()
  

Как создать свой декоратор Python

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

Синтаксис декоратора

Декораторы в Python представлены символом "@", за которым идет имя декоратора. Они применяются непосредственно перед определением функции или метода. Вот пример:

@decorator
def my_function():
    # Код функции
    pass
    

В этом примере "decorator" - это имя декоратора, который будет применен к функции "my_function()".

Как создать простой декоратор

Рассмотрим пример создания простого декоратора, который будет выводить сообщение до и после выполнения функции:

def decorator_function(func):
    def wrapper():
        print("Начало выполнения функции")
        func()
        print("Конец выполнения функции")
    return wrapper

@decorator_function
def my_function():
    print("Привет, мир!")

my_function()
    

В этом примере "decorator_function" - это функция-декоратор. Она принимает функцию "func" в качестве аргумента и возвращает внутреннюю функцию "wrapper", которая добавляет дополнительное поведение до и после выполнения функции.

Затем мы применяем декоратор к функции "my_function" с помощью символа "@". При вызове "my_function" будет выводиться сообщение перед и после выполнения основного кода функции.

Декораторы с аргументами

Иногда требуется передать аргументы в декоратор. Для этого можно использовать функцию-фабрику:

def decorator_factory(arg1, arg2):
    def decorator_function(func):
        def wrapper():
            print("Аргументы декоратора:", arg1, arg2)
            func()
        return wrapper
    return decorator_function

@decorator_factory("Аргумент 1", "Аргумент 2")
def my_function():
    print("Привет, мир!")

my_function()
    

В этом примере "decorator_factory" - это функция-фабрика, которая принимает аргументы "arg1" и "arg2" и возвращает функцию-декоратор "decorator_function". Внутри "decorator_function" мы получаем доступ к переданным аргументам и используем их внутри внутренней функции "wrapper".

Несколько декораторов

Python позволяет применять несколько декораторов к одной функции, применяя их последовательно сверху вниз:

def decorator1(func):
    def wrapper():
        print("Декоратор 1")
        func()
    return wrapper

def decorator2(func):
    def wrapper():
        print("Декоратор 2")
        func()
    return wrapper

@decorator1
@decorator2
def my_function():
    print("Привет, мир!")

my_function()
    

В этом примере мы создали два декоратора "decorator1" и "decorator2", каждый из которых добавляет свое дополнительное поведение к функции "my_function". При вызове "my_function" каждый декоратор будет применяться последовательно.

Заключение

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

Использование декораторов помогает сделать код более модульным, позволяет повторно использовать код и улучшает его читаемость. Знание создания собственных декораторов в Python открывает новые возможности для разработки.