diff --git a/src/newsreader/news/collection/models.py b/src/newsreader/news/collection/models.py index 23d76f6..92dfe51 100644 --- a/src/newsreader/news/collection/models.py +++ b/src/newsreader/news/collection/models.py @@ -64,11 +64,14 @@ class CollectionRule(TimeStampedModel): def __str__(self): return self.name - # TODO add twitter url @property def update_url(self): if self.type == RuleTypeChoices.subreddit: return reverse("news:collection:subreddit-update", kwargs={"pk": self.pk}) + elif self.type == RuleTypeChoices.twitter_timeline: + return reverse( + "news:collection:twitter-timeline-update", kwargs={"pk": self.pk} + ) return reverse("news:collection:feed-update", kwargs={"pk": self.pk})