Refatoração das condicionais de informações

sapl/api/views.py

@@ -60,25 +60,28 @@ def get_mesa_diretora(request):
 
     legislatura = request.GET.get('legislatura')
     if legislatura:
-        kwargs['legislatura_id'] = legislatura
+        kwargs['legislatura_id'] = legislatura        
     
     sessao = request.GET.get('sessao')
     if sessao:
         kwargs['id'] = sessao
 
-    sessao_legislativa = SessaoLegislativa.objects.filter(**kwargs).order_by('-data_inicio').first()
+    sessao_legislativa = SessaoLegislativa.objects.select_related('legislatura').filter(**kwargs).order_by('-data_inicio').first()
 
-    query_legislatura = Legislatura.objects.filter(id=legislatura).order_by('-data_inicio').first()
+    if sessao_legislativa is None:
+        logger.error("Sessão ou legislatura não encontrada!")
+        return JsonResponse({})
 
     composicao_mesa = ComposicaoMesa.objects.select_related('parlamentar', 'cargo').all().filter(
         sessao_legislativa=sessao_legislativa).order_by('cargo_id')
-   
-    if sessao_legislativa is None or legislatura is None or composicao_mesa is None:
-        logger.error("Sessão, legislatura ou mesa não encontrada(s)!")
+
+    if composicao_mesa is None:
+        logger.error("Nenhuma mesa não encontrada!")
         return JsonResponse({})
+   
 
 
-    mesa_diretora = [{'legislatura_id':query_legislatura.id,'legislatura':str(query_legislatura), 
+    mesa_diretora = [{'legislatura_id':sessao_legislativa.legislatura.id,'legislatura':str(sessao_legislativa.legislatura), 
                     'sessao_legislativa_id':sessao_legislativa.id,'sessao_legislativa':str(sessao_legislativa),
                     'parlamentar_id':i[0], 'parlamentar_nome':i[1], 'cargo_id':i[2], 'cargo_descricao':i[3]} 
                     for i in composicao_mesa.values_list('parlamentar_id', 'parlamentar__nome_parlamentar',