Input dicionario [RESOLVIDO]

#!/usr/bin/python

# teste.py



def show_menu():

    print "\n1. Mostrar nome"

    print "2. Adicionar Nome"

    print "3. Apagar Nome"

    print "0. Sair\n"

    return input("Entre com a selecao: ")

def show_lista(lista):

    for keys in lista.keys():

        print keys, lista[keys]



def add_lista(lista):

    nome = raw_input("Entre com o nome: ")

    lista[nome] = nome

    return lista



def del_lista(lista):

    nome = raw_input("Entre com o nome a ser deletado: ")

    if nome in lista.keys():

        del lista[nome]

        print "DELETADO!"

        return lista

    else:

        print "Esse nome nao esta no registro!"



lista = {"AB":"CD"}



running = 1



while running:

    selection = show_menu()

    if selection == 1:

        show_lista(lista)

    elif selection == 2:

        lista = add_lista(lista)

    elif selection == 3:

        lista = del_lista(lista)

    elif selection == 0:

        running = 0

    else:

        print "OPCAO INVALIDA"