Example #1
0
    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: '))