from Leilighet import Leilighet
import random

class Blokk:
    def __init__(self, etasjer, leiligheter):
        self._etasjer = etasjer
        self._leiligheter = leiligheter
        self._alleLeiligheter = []
    
    def marakeshBlokk(self): #Metode som fyller inn _alleLeiligheter med etasjer (med leiligheter)
        for i in range(self._etasjer):
            etasje = [] #For hver etasje, må vi opprette en ny liste der leilighetene skal settes inn
            for j in range(self._leiligheter):
                etasjenr = i
                leilighetnr = j
                antBeboere = random.randint(0, 40) #genererer et tilfeldig heltall fom 0, tom 40.

                nyLeilighet = Leilighet(etasjenr, leilighetnr, antBeboere)

                etasje.append(nyLeilighet) #Legger den nye leiligheten til i etasje-listen

            self._alleLeiligheter.append(etasje) #Legger etasje-listen inn i _alleLeiligheter (etter at den innerste løkken er ferdig)
    
    def printBlokk(self):
        for etasje in self._alleLeiligheter[::-1]: #går gjennom etasjene baklengs, slik at den øverste etasjen blir printet ut først!
            for leilighet in etasje:
                #print(etasje.index(leilighet)) #gir indexen leilighet ligger i i listen etasje (var spørsmål om dette i timen)
                print("[", leilighet.hentAllInformasjon(), "]", end = " ")
            print() #tar med et linjeskift for at ikke alle leiligheter skal bli printet på samme linje