class Sortinghat:

    def __init__(self):
        self._hus = [[], [], [], []]
        self._keys = ["Ravenclaw", "Hufflepuff", "Gryffindor", "Slytherin"]

    def lagHusOrdbok(self, elevListe):

        husOrdbok = {}

        for elevIndeks in range(len(elevListe)):
            elev = elevListe[elevIndeks]
            hus = self.sorter(elev)

            if hus == self._keys[0]:
                self._hus[0].append(elev)

            elif hus == self._keys[1]:
                self._hus[1].append(elev)

            elif hus == self._keys[2]:
                self._hus[2].append(elev)

            else:
                self._hus[3].append(elev)

        indeks = 0

        for key in self._keys: 
            husOrdbok[key] = self._hus[0]
            indeks += 1

        return husOrdbok


    def sorter(self, elev):

        hus = ""

        if elev.sjekkEgenskap("intelligent"):
            hus = self._keys[0]

        elif elev.sjekkEgenskap("snill") and elev.sjekkEgenskap("dristig") == False:
            hus = self._keys[1]

        elif elev.sjekkEgenskap("modig") or elev.sjekkEgenskap("dristig"):
            hus = self._keys[2]

        elif elev.sjekkEgenskap("ambisiøs") or elev.sjekkEgenskap("ond") or elev.sjekkEgenskap("slu"):
            hus = self._keys[3]

        elev.settHus(hus)

        return hus