"""
1. Skriv en klasse Bygaard. Konstruktøren til
klassen skal definere en tom liste for leiligheter.

2. Skriv en metode som legger til en leilighet i
listen med leiligheter.

3. Skriv en metode som tar i mot et leilighetsnummer og
forsøker å finne og returnere leiligheten med dette nummeret.
Hvis det ikke eksisterer en slik leilighet så skal metoden
returnere None."""

from leilighet import Leilighet

class Bygaard:

    def __init__(self):
        self._leiligheter = [] #tom liste med leiligheter

    #legger til antall etasjer. Hver etasje representereres med en tom liste
    def antallEtasjer(self, antall):
        for etasje in range(antall):
            self._leiligheter.append([])

    #legger til leiligheten etter hvilken etasje de er i (etasjer går fra 1-n) derfor - 1
    def leggTilLeilighet(self, leilighet):
        self._leiligheter[leilighet.hentEtasje() - 1].append(leilighet)

    #henter en lilighet fra bygaarden, hvis den ikke eksisterer blir None returnert
    def finnLeilighet(self, nr):
        for etasje in self._leiligheter:
            for leilighet in etasje:
                if leilighet.hentLeilighetsnummer() == nr:
                    return leilighet
        return None

    #Henter info om alle leilighetene i byggarden
    def hentInfo(self):
        print("Info om alle leiligheter")
        for etasje in range(len(self._leiligheter)):
            print("Du er nå i", etasje + 1)
            for leilighet in self._leiligheter[etasje]:
                leilighet.hentInfo()
                print()