Декораторы в Python

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

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

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

import time

def measure_time(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        execution_time = end_time - start_time
        print(f"Время выполнения функции {func.__name__}: {execution_time} секунд.")
        return result
    return wrapper

@measure_time
def slow_function():
    time.sleep(2)
    print("Функция slow_function выполнена.")

slow_function()

В этом примере декоратор measure_time оборачивает функцию slow_function и добавляет логику замера времени выполнения. При вызове функции slow_function, она обернута в декоратор, который выводит время выполнения и возвращает результат. Таким образом, мы можем получить информацию о времени, затраченном на выполнение функции без необходимости изменения самой функции.

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

def uppercase(func):
    def wrapper():
        result = func()
        return result.upper()
    return wrapper

@uppercase
def get_greeting():
    return "Привет, мир!"

print(get_greeting())  # Выводит: "ПРИВЕТ, МИР!"

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

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

def log_init(func):
    def wrapper(self, *args, **kwargs):
        print(f"Инициализация класса {self.__class__.__name__}")
        return func(self, *args, **kwargs)
    return wrapper

class Person:
    @log_init
    def __init__(self, name, age):
        self.name = name
        self.age = age

person = Person("Иван", 30)
print(person.name)  # Выводит: "Иван"
print(person.age)  # Выводит: 30

В этом примере декоратор log_init применяется к конструктору класса Person. При инициализации объекта класса Person выводится сообщение о его создании. Таким образом, мы можем добавить дополнительную логику к конструктору класса без изменения его исходного кода.

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