from leilighet import Leilighet

class Boligblokk:
    def __init__(self, adresse):
        self._adresse = adresse
        self._etasjer = []

        ant_etasjer = 4
        ant_leiligheter_pr_etasje = 4

        for etasje_nr in range(1, ant_etasjer+1):
            etasje = []

            for leilighet_nr in range(1, ant_leiligheter_pr_etasje+1):
                leilighet = Leilighet(self._adresse, etasje_nr, leilighet_nr)
                etasje.append(leilighet)

            self._etasjer.append(etasje)

    def hent_leilighet(self, etasje_nr, leilighet_nr):
        try:
            return self._etasjer[etasje_nr-1][leilighet_nr-1]
        except IndexError:
            print('Leiligheten', etasje_nr, leilighet_nr, 'finnes ikke')

    def hent_adresse(self):
        return self._adresse

    def print_leiligheter(self):
        print(self._adresse)

        for etasje_nr in range(len(self._etasjer)):
            print('Etasje: ', etasje_nr)
            for leilighet in self._etasjer[etasje_nr]:
                print(leilighet)

            print()

    def print_alle_leietakere(self):
        print('Alle som bor i', self._adresse)

        for etasje in self._etasjer:
            for leilighet in etasje:

                if len(leilighet.hent_leietakere()) > 0:
                    print('Leilighet:',leilighet.hent_adresse())
                for person in leilighet.hent_leietakere():
                    print('\t', person)

    def finn_leietaker(self, navn):
        EH_JA_LEIETAKERE = []
        for etasje in self._etasjer:
            for leilighet in etasje:
                for person in leilighet.hent_leietakere():
                    if person.hent_navn() == navn:
                        EH_JA_LEIETAKERE.append(person)

        return EH_JA_LEIETAKERE