Modules

Modules are a namespace for packages from which code (variables, functions and classes) can be imported. A module is a file with the .py suffix.

fruits.py

class Fruit:
    def __init__(self, kind, amount=1):
        self.kind = kind
        self.amount = amount

    def add(self, fruit):
        if isinstance(fruit, Fruit) and self.kind == fruit.kind:
            self.amount += fruit.amount
            return True
        return False

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

    def __len__(self):
        return self.amount

    def __str__(self):
        return f"{len(self)} {self.kind}"


class Basket:
    def __init__(self) -> None:
        self.fruits = []

    def add(self, new_fruit):
        for fruit in self.fruits:
            if fruit.add(new_fruit):
                return True

        self.fruits.append(new_fruit)
        return True

    def __len__(self):
        return len(self.fruits)

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

    def __str__(self):
        output = ["The basket contains:"]
        for fruit in self.fruits:
            output.append(str(fruit))
        return "\n".join(output)


class Apple(Fruit):
    def __init__(self):
        super().__init__("apple")


class Pear(Fruit):
    def __init__(self):
        super().__init__("pear")


class Orange(Fruit):
    def __init__(self):
        super().__init__("orange")


class Kiwi(Fruit):
    def __init__(self):
        super().__init__("kiwi")


class BoxOf(Fruit):
    def __init__(self, fruit, amount):
        super().__init__(fruit.kind, amount)

    def __str__(self):
        return f"A box of {len(self)} {self.kind}s"


class Container(Basket):
    def __str__(self):
        output = ["------------------"]
        for fruit in self.fruits:
            output.append(f"|  {str(fruit)}")
        output.append("------------------")
        return "\n".join(output)

example.py

from fruits import Apple, Basket, BoxOf, Container, Kiwi, Orange, Pear

apple = Apple()
pear = Pear()
orange = Orange()
kiwi = Kiwi()
box_of_oranges = BoxOf(Orange(), 6)
container = Container()

basket = Basket()
basket += apple
basket += box_of_oranges
basket += container
basket += pear
basket += kiwi
basket += orange

print(basket)
# prints out:
# The basket contains:
# 1 apple
# A box of 7 oranges
# ------------------
# | A container with
# |  1 pear
# |  1 kiwi
# ------------------

running the code

cd src/in-depth/modules/1
python3 example.py