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