from beboer import Beboer
from sykehjem import Sykehjem

def hovedprogram():
    # Ordbok med beboere
    beboere = {
        "Kari": Beboer("Kari", "brød", "egg", "pølse"),
        "Hans": Beboer("Hans", "toast", "pannekaker", "kjøttboller"),
        "Hanne": Beboer("Hanne", "frokostblanding", "egg", "burger")
    }

    # Oppretter et Sykehjem-objekt som holder på beboerne
    sykehjem = Sykehjem(beboere)

    # while-løkke styrt av brukerinput (programmet kjører så lenge bruker ønsker)
    fortsett = True
    while fortsett:
        valg = input("\nHva vil du gjøre? (skriv kun inn nummer) \n1. Se oversikt over beboere \n2. Se matplan for en beboer \n3. Legge til beboer \n4. Fjerne en beboer \n5. Avslutte \n\n")

        # Sjekker om input er et tall før jeg typekonverterer for å unngå at programmet ikke kjører
        if valg.isdigit():
            valg = int(valg)

        # Se oversikt over beboere
        if valg == 1:
            sykehjem.skriv_beboere()

        # Se matplan for en beboer
        elif valg == 2:
            navn = input("Hvilken beboer vil du se matplanen til? ")
            beboer = sykehjem.finn_beboer(navn)
            if beboer:
                beboer.skriv_matplan()
            else:
                print("Beboeren du har oppgitt er ikke registrert her")

        # Legge til en beboer
        elif valg == 3:
            print("Vennlist oppgi informasjon om beboer og deres matplan\n")
            navn = input("Navn på beboer: ")
            frokost = input("Beboerens frokost: ")
            lunsj = input("Beboerens lunsj: ")
            middag = input("Beboerens middag: ")
            sykehjem.legg_til_beboer(navn, frokost, lunsj, middag)

        # Fjerne en beboer
        elif valg == 4:
            navn = input("Hvilken beboer ønsker du å fjerne? ")
            sykehjem.fjern_beboer(navn)

        # Avslutte
        elif valg == 5:
            fortsett = False

        # Om bruker skrev in ugyldig input
        else:
            print("Ugyldig valg, vennligst skriv inn et tall")


hovedprogram()