Commit zoado de última hora

Author: maribedran

dados_brasil_io/urls.py

@@ -16,9 +16,12 @@
 from django.contrib import admin
 from django.urls import path
 
-from politicos.views import DeputadoListView
+from empresas.views import EmpresaDetailView
+from politicos.views import DeputadoListView, DeputadoGastosView
 
 urlpatterns = [
     path('admin/', admin.site.urls),
     path('deputados/', DeputadoListView.as_view(), name='deputados'),
+    path('deputados-gastos', DeputadoGastosView.as_view(), name='deputados-gastos'),
+    path('empresa/<pk>/', EmpresaDetailView.as_view(), name='empresa'),
 ]

empresas/models.py

@@ -1,5 +1,5 @@
 from django.db import models
-from django.db.models import Case, Count, F, OuterRef, Q, Subquery, Sum, Value, When
+from django.db.models import Case, Count, Exists, F, OuterRef, Q, Subquery, Sum, Value, When
 from django.db.models.functions import Coalesce
 
 

empresas/views.py

@@ -1,3 +1,18 @@
-from django.shortcuts import render
+from rest_framework import generics, serializers
+from rest_framework.response import Response
+
+from empresas.models import Empresa, Sociedade
+
+
+class EmpresaSerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = Empresa
+        fields = ['cnpj', 'nome', 'sociedades']
+        depth = 10
+
+
+class EmpresaDetailView(generics.RetrieveAPIView):
+    serializer_class = EmpresaSerializer
+    queryset = Empresa.objects.all()
 
-# Create your views here.

politicos/filters.py

@@ -1,3 +1,4 @@
+from datetime import date
 from django_filters import rest_framework as filters
 
 from politicos.models import Deputado
@@ -6,8 +7,7 @@
 class DeputadoFilter(filters.FilterSet):
     nome = filters.CharFilter(field_name='nome', lookup_expr='icontains')
     exibir_gasto_mensal = filters.BooleanFilter(method='filter_exibir_gasto_mensal')
-    gastos_mes = filters.NumberFilter(method='add_prefetch_gastos')
-    gastos_ano = filters.NumberFilter(method='add_prefetch_gastos')
+    fornecedor_gasto = filters.CharFilter(field_name='gastos__empresa__nome', lookup_expr='icontains')
 
     class Meta:
         model = Deputado
@@ -20,10 +20,23 @@ def filter_exibir_gasto_mensal(self, queryset, name, value):
             queryset = queryset.annotate_gasto_mensal_por_deputado()
         return queryset
 
+
+class DeputadoGastosFilter(filters.FilterSet):
+    nome = filters.CharFilter(field_name='nome', lookup_expr='icontains')
+    gastos_mes = filters.NumberFilter(method='add_prefetch_gastos')
+    gastos_ano = filters.NumberFilter(method='add_prefetch_gastos')
+
+    class Meta:
+        model = Deputado
+        fields = [
+            'nome', 'partido', 'uf', 'id_legislatura',
+        ]
+
     def add_prefetch_gastos(self, queryset, name, value):
         filtros = self.data.dict()
-        mes = filtros.get('gastos_mes')
-        ano = filtros.get('gastos_ano')
+        hoje = date.today()
+        mes = filtros.get('gastos_mes', hoje.month)
+        ano = filtros.get('gastos_ano', hoje.year)
         if not hasattr(self, '_prefetch_added') and mes and ano:
             queryset = queryset.prefetch_gastos(mes=mes, ano=ano)
             self._prefetch_added = True

politicos/models.py

@@ -1,5 +1,6 @@
 from django.db import models
-from django.db.models import Avg, F, FilteredRelation, Prefetch, Q, Sum
+from django.db.models import Avg, Case, Count, Exists, F, FilteredRelation, OuterRef, Prefetch, Q, Sum, Value, When
+from django.db.models.functions import Coalesce
 
 
 class DeputadoQuerySet(models.QuerySet):
@@ -54,9 +55,31 @@ def prefetch_gastos(self, **kwargs):
     def annotate_gastos_acima_dobro(self, descricao_gasto):
         media = GastoCotaParlamentar.objects.filter_descricao(descricao_gasto).media()
         acima_dobro = Q(gastos__descricao=descricao_gasto, gastos__valor_liquido__gt=media * 2)
-        count_acima_dobro = Count('pk', filter=acima_dobro)
-        count_geral = Count('pk', filter=Q(gastos__descricao=descricao_gasto))
-        return self.annotate(gastos_acima_dobro=count_acima_dobro, qtd_gastos=count_geral)
+        return self.annotate(
+            qtd_gastos=Count('gastos', filter=Q(gastos__descricao=descricao_gasto)),
+            qtd_acima_dobro=Coalesce(
+                Sum(
+                    Case(
+                        When(
+                            Q(gastos__valor_liquido__gt=media * 2, gastos__descricao=descricao_gasto),
+                            then=Value(1)
+                        ),
+                        output_field=models.IntegerField()
+                    )
+                ),
+                0
+            )
+        )
+
+    def annotate_empresas(self):
+        from empresas.models import Empresa
+        empresas_qs = Empresa.objects.filter(
+            sociedades__socio_pessoa_fisica__nome=OuterRef('nome'),
+            uf=OuterRef('uf')
+        )
+        return self.annotate(
+            empresas=Exists(empresas_qs)
+        )
 
 
 class GastoCotaParlamentarQuerySet(models.QuerySet):

politicos/serializers.py

@@ -28,13 +28,12 @@ class Meta:
 
 class DeputadoSerializer(serializers.ModelSerializer):
     gastos_mensais = serializers.SerializerMethodField()
-    gastos = GastoCotaParlamentarSerializer(many=True)
 
     class Meta:
         model = Deputado
         fields = [
             'id', 'nome', 'partido', 'uf', 'id_legislatura',
-            'gastos', 'gastos_mensais',
+            'gastos_mensais',
         ]
         depth = 2
 
@@ -43,3 +42,15 @@ def get_gastos_mensais(self, obj):
             key: value for key, value in obj.__dict__.items()
             if key.startswith('gastos_')
         }
+
+
+class DeputadoGastosSerializer(serializers.ModelSerializer):
+    gastos = GastoCotaParlamentarSerializer(many=True)
+
+    class Meta:
+        model = Deputado
+        fields = [
+            'id', 'nome', 'partido', 'uf', 'id_legislatura',
+            'gastos',
+        ]
+        depth = 2

politicos/views.py

@@ -2,10 +2,20 @@
 
 from politicos.filters import DeputadoFilter
 from politicos.models import Deputado
-from politicos.serializers import DeputadoSerializer
+from politicos.serializers import DeputadoSerializer, DeputadoGastosSerializer
 
 
 class DeputadoListView(generics.ListAPIView):
     serializer_class = DeputadoSerializer
     filterset_class = DeputadoFilter
-    queryset = Deputado.objects.all().select_related('partido', 'uf')
+    queryset = Deputado.objects.all().select_related(
+        'partido', 'uf'
+    )
+
+
+class DeputadoGastosView(generics.ListAPIView):
+    serializer_class = DeputadoGastosSerializer
+    filterset_class = DeputadoFilter
+    queryset = Deputado.objects.all().select_related(
+        'partido', 'uf'
+    ).prefetch_related('gastos__empresa')