From 20e63023bb0d3c7e4cb36b91b73e79f51889cc90 Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Tue, 15 Nov 2022 18:30:30 -0700
Subject: Get outbound likes/boosts and their undos working

---
 activities/models/timeline_event.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'activities/models/timeline_event.py')

diff --git a/activities/models/timeline_event.py b/activities/models/timeline_event.py
index 6dba32c..29dec19 100644
--- a/activities/models/timeline_event.py
+++ b/activities/models/timeline_event.py
@@ -114,3 +114,20 @@ class TimelineEvent(models.Model):
                 subject_identity_id=interaction.identity_id,
                 subject_post_interaction=interaction,
             )[0]
+
+    @classmethod
+    def delete_post_interaction(cls, identity, interaction):
+        if interaction.type == interaction.Types.like:
+            cls.objects.filter(
+                identity=identity,
+                type=cls.Types.liked,
+                subject_post_id=interaction.post_id,
+                subject_identity_id=interaction.identity_id,
+            ).delete()
+        elif interaction.type == interaction.Types.boost:
+            cls.objects.filter(
+                identity=identity,
+                type__in=[cls.Types.boosted, cls.Types.boost],
+                subject_post_id=interaction.post_id,
+                subject_identity_id=interaction.identity_id,
+            ).delete()
-- 
cgit v1.2.3