from leilighet import Leilighet

class Bygaard:
    def __init__(self, antallEtasjer, antPerEtasje):
        # self._leiligheter skal være en nøstet liste hvor
        # self._leiligheter[0] er første etasje
        # self._leiligheter[1] er andre etasje
        # etc...
        self._leiligheter = []
        self._antallEtasjer = antallEtasjer
        self._antPerEtasje = antPerEtasje

        # Dette gjør man for å kunne legge til leiligheter
        # Man vil få en feilmelding i leggTilLeilighet dersom man ikke har det med
        # fordi man ikke kan hente ut noe på en plass i en tom liste
        for i in range(antallEtasjer):
            self._leiligheter.append([])

    # Oppretter og legger til en leilighet
    def leggTilLeilighet(self, eier, etasje):
        nyLeilighet = Leilighet(eier, etasje)
        # Her kan man gjerne sjekke om etasjen finnes i bygget
        if len(self._leiligheter[etasje]) >= self._antPerEtasje:
            print("Denne etasjen er full!")
        else:
            self._leiligheter[etasje].append(nyLeilighet)

    # Finner og returnerer en leilighet ut ifra navnet på eieren
    def finnLeilighet(self, navn):
        for etasje in self._leiligheter:
            for leilighet in etasje:
                if navn == leilighet.hentEier():
                    return leilighet
        # Dersom leiligheten ikke finnes, returneres None
        return None

    # Skriver ut alle eierne av leilighetene på en ryddig måte
    def skrivUt(self):
        for i in range(self._antallEtasjer):
            print("I etasje", i, "bor:")

            for j in range(self._antPerEtasje):
                leilighet = self._leiligheter[i][j]
                print(leilighet.hentEier())