from django.db import models from .validators import json_email_array, json_not_empty_string_array from django.utils import timezone # TODO! Move to ENV file... PROGRAMMED_REPORTS_SERVICE_AUTH_TOKEN = "juwiV9MjP4r3NTsvd9qRsPQ2FMdk5soJjxksYKhruza3wskG23h4cygTEPiFAGo2" class ProgrammedReport(models.Model): REPORT_FORMATS = [ ("excel", "Excel"), ("csv", "CSV") ] name = models.CharField(max_length=100) user_id = models.CharField(max_length=2048) frequency = models.DurationField(default=timezone.timedelta(days=1)) start_report_date = models.DateTimeField( default=timezone.now) last_report_date = models.DateTimeField( blank=True, null=True, default=None) # Valores que configuran al reporte: format = models.CharField( max_length=5, choices=REPORT_FORMATS) option = models.CharField(max_length=50, default="all") stations = models.JSONField(validators=[json_not_empty_string_array]) modules = models.JSONField(validators=[json_not_empty_string_array]) notified_emails = models.JSONField( validators=[json_not_empty_string_array, json_email_array]) # Metodo que envía la petición a Clima para enviar el reporte. # TODO! Tecnicamente start_date y end_date podrían ser calculados por el modelo, # hay que ver si hay un caso borde que requiere de que el llamador especifique el rango. # Basicamente: Testear def send_report(self, start_date, end_date): import requests from django.conf import settings # Calculamos el formato en el que Clima requiere recibir los rangos. start = start_date.strftime("%d/%m/%Y %H:%M") end = end_date.strftime("%d/%m/%Y %H:%M") response = requests.post(settings.CLIMA_URL + "/async_report_handler", { "format": self.format, "stations": self.stations, "modules": self.modules, "option": self.option, "start_date": start, "end_date": end, "notified_emails": self.notified_emails }, headers={"AUTHORIZATION": PROGRAMMED_REPORTS_SERVICE_AUTH_TOKEN}) if response.status_code != 200: import logging logging.getLogger('django').error( "No se pudo enviar el reporte asincrono: %s\n Reason: %s", self, response.reason) # TODO! Seria ideal enviar un email al administrador para ver que paso. return False # TODO! Si se utiliza una Message Queue para la confirmación del envío entonces # hay que mover esto al listener de la Queue. self.last_report_date = timezone.now() self.save() return True