def all_empresas():
resultado = search_sql("SELECT * FROM Empresa")
if len(resultado) == 0:
return None
else:
empresas = []
for i in range(len(resultado)):
empresa = Empresa()
empresa.set_cod_emp(resultado[int(i)]['CodEmp'])
empresa.set_nome(resultado[int(i)]['Nome'])
empresas.append(empresa)
return empresas
def buscar_empresa_nome(nome):
resultado = search_sql(
"SELECT * FROM Empresa WHERE Nome = {}".format(nome))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
empresa = Empresa()
empresa.set_cod_emp(resultado[0]['CodEmp'])
empresa.set_nome(resultado[0]['Nome'])
return empresa
def all_municipios():
resultado = search_sql("SELECT * FROM Municipio")
if len(resultado) == 0:
return None
else:
municipios = []
for i in range(len(resultado)):
municipio = Municipio()
municipio.set_cod_mun(resultado[int(i)]['CodMunicipio'])
municipio.set_nome(resultado[int(i)]['Nome'])
municipios.append(municipio)
return municipios
def buscar_municipio_nome(nome):
resultado = search_sql(
"SELECT * FROM Municipio WHERE Nome = {}".format(nome))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
municipio = Municipio()
municipio.set_cod_mun(resultado[0][1])
municipio.set_nome(resultado[0][1])
return municipio
def all_viagens():
resultado = search_sql("SELECT * FROM Viagem")
if len(resultado) == 0:
return None
else:
viagens = []
for i in range(len(resultado)):
viagem = Viagem()
viagem.set_cod_traj(resultado[int(i)]['CodTrajetoHorario'])
viagem.set_data(resultado[int(i)]['Data'])
viagem.set_assentos(resultado[int(i)]['TotalAssentosVeiculo'])
viagens.append(viagem)
return viagens
def buscar_municipio(cod_mun):
resultado = search_sql(
"SELECT * FROM Municipio WHERE CodMunicipio = {}".format(cod_mun))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
municipio = Municipio()
municipio.set_cod_mun(cod_mun)
municipio.set_nome(resultado[0]['Nome'])
return municipio
def buscar_empresa(cod_emp):
resultado = search_sql(
"SELECT * FROM Empresa WHERE CodEmp = {}".format(cod_emp))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
# print((resultado[0]))
empresa = Empresa()
empresa.set_cod_emp(cod_emp)
empresa.set_nome(resultado[0]['Nome'])
return empresa
def all_reservas():
resultado = search_sql("SELECT * FROM ReservaAssento")
if len(resultado) == 0:
return None
else:
reservas = []
for i in range(len(resultado)):
reserva = Reserva()
reserva.set_cod_traj(resultado[int(i)]['CodTrajetoHorario'])
reserva.set_data(resultado[int(i)]['Data'])
reserva.set_numero_assento(resultado[int(i)]['NumeroAssento'])
reserva.set_disponibilidade(resultado[int(i)]['Livre'])
reservas.append(reserva)
return reservas
def buscar_empresas(nome):
resultado = search_sql(
"SELECT * FROM Empresa WHERE Nome LIKE %{}%".format(nome))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
empresas = []
for coluna in resultado:
empresa = Empresa()
empresa.set_cod_emp(coluna[0])
empresa.set_nome(coluna[1])
empresas.append(empresa)
return empresas
def buscar_viagem(cod_traj, data):
resultado = search_sql(
"SELECT * FROM Viagem WHERE CodTrajetoHorario = {} AND Data = \"{}\"".
format(cod_traj, data))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
viagem = Viagem()
viagem.set_cod_traj(cod_traj)
viagem.set_data(resultado[0]['Data'])
viagem.set_assentos(resultado[0]['TotalAssentosVeiculo'])
return viagem
def buscar_municipios(nome):
resultado = search_sql(
"SELECT * FROM Empresa WHERE Nome LIKE %{}%".format(nome))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
municipios = []
for coluna in resultado:
municipio = Municipio()
municipio.set_cod_mun(coluna[0])
municipio.set_nome(coluna[1])
municipios.append(municipio)
return municipios
def buscar_reserva(cod_traj, data, numero_assento):
resultado = search_sql(
"SELECT * FROM ReservaAssento WHERE CodTrajetoHorario = {} AND Data = \"{}\""
" AND NumeroAssento = {}".format(cod_traj, data, numero_assento))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
reserva = Reserva()
reserva.set_cod_traj(cod_traj)
reserva.set_data(resultado[0]['Data'])
reserva.set_numero_assento(resultado[0]['NumeroAssento'])
reserva.set_disponibilidade(resultado[0]['Livre'])
return reserva
def buscar_viagens(cod_traj):
resultado = search_sql(
"SELECT * FROM Viagem WHERE CodTrajetoHorario LIKE %{}%".format(
cod_traj))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
viagens = []
for coluna in resultado:
viagem = Viagem()
viagem.set_cod_traj(coluna[0])
viagem.set_data(coluna[1])
viagem.set_assentos(coluna[2])
viagens.append(viagem)
return viagens
def all_trajetos():
resultado = search_sql("SELECT * FROM Trajeto_Horario")
if len(resultado) == 0:
return None
else:
trajetos = []
for i in range(len(resultado)):
trajeto = Trajeto()
trajeto.set_cod_traj(resultado[int(i)]['CodTrajetoHorario'])
trajeto.set_cod_empresa(resultado[int(i)]['CodEmpresa'])
trajeto.set_cod_municipio_origem(resultado[int(i)]['CodMunicio'])
trajeto.set_cod_municipio_destino(
resultado[int(i)]['CodMunicipioDest'])
trajeto.set_dia(resultado[int(i)]['DiaSemana'])
trajeto.set_horario_partida(resultado[int(i)]['HorarioPartida'])
trajeto.set_horario_chegada(resultado[int(i)]['HorarioChegada'])
trajetos.append(trajeto)
return trajetos
def buscar_trajetos():
resultado = search_sql("SELECT * FROM Trajeto_Horario")
if not resultado:
return None
if len(resultado) == 0:
return None
else:
trajetos = []
for coluna in resultado:
trajeto = Trajeto()
trajeto.set_cod_traj(coluna[0])
trajeto.set_cod_empresa(coluna[1])
trajeto.set_cod_municipio_origem(coluna[2])
trajeto.set_cod_municipio_destino(coluna[3])
trajeto.set_dia(coluna[4])
trajeto.set_horario_partida(coluna[5])
trajeto.set_horario_chegada(coluna[6])
trajetos.append(trajeto)
return trajetos
def buscar_trajeto(cod_traj):
resultado = search_sql(
"SELECT * FROM Trajeto_Horario WHERE CodTrajetoHorario = {}".format(
cod_traj))
if not resultado:
return None
if len(resultado) == 0:
return None
else:
coluna = resultado[0]
trajeto = Trajeto()
trajeto.set_cod_traj(cod_traj)
trajeto.set_cod_empresa(coluna['CodEmpresa'])
trajeto.set_cod_municipio_origem(coluna['CodMunicio'])
trajeto.set_cod_municipio_destino(coluna['CodMunicipioDest'])
trajeto.set_dia(coluna['DiaSemana'])
trajeto.set_horario_partida(coluna['HorarioPartida'])
trajeto.set_horario_chegada(coluna['HorarioChegada'])
return trajeto
