Ajustando os retornos do código (#154)

Conforme foi pedido, eu alterei as verificações e retornos da reserva Co-authored-by: GABRIEL ALCANTARA DIAS PRESTES <[email protected]>
Senac-SOR-ADS · Dec 4, 2024 · ff58aa4 · ff58aa4
1 parent 1ce4959
commit ff58aa4
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 20 deletions.
diff --git a/App/controller/reserva.py b/App/controller/reserva.py
@@ -15,7 +15,6 @@ def fazendoReserva(idLogin, dados, diasValidos):
         if diasValidos[diaSemana]:
             Reserva(idLogin, dados['idDocente'], dados['idCurso'], dados['idSala'], diaAtual, dados['inicioCurso'], dados['fimCurso'], 0, dados['observações']).fazer_reserva()
         diaAtual += timedelta(days=1)
-    print('Reserva feita com sucesso!')
     return True
 
 def validarCadastro(dados, diasValidos):
@@ -26,19 +25,14 @@ def validarCadastro(dados, diasValidos):
     diaAtual = diaInicio
     listaDias = []
 
-    if validarDiaSemana(diaInicio, diasValidos):
-        while diaAtual <= diaFim:
-            diaSemana = diaAtual.weekday()
-            validar = Reserva.validar_periodo(dados['idSala'], diaAtual, dados['inicioCurso'], dados['fimCurso'])
-            if diasValidos[diaSemana]:
-                if validar:
-                    listaDias.append(validar[0])
-            diaAtual += timedelta(days=1)
-        if listaDias != []:
-            return listaDias
-        else:
-            return False
-    return True
+    while diaAtual <= diaFim:
+        diaSemana = diaAtual.weekday()
+        validar = Reserva.validar_periodo(dados['idSala'], diaAtual, dados['inicioCurso'], dados['fimCurso'])
+        if diasValidos[diaSemana]:
+            if validar:
+                listaDias.append(validar[0])
+        diaAtual += timedelta(days=1)
+    return listaDias
 
 def trocar_reserva(dados1, dados2):
     if Reserva.atualizar(dados1['idLogin'], dados1['idPessoa'], dados1['idcurso'], dados1['idSala'], dados1['dia'], dados1['inicioCurso'], dados1['fimCurso'], dados1['observações'],  dados1['idReserva']):
@@ -55,7 +49,9 @@ def atualizarReserva(idLogin, idPessoa, idCurso, idSala, dia, hrInicio, hrFim, o
     return False
 
 def validarDiaSemana(dia, diaSemana):
-    dia = datetime.weekday(dia)
+    formatoDia = modificarDataReserva(dia)
+    formatoDia = datetime.strptime(formatoDia, "%d/%m/%Y")
+    dia = datetime.weekday(formatoDia)
     if diaSemana[dia]:
         return True
     print('Selecione o dia da semana certo!')

diff --git a/App/view/reserva.py b/App/view/reserva.py
@@ -10,7 +10,7 @@
 from App.controller.pessoa import buscarPessoas
 from App.controller.sala import listarSala
 from App.controller.utils import modificarData
-from App.controller.reserva import fazendoReserva, validarCadastro
+from App.controller.reserva import fazendoReserva, validarCadastro, validarDiaSemana
 
 
 class ReservaInterface(QWidget):
@@ -81,11 +81,14 @@ def on_btnFazerReserva_clicked(self):
         info = self.getDados()
         idLogin = 8
         diasValidos = (info['seg'], info['ter'], info['qua'], info['qui'], info['sexta'], info['sab'], False)
-        validacao = validarCadastro(info, diasValidos)
-        if type(validacao) == list:
-            print('Não foi possível fazer a reserva, já existe uma reserva nesse horário')
-        elif not validacao:
+        if validarDiaSemana(info['diaInicio'], diasValidos):
+            validacao = validarCadastro(info, diasValidos)
+            if len(validacao):
+                print('Não foi possível fazer a reserva, já existe uma reserva nesse horário')
+                return
             fazendoReserva(idLogin, info, diasValidos)
+            print('Reserva feita com sucesso!')
+        return
 
 
     def setDataMinima(self):