Adaptando logica para que na legislatura atual não apresentar parlamentares desligados

sapl/api/serializers.py

@@ -70,16 +70,21 @@ class ParlamentarResumeSerializer(serializers.ModelSerializer):
     def check_titular(self,obj):
         is_titular = None
         if Legislatura.objects.exists():
+            try:
                 legislatura = Legislatura.objects.get(id=self.context.get('legislatura'))
-            if not legislatura:
+            except ObjectDoesNotExist:
                 legislatura = Legislatura.objects.first()
+
+        
             mandato = Mandato.objects.filter(
                 parlamentar=obj,
                 data_inicio_mandato__gte=legislatura.data_inicio,
                 data_fim_mandato__lte=legislatura.data_fim
             ).order_by('-data_inicio_mandato').first()
-            
-            is_titular = mandato.titular if mandato else False
+            if mandato:
+                is_titular = 'Sim' if mandato.titular else 'Não'
+            else:
+                is_titular = '-'        
         return is_titular
 
     def check_partido(self,obj):
@@ -91,8 +96,9 @@ class ParlamentarResumeSerializer(serializers.ModelSerializer):
         
         username =  self.context['request'].user.username
         if Legislatura.objects.exists():
+            try:
                 legislatura = Legislatura.objects.get(id=self.context.get('legislatura'))
-            if not legislatura:
+            except ObjectDoesNotExist:
                 legislatura = Legislatura.objects.first()
     
         try:

sapl/api/views.py

@@ -13,6 +13,7 @@ from django_filters.filters import CharFilter
 from django_filters.rest_framework.backends import DjangoFilterBackend
 from django_filters.rest_framework.filterset import FilterSet
 from django_filters.utils import resolve_field
+from django.utils import timezone
 from rest_framework import serializers as rest_serializers
 from rest_framework.decorators import action
 from rest_framework.fields import SerializerMethodField
@@ -351,6 +352,15 @@ class _ParlamentarViewSet:
         Pega lista de parlamentares pelo id da legislatura.
         """
         legislatura = Legislatura.objects.get(pk=kwargs['pk'])
+        data_atual = timezone.now().date()
+
+        if legislatura.data_inicio < data_atual < legislatura.data_fim:
+            mandatos = Mandato.objects.filter(
+                    legislatura=legislatura,
+                    data_inicio_mandato__gte=legislatura.data_inicio,
+                    data_fim_mandato__gte=data_atual,
+            ).order_by('-data_inicio_mandato')
+        else:
             mandatos = Mandato.objects.filter(
                     legislatura=legislatura,
                     data_inicio_mandato__gte=legislatura.data_inicio,
@@ -367,7 +377,7 @@ class _ParlamentarViewSet:
     def search_parlamentares(self,request,*args,**kwargs):
         nome = request.query_params.get('nome_parlamentar','')
         parlamentares = Parlamentar.objects.filter(nome_parlamentar__icontains=nome)
-        serializer_class= ParlamentarResumeSerializer(parlamentares,many=True)
+        serializer_class= ParlamentarResumeSerializer(parlamentares,many=True,context={'request':request})
         return Response(serializer_class.data)
 
 

sapl/templates/parlamentares/parlamentares_list.html

@@ -71,8 +71,7 @@
               <p v-else>Não</p>
             </th>
             <th v-if="!is_pesquisa">
-              <p v-if='[[ parlamentar.titular]]'>Sim</p>
-              <p v-else>Não</p>
+              <p>[[ parlamentar.titular]]</p>
             </th>
           </tr>
           </tbody>