From fed811a2272889d37771304a9e7584d54f24caef Mon Sep 17 00:00:00 2001 From: Sonny Bakker Date: Mon, 21 Sep 2020 22:22:29 +0200 Subject: [PATCH] Add twitter time line url to property --- src/newsreader/news/collection/models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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})