def cadastrar_veiculo(self, nome, matricula, curso, placa, modelo, categoria): if self.validar_veiculo(placa.upper()) == None: #cadastra o proprietário no dicionario e no BD prop = Proprietario(nome.upper(), matricula.upper(), curso.upper()) c.execute( 'CREATE TABLE IF NOT EXISTS proprietarios(placa TEXT PRIMARY KEY, nome TEXT, matricula TEXT, curso TEXT)' ) c.execute('INSERT INTO proprietarios VALUES (?, ?, ?,?)', (placa.upper(), prop.get_nome(), prop.get_matricula(), prop.get_curso())) con.commit() self.armazenar_proprietarios() #Cadastra o veiculo no dicionario e no array veic = Veiculo(placa.upper(), prop.get_nome(), modelo.upper(), categoria.upper()) c.execute( 'CREATE TABLE IF NOT EXISTS veiculos(placa TEXT PRIMARY KEY, proprietario TEXT, modelo TEXT, categoria TEXT)' ) c.execute('INSERT INTO veiculos VALUES (?, ?, ?,?)', (veic.get_placa(), prop.get_nome(), veic.get_modelo(), veic.get_categoria())) con.commit() self.armazenar_veiculos() else: print('Veículo já está cadastrado!') resposta = input('Deseja cadastrá-lo novamente? (S/N) ') if resposta.upper() == 'S': self.remover_veiculo(placa.upper()) self.cadastrar_veiculo(input('Nome: '), input('Matícula: '), input('Curso: '), input('Placa: '), input('Modelo: '), input('Categoria: '))
def cadastrar_veiculo(self, nome, matricula, curso, placa, modelo, categoria): '''Este método irá instanciar o objeto proprietário e o objeto veículo a partir de uma única solicitacao ao usuario.''' # primeiro valida se o veículo já existe no cadastro. if self.validar_veiculo(placa.upper()) == None: # cadastra o proprietário no dicionario e no BD prop = Proprietario(nome.upper(), matricula.upper(), curso.upper()) self.cadastro_proprietarios[placa.upper()] = prop c.execute( 'CREATE TABLE IF NOT EXISTS proprietarios(placa TEXT PRIMARY KEY, nome TEXT, matricula TEXT, curso TEXT)' ) c.execute('INSERT INTO proprietarios VALUES (?, ?, ?,?)', (placa.upper(), prop.get_nome(), prop.get_matricula(), prop.get_curso())) con.commit() # Cadastra o veiculo no array e no BD veic = Veiculo(placa.upper(), prop.get_nome(), modelo.upper(), categoria.upper()) try: c.execute( 'CREATE TABLE IF NOT EXISTS veiculos(placa TEXT PRIMARY KEY, proprietario TEXT, modelo TEXT, categoria TEXT)' ) c.execute('INSERT INTO veiculos VALUES (?, ?, ?,?)', (veic.get_placa(), prop.get_nome(), veic.get_modelo(), veic.get_categoria())) con.commit() self.cadastro_veiculos.append(veic) print("Veículo cadastrado com sucesso") except: print("Erro ao cadastrar veículo!") else: print('Veículo já está cadastrado!') resposta = input('Deseja cadastrá-lo novamente? (S/N) ') if resposta.upper() == 'S': self.remover_veiculo(placa.upper()) self.cadastrar_veiculo(input('Nome: '), input('Matícula: '), input('Curso: '), input('Placa: '), input('Modelo: '), input('Categoria: '))