Код для тестирования цепочки декораторов

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

Метод 1: использование нескольких декораторов

def decorator1(func):
    def wrapper(*args, kwargs):
        print("Decorator 1")
        return func(*args, kwargs)
    return wrapper
def decorator2(func):
    def wrapper(*args, kwargs):
        print("Decorator 2")
        return func(*args, kwargs)
    return wrapper
@decorator1
@decorator2
def my_function():
    print("Original function")
my_function()

Выход:

Decorator 1
Decorator 2
Original function

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

Метод 2: использование классов декораторов

class Decorator1:
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, kwargs):
        print("Decorator 1")
        return self.func(*args, kwargs)
class Decorator2:
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, kwargs):
        print("Decorator 2")
        return self.func(*args, kwargs)
@Decorator1
@Decorator2
def my_function():
    print("Original function")
my_function()

Выход:

Decorator 1
Decorator 2
Original function

В этом методе мы определяем классы декораторов Decorator1и Decorator2. Каждый класс-декоратор реализует метод __call__, который позволяет вызывать экземпляры класса, как если бы они были функциями. Применяя декораторы с использованием синтаксиса @в порядке @Decorator1и @Decorator2, функция my_functionдекорируется обоими классы декораторов. При вызове my_functionдекораторы выполняются в том порядке, в котором они были применены.

Метод 3: использование декораторов высшего порядка

def decorator1(arg):
    def wrapper(func):
        def inner_wrapper(*args, kwargs):
            print("Decorator 1 with arg:", arg)
            return func(*args, kwargs)
        return inner_wrapper
    return wrapper
def decorator2(arg):
    def wrapper(func):
        def inner_wrapper(*args, kwargs):
            print("Decorator 2 with arg:", arg)
            return func(*args, kwargs)
        return inner_wrapper
    return wrapper
@decorator1("arg1")
@decorator2("arg2")
def my_function():
    print("Original function")
my_function()

Выход:

Decorator 1 with arg: arg1
Decorator 2 with arg: arg2
Original function

В этом методе мы определяем декораторы более высокого порядка decorator1и decorator2, которые принимают аргументы. Декораторы возвращают функции-обертки, которые принимают исходную функцию в качестве параметра. Применяя декораторы с использованием синтаксиса @с нужными аргументами, функция my_functionдекорируется обоими декораторами. При вызове my_functionдекораторы выполняются в том порядке, в котором они были применены, с указанными аргументами.