Add optin as field to channelevents

temba/channels/migrations/0177_channelevent_optin.py

@@ -0,0 +1,21 @@
+# Generated by Django 4.2.3 on 2023-10-04 15:54
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("msgs", "0250_alter_msg_msg_type"),
+        ("channels", "0176_remove_channel_alert_email_delete_alert"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="channelevent",
+            name="optin",
+            field=models.ForeignKey(
+                null=True, on_delete=django.db.models.deletion.PROTECT, related_name="optins", to="msgs.optin"
+            ),
+        ),
+    ]

temba/channels/models.py

@@ -852,6 +852,7 @@ class ChannelEvent(models.Model):
     contact_urn = models.ForeignKey(
         "contacts.ContactURN", on_delete=models.PROTECT, null=True, related_name="channel_events"
     )
+    optin = models.ForeignKey("msgs.OptIn", null=True, on_delete=models.PROTECT, related_name="optins")
     extra = JSONAsTextField(null=True, default=dict)
     occurred_on = models.DateTimeField()
     created_on = models.DateTimeField(default=timezone.now)

temba/contacts/models.py

@@ -778,7 +778,7 @@ def get_history(self, after: datetime, before: datetime, include_event_types: se
             self.msgs.filter(created_on__gte=after, created_on__lt=before)
             .exclude(status=Msg.STATUS_PENDING)
             .order_by("-created_on", "-id")
-            .select_related("channel", "contact_urn", "broadcast")[:limit]
+            .select_related("channel", "contact_urn", "broadcast", "optin")[:limit]
         )
 
         # get all runs start started or ended in this period
@@ -797,7 +797,7 @@ def get_history(self, after: datetime, before: datetime, include_event_types: se
         channel_events = (
             self.channel_events.filter(created_on__gte=after, created_on__lt=before)
             .order_by("-created_on")
-            .select_related("channel")[:limit]
+            .select_related("channel", "optin")[:limit]
         )
 
         campaign_events = (

temba/mailroom/events.py

@@ -252,7 +252,7 @@ def from_channel_event(cls, org: Org, user: User, obj: ChannelEvent) -> dict:
         if obj.event_type in ChannelEvent.CALL_TYPES:
             ch_event["duration"] = extra.get("duration")
         elif obj.event_type in (ChannelEvent.TYPE_OPTIN, ChannelEvent.TYPE_OPTOUT):
-            ch_event["optin"] = _optin(OptIn.objects.get(org=org, id=extra.get("optin_id")))
+            ch_event["optin"] = _optin(obj.optin) if obj.optin else None
 
         return {
             "type": cls.TYPE_CHANNEL_EVENT,