def menu_obter():
cod_filme = int(input('Obter filme por código: '))
o = filme.obter_filme(cod_filme)
if o == None:
print('filme não encontrado! ')
else:
imprimir_filme(o)
def test_alterar_filme(self):
elenco = {"Ator": "Beltrano", "Diretor": "Fulano"}
filme.criar_filme(1, "Filme", "terror", elenco, 120, "18 Anos", "UCI")
filme.alterar_filme(1, "NovoFilme", "aventura", {
"Ator": "Ciclano",
"Diretor": "Fulano"
}, 100, "10 Anos", "CineMark")
f = filme.obter_filme(1)
self.assertEqual("NovoFilme", f.nome)
def test_obter_filme(self):
filme.criar_filme(1, "Filme", "terror", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 120, "18 Anos", "UCI")
filme.criar_filme(2, "Filme2", "terror2", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 130, "16 Anos", "UCA")
filmeObtido = filme.obter_filme(1)
self.assertEqual(1, filmeObtido.cod)
def test_excluir_filme(self):
filme.criar_filme(1, "Filme", "terror", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 120, "18 Anos", "UCI")
filme.criar_filme(2, "Filme2", "terror2", {
"Ator": "Beltrano",
"Diretor": "Fulano"
}, 130, "16 Anos", "UCA")
filme.excluir_filme(2)
f = filme.obter_filme(2)
self.assertIsNone(f)
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)