PARTE 2
main.py
from models.convidado import Convidado
from models.lista_convidados import ListaConvidados
def exibir_apresentacao():
"""Exibe no console uma apresentação amigável ao usuário"""
print ("\nOlá, seja bem-vindo à festa! \n")
def selecionar_opcoes():
"""Permite o usuário escolher uma opção dentre os serviços disponíveis"""
opcao_usuario = input (
"1 - Adicionar convidados" \
"\n2 - Remover convidado" \
"\n3 - Mostrar a lista de convidados" \
"\n4 - Sair; \n Digite o número da opção desejada: ")
return opcao_usuario
def confirmador():
""" Trava o sistema até o usuário digitar uma tecla para que o texto do console não passe muito rápido"""
input("\nAperte qualquer tecla para continuar")
print("\n\n\n")
def main():
while True:
try:
exibir_apresentacao()
opcao_usuario = int (selecionar_opcoes())
if opcao_usuario == 1:
nome_convidado = input("\nDigite o nome do convidado: ").title()
if ListaConvidados.buscar_convidado_por_nome (nome_convidado):
print ("\nEsse nome já está sendo usado. Por favor, tente novamente")
confirmador()
else:
posicao = input ("\nDigite a posição que deseja adicionar o convidado (ou pressione Enter para adicionar no final da lista): ")
if posicao == "":
posicao = None
else:
posicao = int (posicao)
ListaConvidados.adicionar_convidado (Convidado (nome_convidado), posicao)
print (f"\nO(A) convidado(a) {nome_convidado} foi adicionado(a) com sucesso!")
elif opcao_usuario == 2:
nome_convidado = input ("\nDigite o nome do convidado que se deseja excluir: ").title()
convidado_selecionado = ListaConvidados.buscar_convidado_por_nome (nome_convidado)
if not convidado_selecionado:
print ("\nErro, este convidado não existe em nosso banco de dados. Verifique se não houve erros de digitação.")
confirmador()
else:
confirmacao = input (f"Deseja realmente deletar o(a) convidado(a) {convidado_selecionado.nome}? (S / qualquer outra tecla para cancelar): ").upper()
if confirmacao == "S":
ListaConvidados.remover_convidado (convidado_selecionado)
else:
print ("Cancelando a exclusão. Voltando ao Menu Principal")
confirmador()
elif opcao_usuario == 3:
ListaConvidados.exibir_lista_convidados()
confirmador()
elif opcao_usuario == 4:
print ("\n\nObrigado por acessar nosso site. Até mais!")
break
else:
print ("\nOpção inválida. Por favor, tente novamente.")
confirmador()
except ValueError:
print ("\nDigite um número válido. Por favor, tente novamente.")
confirmador()
if __name__ == "__main__":
main()