Магические методы __add__, __mul__, __sub__ и __truediv__

Магические методы Python позволяют изменять поведение объектов при использовании стандартных операторов. Например, метод __add__ отвечает за поведение объекта при использовании оператора сложения +. 

Пример работы с методом __add__:

class MyString:
    def __init__(self, value):
        self.value = value

    def __add__(self, other):
        return self.value + other

Создав экземпляр класса, можно складывать объекты как строки:

s = MyString("Hello")
print(s + " World")  # Выведет: "Hello World"

Поведение классов при сложении

При создании собственного класса, его экземпляры изначально не могут участвовать в операциях сложения. Если попытаться сложить объект с чем-то, Python выдаст ошибку:

class BankAccount:
    def __init__(self, balance):
        self.balance = balance

acc = BankAccount(100)
print(acc + 50)  # TypeError: unsupported operand type(s) for +: 'BankAccount' and 'int'

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

class BankAccount:
    def __init__(self, balance):
        self.balance = balance

    def __add__(self, other):
        if isinstance(other, (int, float)):
            return BankAccount(self.balance + other)
        raise TypeError("Можно сложить только с числом или другим банковским счетом")

Теперь можно складывать объект с числами:

acc = BankAccount(100)
new_acc = acc + 50
print(new_acc.balance)  # Выведет: 150

Однако если поменять местами операнды (50 + acc), мы снова получим ошибку.

Метод __radd__

Метод __radd__ отвечает за операции сложения, когда экземпляр класса стоит справа от оператора сложения:

class BankAccount:
    def __init__(self, balance):
        self.balance = balance

    def __add__(self, other):
        if isinstance(other, (int, float)):
            return BankAccount(self.balance + other)
        raise TypeError("Можно сложить только с числом")

    def __radd__(self, other):
        return self.__add__(other)

Теперь операция 50 + acc тоже будет работать:

new_acc = 50 + acc
print(new_acc.balance)  # Выведет: 150

Другие магические методы для арифметических операций

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

Отраженные арифметические методы

Кроме обычных магических методов, существуют отраженные версии, которые вызываются, когда левый операнд не реализует соответствующий метод:

Магические методы позволяют настраивать поведение ваших классов при работе с операторами. Правильное использование таких методов, как __add__, __radd__ и других, позволяет создавать более гибкие и адаптируемые классы, взаимодействующие с различными операциями корректно и предсказуемо.

Перейти к следующему шагу