Follow's save() and delete() methods gain a "send_signal" param, defaulting to False

kepi/trilby_api/models/follow.py

@@ -73,19 +73,24 @@ class Follow(models.Model):
                     self.following,
                     )
 
-    def save(self, *args, **kwargs):
+    def save(self,
+            send_signal = False,
+            *args, **kwargs):
 
         newly_made = self.pk is None
 
         super().save(*args, **kwargs)
 
-        if newly_made:
+        if send_signal and newly_made:
             logger.debug("%s: sending 'followed'", self)
             trilby_signals.followed.send(sender=self)
 
-    def delete(self, *args, **kwargs):
+    def delete(self,
+            send_signal = False,
+            *args, **kwargs):
 
-        logger.debug("%s: sending 'unfollowed'", self)
-        trilby_signals.unfollowed.send(sender=self)
+        if send_signal:
+            logger.debug("%s: sending 'unfollowed'", self)
+            trilby_signals.unfollowed.send(sender=self)
 
         super().delete(*args, **kwargs)

kepi/trilby_api/views.py

@@ -265,7 +265,9 @@ class Follow(DoSomethingWithPerson):
                 )
 
             with transaction.atomic():
-                follow.save()
+                follow.save(
+                        send_signal = True,
+                        )
 
             logger.info('  -- follow: %s', follow)
             logger.debug('    -- offer ID: %s', offer)
@@ -278,7 +280,9 @@ class Follow(DoSomethingWithPerson):
                     )
 
                 with transaction.atomic():
-                    follow_back.save()
+                    follow_back.save(
+                            send_signal = True,
+                            )
 
                 logger.info('  -- follow back: %s', follow_back)
 
@@ -300,7 +304,9 @@ class Unfollow(DoSomethingWithPerson):
             logger.info('  -- unfollowing: %s', follow)
 
             with transaction.atomic():
-                follow.delete()
+                follow.delete(
+                    send_signal = True,
+                    )
 
             return the_person