hp-oop-7-8.py

# schools.py

from hogwarts_magic import House


class Hogwarts:
    _houses = {
        "Gryffindor": House(
            "Gryffindor", "Godric Gryffindor", ["scarlet", "gold"], "lion"
        ),
        "Slytherin": House(
            "Slytherin", "Salazar Slytherin", ["green", "silver"], "serpent"
        ),
        "Ravenclaw": House(
            "Ravenclaw", "Rowena Ravenclaw", ["blue", "bronze"], "eagle"
        ),
        "Hufflepuff": House(
            "Hufflepuff", "Helga Hufflepuff", ["yellow", "black"], "badger"
        ),
    }

    @classmethod
    def get_house(cls, house_name):
        return cls._houses.get(house_name)