data.py

data = dict()


def add_data():
    while True:
        name = input("Name: (type n to exit):")
        if name == "n":
            break
        while True:
            number = input("Number: (type n to exit):")
            if number == "n":
                break
            data[name] = data[name] + [number] if name in data else [number]


def delete_data():
    while True:
        name = input("Name: (type n to exit):")
        if name == "n":
            break
        if name in data:
            del data[name]
        else:
            print("Name not found")
            continue


def check_if_present():
    while True:
        name = input("Name: (type n to exit):")
        if name == "n":
            break
        if name in data:
            print("Present")
        else:
            print("Not present")


while True:
    i = input(
        "Choose a option: (1)Add, (2)Delete, (3)Check if present,(4) Print all data, (5) exit:")
    if i == "1":
        add_data()
    elif i == "2":
        delete_data()
    elif i == "3":
        check_if_present()
    elif i == "4":
        print(data)
    elif i == "5":
        break
    else:
        print("Invalid input")
        continue