def adicionar(self):
titulo = self.titulo.get()
genero = self.genero.get()
tempo = self.tempo.get()
atores = str(self.atores.get()).split(';')
ano = self.ano.get()
if self.escolha.get() == 'No Início':
a.inserir_inicio(Filme(titulo,genero,tempo,atores,ano))
elif self.escolha.get() == 'No Final':
a.inserir_final(Filme(titulo,genero,tempo,atores,ano))
else:
a.inserir_index(Filme(titulo,genero,tempo,atores,ano),int(self.indice.get()))
def create_avl_tree():
tree = AVLTree()
id = create_random_id()
filme = Filme(id, 'Eu', '2019')
tree.insert(id, filme)
id = create_random_id()
filme = Filme(id, 'Tu', '2000')
tree.insert(id, filme)
id = create_random_id()
filme = Filme(id, 'Ele', '1999')
tree.insert(id, filme)
return tree
def menuInserir(lista):
os.system('cls') or None
print('- Filmes favoritos -')
print('------ Inserir -----\n')
print('1 - Início\n2 - Índice\n3 - Final\n0 - Voltar\n')
op = (input('> '))
os.system('cls') or None
titulo = input('Título:')
genero = input('Gênero:')
tempo = (input('Tempo:'))
atores = input('Atores (separe por ;):')
ano = (input('Ano:'))
ano = ano.split(';')
no = No(Filme(titulo, genero, tempo, atores, ano))
print(op)
if op == 1:
lista.inserir_inicio(no)
elif op == 2:
index = (input('Índice:'))
lista.inserir_index(no, index)
elif op == 3:
lista.inserir_final(no)
elif op == 0:
return
else:
input('Opção inválida! Pressione ENTER')
def adicionar(self):
titulo = self.titulo.get()
genero = self.genero.get()
tempo = self.tempo.get()
atores = str(self.atores.get()).split(';')
ano = self.ano.get()
a.add(Filme(titulo,genero,tempo,atores,ano))
def listarJSON(lista):
#https://www.programiz.com/python-programming/json
with open("filmes.json", "r", encoding='utf-8') as f:
filmes = json.load(f)
for f in filmes:
f = Filme(f['titulo'], f['genero'], f['tempo'], f['atores'], f['ano'])
lista.inserir_final(f)
def create_Filme_and_insert(root):
nome = input("Nome do Filme: ")
global ids
ids += 1
ano = int(input("Ano de lançamento: "))
filme = Filme(nome, ano, ids)
root = root.insert(filme)
return root
print('3 - Buscar filmes pelo ano')
print('4 - Listar filmes em ordem alfabética')
print('5 - Altura da árvore')
print('6 - Exibir a árvore')
print('0 - Sair do programa')
op = int(input('> '))
os.system('cls') or None
if op == 1:
print('Inserir filme')
nome = input('Título do filme: ')
ano = input('Ano de lançamento: ')
id = create_random_id()
filme = Filme(id, nome, ano)
avl.insert(id, filme)
print('\n')
print('Filme inserido\nTítulo: {}\nAno: {}\nID: {}'.format(
nome, ano, id))
elif op == 2:
id = int(input('Id procurado? '))
avl.find_by_id(id)
elif op == 3:
ano = int(input('Ano procurado? '))
avl.find_by_year(ano)
elif op == 4:
filmes = avl.tree_in_order()
class Principal(Emprestimo, Copia):
emp = Emprestimo()
cop = Copia()
soc = Socio()
film = Filme()
def setUp(self):
self.filme = Filme()
class TestFilme(unittest.TestCase):
def setUp(self):
self.filme = Filme()
def testFilmeExiste(self):
assert self.filme != None, "Filme nao existe"
def testInsereCodigo(self):
self.filme.setCod(001)
assert self.filme.cod != None, "Codigo nao existe"
def testInsereTitulo(self):
self.filme.setTitulo("Avatar")
assert self.filme.titulo != None, "Titulo nao existe"
def testInsereDuracao(self):
self.filme.setDuracao("02:00")
assert self.filme.duracao != None, "Duracao nao existe"
def testInsereAno(self):
self.filme.setAno(2010)
assert self.filme.ano != None, "Ano nao existe"
def testInsereGenero(self):
self.filme.setGenero("Ficcao Cientifica")
assert self.filme.genero != None, "Genero nao existe"
def testInsereDiretor(self):
self.filme.setDiretor("James Cameron")
assert self.filme.diretor != None, "Diretor nao existe"
def testInsereArtista1(self):
self.filme.setArtista1("A1")
assert self.filme.artista1 != None, "Artista1 nao existe"
def testInsereArtista2(self):
self.filme.setArtista2("A2")
assert self.filme.artista2 != None, "Artista2 nao existe"
def testImprimeCodigo(self):
self.filme.setCod(001)
cod = self.filme.getCod()
assert cod != None, "Codigo nao existe"
def testImprimeTitulo(self):
self.filme.setTitulo("Avatar")
titulo = self.filme.getTitulo()
assert titulo != None, "Titulo nao existe"
def testImprimeDuracao(self):
self.filme.setDuracao("02:00")
duracao = self.filme.getDuracao()
assert duracao != None, "Duracao nao existe"
def testImprimeAno(self):
self.filme.setAno(2010)
ano = self.filme.getAno()
assert ano != None, "Ano nao existe"
def testImprimeGenero(self):
self.filme.setGenero("Ficcao Cientifica")
genero = self.filme.getGenero()
assert genero != None, "Genero nao existe"
def testImprimeDiretor(self):
self.filme.setDiretor("James Cameron")
diretor = self.filme.getDiretor()
assert diretor != None, "Diretor nao existe"
def testImprimeArtista1(self):
self.filme.setArtista1("A1")
artista1 = self.filme.getArtista1()
assert artista1 != None, "Artista1 nao existe"
def testImprimeArtista2(self):
self.filme.setArtista2("A2")
artista2 = self.filme.getArtista2()
assert artista2 != None, "Artista2 nao existe"
def testRelatorioFilmeGenero(self):
cod = self.filme.getCod()
tit = self.filme.getTitulo()
gen = self.filme.getGenero()
relFilmeGenero = self.filme.relatorioFilmeGenero(cod, tit, gen)
assert relFilmeGenero != None, "Relatorio nao existe"
def testRelatorioFilmeDiretor(self):
cod = self.filme.getCod()
tit = self.filme.getTitulo()
dir = self.filme.getDiretor()
relFilmeDiretor = self.filme.relatorioFilmeDiretor(cod, tit, dir)
assert relFilmeDiretor != None, "Relatorio nao existe"
def testRelatorioFilmeArtista(self):
cod = self.filme.getCod()
tit = self.filme.getTitulo()
art1 = self.filme.getArtista1()
relFilmeArtista = self.filme.relatorioFilmeArtista(cod, tit, art1)
assert relFilmeArtista != None, "Relatorio nao existe"
#adicionar pontos de locador frequente
pontosFrequenciaAlocacao = pontosFrequenciaAlocacao + 1
#adicionar bonus para uma locação de dois dias para lançamentos
if cada.getFilme().getPrecoCodigo(
) == Filme.NOVA_RELEASE and cada.getDiasAlocados() > 1:
pontosFrequenciaAlocacao = pontosFrequenciaAlocacao + 1
#mostrar informacoes para esta locacao
resultado = resultado + ' ' + cada.getFilme().getTitulo(
) + ' ' + str(estaQuantidade) + '\n'
#adicionar rodape do relatorio
resultado = resultado + "Quantia devida é " + str(
totalQuantidade) + "\n"
resultado = resultado + 'Voce ganhou ' + str(
pontosFrequenciaAlocacao) + ' pontos de locacao.'
return resultado
if __name__ == '__main__':
meuCliente = Cliente('Ruben')
print meuCliente.getNome()
fil01 = Filme('Titanic', 2)
alo01 = Alocacao(fil01, 5) #filme, dias locados
alo02 = Alocacao(fil01, 2) #filme, dias locados
alo03 = Alocacao(fil01, 1) #filme, dias locados
meuCliente.adicionarAlocacao(alo01)
meuCliente.adicionarAlocacao(alo02)
meuCliente.adicionarAlocacao(alo03)
print meuCliente.Expressao()
def setUp(self):
self.filme = Filme()
class TestFilme(unittest.TestCase):
def setUp(self):
self.filme = Filme()
def testFilmeExiste(self):
assert self.filme != None, "Filme nao existe"
def testInsereCodigo(self):
self.filme.setCod(001)
assert self.filme.cod != None,"Codigo nao existe"
def testInsereTitulo(self):
self.filme.setTitulo("Avatar")
assert self.filme.titulo != None,"Titulo nao existe"
def testInsereDuracao(self):
self.filme.setDuracao("02:00")
assert self.filme.duracao != None,"Duracao nao existe"
def testInsereAno(self):
self.filme.setAno(2010)
assert self.filme.ano != None,"Ano nao existe"
def testInsereGenero(self):
self.filme.setGenero("Ficcao Cientifica")
assert self.filme.genero != None,"Genero nao existe"
def testInsereDiretor(self):
self.filme.setDiretor("James Cameron")
assert self.filme.diretor != None,"Diretor nao existe"
def testInsereArtista1(self):
self.filme.setArtista1("A1")
assert self.filme.artista1 != None,"Artista1 nao existe"
def testInsereArtista2(self):
self.filme.setArtista2("A2")
assert self.filme.artista2 != None,"Artista2 nao existe"
def testImprimeCodigo(self):
self.filme.setCod(001)
cod=self.filme.getCod()
assert cod != None,"Codigo nao existe"
def testImprimeTitulo(self):
self.filme.setTitulo("Avatar")
titulo=self.filme.getTitulo()
assert titulo != None,"Titulo nao existe"
def testImprimeDuracao(self):
self.filme.setDuracao("02:00")
duracao=self.filme.getDuracao()
assert duracao != None,"Duracao nao existe"
def testImprimeAno(self):
self.filme.setAno(2010)
ano=self.filme.getAno()
assert ano != None,"Ano nao existe"
def testImprimeGenero(self):
self.filme.setGenero("Ficcao Cientifica")
genero=self.filme.getGenero()
assert genero != None,"Genero nao existe"
def testImprimeDiretor(self):
self.filme.setDiretor("James Cameron")
diretor=self.filme.getDiretor()
assert diretor != None,"Diretor nao existe"
def testImprimeArtista1(self):
self.filme.setArtista1("A1")
artista1=self.filme.getArtista1()
assert artista1 != None,"Artista1 nao existe"
def testImprimeArtista2(self):
self.filme.setArtista2("A2")
artista2=self.filme.getArtista2()
assert artista2 != None,"Artista2 nao existe"
def testRelatorioFilmeGenero(self):
cod=self.filme.getCod()
tit=self.filme.getTitulo()
gen=self.filme.getGenero()
relFilmeGenero=self.filme.relatorioFilmeGenero(cod,tit,gen)
assert relFilmeGenero != None,"Relatorio nao existe"
def testRelatorioFilmeDiretor(self):
cod=self.filme.getCod()
tit=self.filme.getTitulo()
dir=self.filme.getDiretor()
relFilmeDiretor=self.filme.relatorioFilmeDiretor(cod,tit,dir)
assert relFilmeDiretor != None,"Relatorio nao existe"
def testRelatorioFilmeArtista(self):
cod=self.filme.getCod()
tit=self.filme.getTitulo()
art1=self.filme.getArtista1()
relFilmeArtista=self.filme.relatorioFilmeArtista(cod,tit,art1)
assert relFilmeArtista != None,"Relatorio nao existe"
def cadastrarFilme(self):
filme = Filme()
while 1:
cont = 0
codEscolhido = raw_input(
"Digite o codigo do filme que voce quer cadastrar:")
filme.setCod(codEscolhido)
codCopEscolhido = raw_input(
"Digite o codigo da copia que voce quer cadastrar:")
self.setCod(codCopEscolhido)
dataAqEscolhido = raw_input(
"Digite a data da aquisicao referente ao filme que voce quer cadastrar:"
)
self.setDataAq(dataAqEscolhido)
codCopia = raw_input(
"Digite o codigo da copia que voce quer cadastrar:")
self.setCod(codCopia)
tituloEscolhido = raw_input(
"Digite o titulo do filme que voce quer cadastrar:")
filme.setTitulo(tituloEscolhido)
duracaoEscolhida = raw_input(
"Digite a duracao do filme que voce quer cadastrar:")
filme.setDuracao(duracaoEscolhida)
anoEscolhido = raw_input(
"Digite o ano do filme que voce quer cadastrar:")
filme.setAno(anoEscolhido)
generoEscolhido = raw_input(
"Digite o genero do filme que voce quer cadastrar:")
filme.setGenero(generoEscolhido)
diretorEscolhido = raw_input(
"Digite o diretor do filme que voce quer cadastrar:")
filme.setDiretor(diretorEscolhido)
artista1Escolhido = raw_input(
"Digite o nome de um ator do filme que voce quer cadastrar:")
filme.setArtista1(artista1Escolhido)
artista2Escolhido = raw_input(
"Digite o nome de um ator do filme que voce quer cadastrar:")
filme.setArtista2(artista2Escolhido)
listaCadastro = [
codEscolhido, codCopEscolhido, dataAqEscolhido, codCopia,
tituloEscolhido, duracaoEscolhida, anoEscolhido,
generoEscolhido, diretorEscolhido, artista1Escolhido,
artista2Escolhido
]
print listaCadastro
++cont
opcaoContinuar = raw_input(
"Voce deseja continuar a cadastrar?(sim/nao)")
if opcaoContinuar == 'nao': break
return listaCadastro
def preencher_filme():
filme = Filme()
filme.set_nome(str(input("Informe o nome do Filme: ")))
filme.set_sinopse(str(input("Informe a sinopse do Filme: ")))
id_genero = str(input("\nInforme o id do Gênero: "))
generos = [i for i in genero_col.find({"_id": ObjectId(id_genero)})]
id_diretor = str(input("\nInforme o id do Diretor: "))
diretores = [i for i in diretor_col.find({"_id": ObjectId(id_diretor)})]
filme.set_genero(generos[0])
filme.set_diretor(diretores[0])
filme.set_avaliacao(Avaliacao().__dict__)
return filme
def cadastrarFilme(self):
filme = Filme()
while 1:
cont = 0
codEscolhido = raw_input("Digite o codigo do filme que voce quer cadastrar:")
filme.setCod(codEscolhido)
codCopEscolhido = raw_input("Digite o codigo da copia que voce quer cadastrar:")
self.setCod(codCopEscolhido)
dataAqEscolhido = raw_input("Digite a data da aquisicao referente ao filme que voce quer cadastrar:")
self.setDataAq(dataAqEscolhido)
codCopia = raw_input("Digite o codigo da copia que voce quer cadastrar:")
self.setCod(codCopia)
tituloEscolhido = raw_input("Digite o titulo do filme que voce quer cadastrar:")
filme.setTitulo(tituloEscolhido)
duracaoEscolhida = raw_input("Digite a duracao do filme que voce quer cadastrar:")
filme.setDuracao(duracaoEscolhida)
anoEscolhido = raw_input("Digite o ano do filme que voce quer cadastrar:")
filme.setAno(anoEscolhido)
generoEscolhido = raw_input("Digite o genero do filme que voce quer cadastrar:")
filme.setGenero(generoEscolhido)
diretorEscolhido = raw_input("Digite o diretor do filme que voce quer cadastrar:")
filme.setDiretor(diretorEscolhido)
artista1Escolhido = raw_input("Digite o nome de um ator do filme que voce quer cadastrar:")
filme.setArtista1(artista1Escolhido)
artista2Escolhido = raw_input("Digite o nome de um ator do filme que voce quer cadastrar:")
filme.setArtista2(artista2Escolhido)
listaCadastro = [
codEscolhido,
codCopEscolhido,
dataAqEscolhido,
codCopia,
tituloEscolhido,
duracaoEscolhida,
anoEscolhido,
generoEscolhido,
diretorEscolhido,
artista1Escolhido,
artista2Escolhido,
]
print listaCadastro
++cont
opcaoContinuar = raw_input("Voce deseja continuar a cadastrar?(sim/nao)")
if opcaoContinuar == "nao":
break
return listaCadastro
