from kategori import Kategori
from rett import Rett


class Meny:
    def __init__(self, kategorier):
        self._meny = {}  # str: Kategori
        for kategori in kategorier:
            self._meny[kategori] = self._les_kategori_fil(kategori + ".txt")

    def hent_redusert_meny(self, stoffer_unng?):
        ok_meny = {}  # str: Kategori
        for kategori in self._meny:
            ok_retter = self._meny[kategori].hent_ok_retter(stoffer_unng?)
            if len(ok_retter) > 0:
                ok_kategori = Kategori(kategori, ok_retter)
                ok_meny[kategori] = ok_kategori
        return ok_meny

    # Denne metoden var ikke n?dvendig ? skrive
    def _les_kategori_fil(self, filnavn):
        fil = open(filnavn)
        retter = []
        for linje in fil:
            verdier = linje.strip().split()
            if len(verdier) > 2 and not linje.startswith("#"):
                retter.append(Rett(verdier[0], float(verdier[1]), verdier[2:]))
        fil.close()
        return Kategori(filnavn.rsplit(".", 1)[0], retter)