from rom import Rom

class Hus:
    def __init__(self):
        self._størrelse = 0

        self._romtyper = {"bad":1, "kjøkken":1, "soverom":2, "stue":1}
        
        self._rom = []

    def legg_til_rom(self, rom):
        teller = 0
        if rom.type() in romtyper and len(self._rom) < 5:
            for x in self._rom:
                if rom.type() == x.type():
                    teller += 1
            if teller >= self._romtyper[rom.type()]:
                print("For mange av den type rom")
                return
            self._rom.append(rom)
            self._størrelse += rom.størrelse()
        else:
            print("Feil romtype")

    def __str__(self):
        outputstreng = ""
        for rom in self._rom:
            outputstreng += f"{rom.type()} : {rom.størrelse()}\n"

        outputstreng += f"Den totale størrelsen er: {self._størrelse}"

        return outputstreng