from leilighet import Leilighet
from bygaard import Bygaard

gaarden = Bygaard()

antall_etasjer = 5
leil_per_etasje = 3

for etasje in range(1, antall_etasjer + 1):
    for nummerering in range(1, leil_per_etasje + 1):
        leilnr = etasje*100 + nummerering
        ny_leil = Leilighet(leilnr, etasje)
        gaarden.legg_til_leilighet(ny_leil)


nr = 506
leilighet = gaarden.finn_leiliget(nr)
if leilighet is None:
    print("Fant ikke leilighet nummer", nr)
else:
    print("Leilighet nr", leilighet.hent_leilnr(), "ligger i etasje:", 
          leilighet.hent_etasje())