Infotanar.eu - Filmek nyilvántartása szótárral és függvénnyel

Kód

# Filmek nyilvántartása szótárral és függvénnyel
# YT: https://www.youtube.com/watch?v=tX9wsGdpYVk
# Kategória: szótárak

def uj_film_hozzaadasa(filmek):
    cim = input("Add meg a film címét: ")
    rendezo = input("Add meg a rendezőt: ")
    ev = int(input("Add meg az évszámot: "))
    filmek[cim] = {"rendezo": rendezo, "ev": ev}

def film_modositasa(filmek):
    cim = input("Melyik film adatait szeretnéd módosítani? ")
    if cim in filmek:
        uj_rendezo = input("Add meg az új rendezőt: ")
        uj_ev = int(input("Add meg az új évszámot: "))
        filmek[cim] = {"rendezo": uj_rendezo, "ev": uj_ev}
    else:
        print("Ez a film nincs a nyilvántartásban.")

def film_torlese(filmek):
    cim = input("Melyik filmet szeretnéd törölni? ")
    if cim in filmek:
        del filmek[cim]
        print(f"A(z) {cim} című film törölve lett.")
    else:
        print("Ez a film nincs a nyilvántartásban.")

def filmek_listazasa(filmek):
    if not filmek:
        print("Nincs film a nyilvántartásban.")
    else:
        for cim, adatok in filmek.items():
            print(f"{cim} - Rendező: {adatok['rendezo']}, Év: {adatok['ev']}")

def menu_filmek():
    filmek = {}
    while True:
        print("\n1. Új film hozzáadása")
        print("2. Film adatainak módosítása")
        print("3. Film törlése")
        print("4. Filmek listázása")
        print("5. Kilépés")
        valasz = input("Válassz egy lehetőséget (1-5): ")

        if valasz == "1":
            uj_film_hozzaadasa(filmek)
        elif valasz == "2":
            film_modositasa(filmek)
        elif valasz == "3":
            film_torlese(filmek)
        elif valasz == "4":
            filmek_listazasa(filmek)
        elif valasz == "5":
            break
        else:
            print("Érvénytelen választás.")

if __name__ == "__main__":
    menu_filmek()