super()

В прошлом уроке подкласс либо брал метод базового класса как есть, либо заменял его целиком. Чаще нужно третье: дополнить метод базового класса — сделать всё то же самое и ещё немного. Для этого есть super().

Свой __init__ в подклассе

У выпечки, кроме названия и цены, есть вес. Значит, подклассу Pastry нужен свой __init__ с параметром weight. Но переписывать в нём self.name = name и self.price = price не стоит: это уже умеет __init__ базового класса. Вызов super().__init__(…) выполняет __init__ базового класса для этого же экземпляра:

class Product:
    def __init__(self, name, price):
        self.name = name
        self.price = price

    def __repr__(self):
        return f"{type(self).__name__}({self.name!r}, {self.price})"

    def label(self):
        return f"{self.name} — {self.price} ₽"

    def cost(self, quantity):
        return self.price * quantity


class Pastry(Product):
    def __init__(self, name, price, weight):
        super().__init__(name, price)  # name и price записывает Product
        self.weight = weight


croissant = Pastry("Круассан", 140, 70)
print(croissant.name, croissant.price, croissant.weight)
print(croissant.label())

Круассан 140 70
Круассан — 140 ₽

super() — это «тот же self, но методы ищутся начиная с класса, следующего за тем, где написан super()»; пока у класса один базовый класс, это и есть базовый класс. Поэтому self в скобки не передают: super().__init__(name, price).

class Drink(Product):
    def __init__(self, name, price, volume):
        ...

    def per_100ml(self):
        ...

class Drink(Product):
    def __init__(self, name, price, volume):
        super().__init__(name, price)
        self.volume = volume

    def per_100ml(self):
        return round(self.price / self.volume * 100, 1)

class Cake(Product):
    def __init__(self, name, price, slices):
        self.slices = slices  # забыли super().__init__


cake = Cake("Медовик", 1200, 8)
print(cake.slices)
print(cake.label())

8

AttributeError: 'Cake' object has no attribute 'name'

Расширить, а не скопировать

Так же super() работает в любом методе: подкласс вызывает версию базового класса и что-то добавляет к её результату. Если потом формат строки в Product.label поменяется, подкласс подхватит изменение сам:

class Pastry(Product):
    def __init__(self, name, price, weight):
        super().__init__(name, price)
        self.weight = weight

    def label(self):
        return f"{super().label()}, {self.weight} г"


print(Pastry("Круассан", 140, 70).label())

Круассан — 140 ₽, 70 г

class SeasonalProduct(Product):
    def __init__(self, name, price, until):
        ...

    def label(self):
        ...

class SeasonalProduct(Product):
    def __init__(self, name, price, until):
        super().__init__(name, price)
        self.until = until

    def label(self):
        return f"{super().label()} (до {self.until})"

class WholesaleProduct(Product):
    def cost(self, quantity):
        ...

class WholesaleProduct(Product):
    def cost(self, quantity):
        full = super().cost(quantity)
        if quantity >= 10:
            return round(full * 85 / 100)
        return full

Порядок поиска методов: MRO

Куда именно смотрит super()? У каждого класса есть порядок разрешения методов (method resolution order, MRO) — список классов, в котором Python ищет атрибуты: сам класс, его базовый класс, базовый класс базового и так далее. В конце всегда стоит object — базовый класс всех классов. MRO хранится в атрибуте __mro__:

print([cls.__name__ for cls in Pastry.__mro__])
print([cls.__name__ for cls in WholesaleProduct.__mro__])

['Pastry', 'Product', 'object']
['WholesaleProduct', 'Product', 'object']

super() берёт следующий класс в MRO после текущего. Пока у класса один базовый класс, MRO — просто цепочка вверх. У класса может быть и несколько базовых — множественное наследование (multiple inheritance), — тогда MRO решает, чей метод найдётся первым. В этом курсе оно не понадобится; для начала хватает правила: один базовый класс, остальное — композицией (модуль 6).