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

Kód

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

def uj_osztalytars_hozzaadasa(osztaly):
    while True:
        nev = input("Add meg az osztálytárs nevét (vagy 'kilép' a kilépéshez): ")
        if nev.lower() == "kilép":
            break
        kor = input(f"Add meg {nev} korát: ")
        telepules = input(f"Add meg {nev} települését: ")
        osztaly[nev] = {"kor": kor, "telepules": telepules}
        print(f"{nev} hozzáadva az osztályhoz.")

def osztalytars_modositasa(osztaly):
    nev = input("Add meg annak az osztálytársnak a nevét, akit módosítani szeretnél: ")
    if nev in osztaly:
        kor = input(f"Add meg {nev} új korát: ")
        telepules = input(f"Add meg {nev} új települését: ")
        osztaly[nev] = {"kor": kor, "telepules": telepules}
        print(f"{nev} adatai frissítve.")
    else:
        print("Ez az osztálytárs nincs a nyilvántartásban.")

def osztalytars_torlese(osztaly):
    nev = input("Add meg annak az osztálytársnak a nevét, akit törölni szeretnél: ")
    if nev in osztaly:
        del osztaly[nev]
        print(f"{nev} törölve az osztályból.")
    else:
        print("Ez az osztálytárs nincs a nyilvántartásban.")

def osztalytarsak_listazasa(osztaly):
    if not osztaly:
        print("Az osztály még üres.")
    else:
        print("\nOsztálytársak listája:")
        for nev, adatok in osztaly.items():
            print(f"Név: {nev}, Kor: {adatok['kor']}, Település: {adatok['telepules']}")

def menu():
    osztaly = {}
    while True:
        print("\nFőmenü")
        print("1. Új osztálytárs hozzáadása")
        print("2. Osztálytárs módosítása")
        print("3. Osztálytárs törlése")
        print("4. Osztálytársak listázása")
        print("5. Kilépés")
        valasztas = input("Válassz egy lehetőséget (1-5): ")

        if valasztas == "1":
            uj_osztalytars_hozzaadasa(osztaly)
        elif valasztas == "2":
            osztalytars_modositasa(osztaly)
        elif valasztas == "3":
            osztalytars_torlese(osztaly)
        elif valasztas == "4":
            osztalytarsak_listazasa(osztaly)
        elif valasztas == "5":
            print("Kilépés a programból.")
            break
        else:
            print("Érvénytelen választás!")

if __name__ == "__main__":
    menu()