from rom import Rom

class Hus:
    
    def __init__(self):
        self._rom = []
        self._ledige_rom = ["stue", "kj?kken", "bad", "soverom", "soverom"]
        self._total_st?rrelse = 0
    

    def legg_til_rom(self, type, st?rrelse):

        #sjekke om det er 5 rom - hvis det er 5 rom er det fullt
        if len(self._rom) >= 5:
            print("Ikke plass til flere rom!")
        
        #sjekke om romtypen er ledig for ? legge til
        elif type not in self._ledige_rom:
            print("Kan ikke legge til " + type)
            print("Ledige rom: " + str(self._ledige_rom))
        
        #legge til rom:
        else:
            nytt_rom = Rom(st?rrelse, type)
            self._rom.append(nytt_rom)
            self._ledige_rom.remove(type)
            self._total_st?rrelse += st?rrelse
    

    def __str__(self):
        hus = "Informasjon om huset:\n"

        for rom in self._rom:
            hus += f" - {rom.type()}: {rom.st?rrelse()} m2\n"

        hus += "\nTotal husst?rrelse: " + str(self._total_st?rrelse)
        
        return hus