Как работать с декораторами классов в Python

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

Освойте Python на курсе от Skypro. Вас ждут 400 часов обучения и практики (достаточно десяти часов в неделю), подготовка проектов для портфолио, индивидуальная проверка домашних заданий и помощь опытных наставников. Получится, даже если у вас нет опыта в IT.

Что такое декоратор класса?

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

Пример простого декоратора класса

Давайте рассмотрим пример простого декоратора класса, который добавляет новый метод к классу:

def add_new_method(class_to_decorate):
    class DecoratedClass(class_to_decorate):
        def new_method(self):
            return "This is a new method"
    return DecoratedClass

@add_new_method
class MyClass:
    def my_method(self):
        return "This is my method"

my_object = MyClass()
print(my_object.my_method())  # вывод: This is my method
print(my_object.new_method())  # вывод: This is a new method

В этом примере мы создали декоратор add_new_method, который принимает класс и возвращает новый класс DecoratedClass с добавленным методом new_method. Затем мы применили этот декоратор к классу MyClass с помощью синтаксиса @add_new_method.

Пример декоратора класса с параметрами

Иногда может потребоваться создать декоратор класса с параметрами. В таком случае, нужно создать функцию, которая возвращает декоратор. Рассмотрим пример:

def add_prefix(prefix):
    def class_decorator(class_to_decorate):
        class DecoratedClass(class_to_decorate):
            def __init__(self, *args, **kwargs):
                super().__init__(*args, **kwargs)
                self.prefix = prefix

            def get_prefixed_name(self):
                return self.prefix + self.name
        return DecoratedClass
    return class_decorator

@add_prefix("Mr. ")
class Person:
    def __init__(self, name):
        self.name = name

person = Person("John")
print(person.get_prefixed_name())  # вывод: Mr. John

В этом примере мы создали функцию add_prefix, которая принимает параметр prefix и возвращает декоратор класса class_decorator. Затем мы применили этот декоратор к классу Person с параметром "Mr. ".

На курсе Skypro «Python-разработчик» освоите основные инструменты программирования, получите опыт на реальных проектах и сможете стартовать в профессии уверенным новичком. Преподаватели — практикующие программисты с большим опытом, а в центре карьеры помогут составить цепляющее резюме и подготовиться к собеседованию.

Заключение

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

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