|
1 | | -# from django.db import models |
2 | | - |
3 | 1 | # Create your models here. |
| 2 | +from copy import deepcopy |
| 3 | + |
| 4 | +from common.models import TimestampedModel, UUIDPrimaryKeyModel |
| 5 | +from django.contrib.postgres.fields import JSONField |
| 6 | +from django.db import models |
| 7 | +from django.utils.translation import ugettext_lazy as _ |
| 8 | + |
| 9 | + |
| 10 | +class DeliveryLog(UUIDPrimaryKeyModel, TimestampedModel): |
| 11 | + user = models.ForeignKey( |
| 12 | + "users.User", |
| 13 | + related_name="delivery_logs", |
| 14 | + on_delete=models.CASCADE, |
| 15 | + verbose_name=_("user"), |
| 16 | + ) |
| 17 | + report = JSONField(verbose_name=_("report"), blank=True, null=True) |
| 18 | + |
| 19 | + class Meta: |
| 20 | + verbose_name = _("delivery log") |
| 21 | + verbose_name_plural = _("delivery logs") |
| 22 | + ordering = ["-updated_at"] |
| 23 | + |
| 24 | + def update_report(self, report_data): |
| 25 | + """ |
| 26 | + Update delivery of single message base on json data sent from Quriiri to |
| 27 | + our service |
| 28 | + :param report_data: JSON |
| 29 | + :return: |
| 30 | + """ |
| 31 | + report = self.report |
| 32 | + updated_messages = deepcopy(report["messages"]) |
| 33 | + destination = report_data["destination"] |
| 34 | + for k, v in report["messages"].items(): |
| 35 | + if k == destination or v["converted"] == destination: |
| 36 | + updated_messages[k] = report_data |
| 37 | + report["messages"] = updated_messages |
| 38 | + self.report = report |
| 39 | + self.save() |
0 commit comments