from django.db import models from .validators import json_email_array, json_not_empty_string_array, valid_frequency_format from django.utils import timezone from django.conf import settings from api.config import CALENDAR_FREQUENCIES 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.CharField(max_length=20, default="Mensual", validators=[valid_frequency_format]) start_report_date = models.DateTimeField( default=timezone.now) last_report_date = models.DateTimeField( blank=True, null=True, default=None) due_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]) # Actualiza la due_date en cada save del modelo. def save(self, *args, **kwargs): base_date = self.last_report_date if self.last_report_date is not None else self.start_report_date self.due_date = base_date + CALENDAR_FREQUENCIES.get( self.frequency) if self.frequency in CALENDAR_FREQUENCIES else base_date + timezone.timedelta(seconds=int(self.frequency)) super().save(*args, **kwargs) # Metodo que envía la petición a Clima para enviar el reporte. def send_report(self): import requests from django.conf import settings # Calculamos el formato en el que Clima requiere recibir los rangos. start_date = (self.last_report_date if self.last_report_date is not None else self.start_report_date).strftime( "%d/%m/%Y %H:%M") end_date = self.due_date.strftime("%d/%m/%Y %H:%M") # Enviamos la request: response = requests.post(settings.CLIMA_URL + "/async_report_handler", { "id": self.id, "user_id": self.user_id, "format": self.format, "stations": self.stations, "modules": self.modules, "option": self.option, "start_date": start_date, "end_date": end_date, "notified_emails": self.notified_emails }, headers={"AUTHORIZATION": settings.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 self.last_report_date = timezone.now() self.save()