from bygaard import Bygaard
from leilighet import Leilighet

def hovedprogram():
    bygaard = Bygaard()

    cnt = 1 #brukes til å lage unike leilighetsnummer
    antallEtasjer = 5
    bygaard.antallEtasjer(antallEtasjer) #legger inn etasjene

    #oppretter leilighetsobjektene ved en dobbel for-loop
    for etasje in range(1, antallEtasjer + 1):
        for i in range(1, 6):
            #leilighet = Leilighet(cnt, etasje)
            #bygaard.leggTilLeilighet(leilighet)
            bygaard.leggTilLeilighet(Leilighet(cnt, etasje))
            cnt += 1

    bygaard.hentInfo()
    bygaard.finnLeilighet(2).hentInfo()



hovedprogram()