from person import Person

class Bygning:
    def __init__(self, listeMLister, etasjer, leiligheterIHverEtg):
        self._bygg = listeMLister
        self._etasjer = etasjer
        self._leiligheterIHverEtg = leiligheterIHverEtg

    # Vil skrive ut på følgende format:
    # "I 0 etasje bor: Lise Per Kari Martin"
    def skrivUt(self):
        for i in range(self._etasjer):
            etasje = self._bygg[i]
            print("I", i, "etasje bor: ", end="")

            # Går gjennom etasjen
            for j in range(self._leiligheterIHverEtg):
                person = self._bygg[i][j]
                print(person.hentNavn(), end=" ")

                # Kan skrive ut objektet direkte siden vi har overskrevet __str__
                # print(person, end=" ")

                # Kan også skrives slik:
                # print(self._bygg[i][j].hentNavn(), end=" ")

            # Skriver et ekstra linjeskift mellom hver etasje
            print()


def hovedprogram():
    etasje0 = [Person("Lisa"), Person("Siri"), Person("Silje"), Person("Hanne")]
    etasje1 = [Person("Haaken"), Person("Maren"), Person("Kai"), Person("Jens")]
    etasje2 = [Person("Petter"), Person("Per"), Person("Ole"), Person("Kristoffer")]
    bygg = [etasje0, etasje1, etasje2]

    vaartBygg = Bygning(bygg, len(bygg), len(bygg[0]))
    vaartBygg.skrivUt()

hovedprogram()