from leilighet import Leilighet

class Bygaard:
    def __init__(self):
        self._leiligheter = []

    # Mulighet 1
    def leggTilLeilighet(self, leilighet):
        self._leiligheter.append(leilighet)

    # Mulighet 2
    def leggTilLeilighet(self, nummer, etasje):
        nyLeilighet = Leilighet(nummer, etasje)
        self._leiligheter.append(nyLeilighet)


    def finnLeilighet(self, nummer):
        for leilighet in leiligheter:
            if leilighet.hentNummer() == nummer:
                return leilighet