dataMhs = {}


def tambahDataMhs(npm, nama, jurusan, kelas):
    result = dataMhs.update({npm: [nama, jurusan, kelas]})
    return result


def tampilDataMhs(cetak):

    if(len(dataMhs) != 0):
        if dataMhs.get(cetak, None):
            print("======== [ Cetak Data Mahasiswa ] =======")
            for k, v in (dataMhs[cetak]):
                print(k+": ".join(v))
                print("="*20)
                return cetak
        else:
            print("======== [ Cetak Data Mahasiswa ] =======")
            for k, v in dataMhs.items():
                print(k+": "+str(v))
                print("="*20)
    else:
        print("belum ada data terisi")


def ambilDataMhs():
    print("======== [ List Data Mahasiswa ] =======")
    print("\n".join(v for v in dataMhs.keys()))


while 1:
    print("1. Tambah Data Mahasiswa \n2. Cek Data Mahasiswa\n3. Cetak")
    tanyaMenu = input("Pilih Menu :")
    if(tanyaMenu == "1"):
        data = tambahDataMhs(input("Masukkan NPM: "),
                             input("Masukkan Nama: "), input("Masukkan Jurusan: "), input("Masukkan Kelas "))
    elif(tanyaMenu == "2"):
        ambilDataMhs()
    elif(tanyaMenu == "3"):
        cetak = tampilDataMhs(
            input("Cetak semua atau masukkan salah satu npm : "))
    else:
        print("Tidak ada menu")