From ef4b31d5d99cdeb2cdc03ebb16f9da757719f001 Mon Sep 17 00:00:00 2001 From: Thiago Nogueira Date: Fri, 8 Jun 2018 20:29:09 -0300 Subject: [PATCH] =?UTF-8?q?#4=20extraindo=20m=C3=A9todo=20devido=20a=20com?= =?UTF-8?q?plexidade=20ciclom=C3=A1tica?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../exportadores/exportador_csv_r.py | 30 ++++++++++++------- 1 file changed, 20 insertions(+), 10 deletions(-) diff --git a/radar_parlamentar/exportadores/exportador_csv_r.py b/radar_parlamentar/exportadores/exportador_csv_r.py index d04e4074..2d7ab358 100644 --- a/radar_parlamentar/exportadores/exportador_csv_r.py +++ b/radar_parlamentar/exportadores/exportador_csv_r.py @@ -57,27 +57,37 @@ def exportar_csv(self): self.retrieve_votacoes() self.transform_data() self.write_csv() - - def retrieve_votacoes(self): - try: - casa = models.CasaLegislativa.objects.get(nome_curto='cdep') - except: - casa = None - if self.ini is None and self.fim is None: + + def _ini_is_none(self, casa): + if self.fim is None: self.votacoes = models.Votacao.objects.filter( proposicao__casa_legislativa=casa).order_by('data') - if self.ini is None and self.fim is not None: + else: self.votacoes = models.Votacao.objects.filter( proposicao__casa_legislativa=casa ).filter(data__lte=self.fim).order_by('data') - if self.ini is not None and self.fim is None: + + def _ini_is_not_none(self, casa): + if self.fim is None: self.votacoes = models.Votacao.objects.filter( proposicao__casa_legislativa=casa ).filter(data__gte=self.ini).order_by('data') - if self.ini is not None and self.fim is not None: + else: self.votacoes = models.Votacao.objects.filter( proposicao__casa_legislativa=casa ).filter(data__gte=self.ini, data__lte=self.fim).order_by('data') + + + def retrieve_votacoes(self): + try: + casa = models.CasaLegislativa.objects.get(nome_curto='cdep') + except: + casa = None + if self.ini is None: + self._ini_is_none(casa) + else: + self._ini_is_not_none(casa) + def transform_data(self): self.csv_rows.append(LABELS)