from elev import Elev

class Valghatt:

    def __init__(self):
        self._hus_ordbok = {"Ravnklo":[], "H?sbl?s":[], "Griffing":[], "Smygard":[]}

    def sorter_i_hus(self, elev_liste):
        for elev in elev_liste:
            self._sorter(elev)

    def _sorter(self, elev):
        hus = ""

        if elev.sjekk_egenskap("intelligent"):
            hus = "Ravnklo"

        elif elev.sjekk_egenskap("snill") and not elev.sjekk_egenskap("dristig"):
            hus = "H?sbl?s"

        elif elev.sjekk_egenskap("modig") or elev.sjekk_egenskap("dristig"):
            hus = "Griffing"

        elif elev.sjekk_egenskap("ambisi?s") or elev.sjekk_egenskap("ond") or elev.sjekk_egenskap("slu"):
            hus = "Smygard"

        if hus == "":
            return

        self._hus_ordbok[hus].append(elev)
        elev.sett_hus(hus)

    def hent_hus_ordbok(self):
        return self._hus_ordbok