🎨 Как писать декораторы python и сделать код еще красивее 🖌️

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

def decorator(func):
    def wrapper(*args, **kwargs):
        # выполнять код перед вызовом функции
        result = func(*args, **kwargs)
        # выполнять код после вызова функции
        return result
    return wrapper

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

my_function() # вызов функции с применением декоратора

Как писать декораторы Python?

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

1. Что такое декораторы?

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

    def decorator_function(func):
        def wrapper():
            print("Before function execution")
            func()
            print("After function execution")
        return wrapper
    
    @decorator_function
    def hello():
        print("Hello, World!")
    
    hello()
  

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

2. Как применять декораторы?

Декораторы применяются с помощью символа @ перед определением функции, которую мы хотим декорировать. Это называется "синтаксическим сахаром".

    def decorator_function(func):
        def wrapper():
            print("Before function execution")
            func()
            print("After function execution")
        return wrapper
    
    @decorator_function
    def hello():
        print("Hello, World!")
    
    hello()
  

В этом примере мы применяем декоратор decorator_function к функции hello. Теперь при вызове функции hello будет выводиться дополнительная информация до и после ее выполнения.

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

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

    def decorator_function(arg1, arg2):
        def decorator_inner(func):
            def wrapper():
                # Доступ к аргументам декоратора
                print("Decorator arguments:", arg1, arg2)
                print("Before function execution")
                func()
                print("After function execution")
            return wrapper
        return decorator_inner
    
    @decorator_function("arg1_value", "arg2_value")
    def hello():
        print("Hello, World!")
    
    hello()
  

В этом примере мы определяем дополнительную оберточную функцию decorator_inner, которая принимает аргументы декоратора и возвращает функцию-обертку wrapper. Затем мы применяем декоратор decorator_function к функции hello, передавая ему аргументы arg1_value и arg2_value.

4. Декораторы классов

Кроме того, декораторы могут быть применены к классам. Для этого мы можем определить декоратор, который принимает класс в качестве аргумента и возвращает новый класс.

    def decorator_class(cls):
        class WrapperClass:
            def __init__(self, *args, **kwargs):
                self.wrapped = cls(*args, **kwargs)
            
            def decorated_method(self):
                print("Before method execution")
                self.wrapped.method()
                print("After method execution")
        
        return WrapperClass
    
    @decorator_class
    class MyClass:
        def method(self):
            print("Hello from MyClass")
    
    obj = MyClass()
    obj.decorated_method()
  

В этом примере мы определяем декоратор decorator_class, который принимает класс cls в качестве аргумента и возвращает новый класс WrapperClass. В этом новом классе мы можем добавить дополнительную логику перед и после вызова метода оригинального класса.

5. Декораторы с параметрами

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

    def decorator_factory(arg1, arg2):
        def decorator_function(func):
            def wrapper():
                print("Decorator arguments:", arg1, arg2)
                print("Before function execution")
                func()
                print("After function execution")
            return wrapper
        return decorator_function
    
    @decorator_factory("arg1_value", "arg2_value")
    def hello():
        print("Hello, World!")
    
    hello()
  

В этом примере мы определяем функцию-фабрику decorator_factory, которая принимает аргументы декоратора и возвращает фактический декоратор decorator_function. Затем мы применяем фабрику decorator_factory к функции hello, передавая ей аргументы arg1_value и arg2_value.

6. Стек декораторов

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

    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 hello():
        print("Hello, World!")
    
    hello()
  

В этом примере мы применяем два декоратора decorator1 и decorator2 к функции hello. Сначала будет применен decorator1, а затем decorator2. Порядок декорации важен и может влиять на результат выполнения функции.

Заключение

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

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

Надеюсь, эта статья помогла вам лучше понять, как писать декораторы в Python. Удачи!