def test_remover_todos_atores(self):
ator.adicionar_ator(2, "Jason Statham", "Ingles", 49)
ator.adicionar_ator(1, "Chris Hemsworth", "Australiano", 33)
ator.remover_todos_atores()
a = ator.obter_ator(1)
self.assertIsNone(a)
def test_excluir_ator(self):
ator.adicionar_ator(2, "Jason Statham", "Ingles", 49)
ator.adicionar_ator(1, "Chris Hemsworth", "Australiano", 33)
ator.excluir_ator(1)
a = ator.obter_ator(1)
self.assertIsNone(a)
def obter_ator():
print("\nBuscar Ator pelo código. \n")
cod = int(input("Código do Ator: "))
a = ator.obter_ator(cod)
if (a == None):
print("Ator não encontrado")
else:
imprimir_ator(a)
def test_buscar_atores(self):
ator.adicionar_ator(2, "Jason Statham", "Ingles", 49)
ator.adicionar_ator(1, "Chris Hemsworth", "Australiano", 33)
a = ator.obter_ator(1)
self.assertEqual(1, a[0])
self.assertEqual("Chris Hemsworth", a[1])
self.assertEqual("Australiano", a[2])
self.assertEqual(33, a[3])
def menu_alterar():
print('\n Alterar filme')
cod_filme = int(input('Cod. filme: '))
f = filme.obter_filme(cod_filme)
if (f == None):
print("Filme não encontrado")
else:
nome = str(input('Digite o nome do filme :'))
genero = str(input('Digite o genero do filme :'))
menuElenco = 0
listaElenco = []
while menuElenco == 0:
print("\n----------------\n" + "(1) Criar Ator \n" +
"(2) Listar Atores \n" + "(3) Adicionar Ator ao Elenco \n" +
"(4) Ver Elenco \n" + "(0) Terminar Elenco \n" +
"----------------")
opt = int(input("Escolha a opção: "))
if opt == 1:
menu_ator.adicionar_ator()
if opt == 2:
menu_ator.listar_ator()
if opt == 3:
cod_ator = int(input("Insira o codigo do ator: "))
atorElenco = ator.obter_ator(cod_ator)
if atorElenco == None:
print("Ator não encontrado")
else:
print("Deseja adicionar: ", atorElenco[1],
" ao elenco? (S/N)")
rodando = 0
while rodando == 0:
resp = str(input(": "))
if resp == "S" or resp == "s":
nomeAtor = atorElenco[1]
listaElenco.append(nomeAtor)
print("Ator ADICIONADO")
rodando = 1
elif resp == "N" or resp == "n":
print("Ator NAO adicionado")
rodando = 1
else:
print("Insira uma resposta valida")
if opt == 4:
print(listaElenco)
if opt == 0:
menuElenco = 1
dura = int(input('Digite a duracao do filme :'))
clasi = str(input('Digite a classificacao do filme :'))
distr = str(input('Digite a distribuidora do filme :'))
filme.alterar_filme(cod_filme, nome, genero, listaElenco, dura, clasi,
distr)