from bygg import Bygg 
from by import By
import random

def leggTilBygg(byen, n, farge_liste): #legger til n-bygg i et by-objekt
    for i in range(n):
        f_index = random.randint(0, len(farge_liste)-1) #randint genererer tilfeldige tall
        etasjer = random.randint(1, 100)
        byen.leggTilBygg(Bygg(farge_liste[f_index], etasjer)) #legger til bygg gitt en by

def main():
    farger = ["hvitt", "r?dt", "svart", "gult", "bl?tt"]
    by = By("Oslo") #opprettelse av objekt
    leggTilBygg(by, 1000, farger) 
    by.printInfo()
    print(f"Antall bygg med over 50 etasjer: {by.antallByggOverFemti()}")

main()