Add twitter time line url to property

This commit is contained in:
Sonny Bakker 2020-09-21 22:22:29 +02:00
parent c6fda4fcee
commit fed811a227

View file

@ -64,11 +64,14 @@ class CollectionRule(TimeStampedModel):
def __str__(self): def __str__(self):
return self.name return self.name
# TODO add twitter url
@property @property
def update_url(self): def update_url(self):
if self.type == RuleTypeChoices.subreddit: if self.type == RuleTypeChoices.subreddit:
return reverse("news:collection:subreddit-update", kwargs={"pk": self.pk}) 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}) return reverse("news:collection:feed-update", kwargs={"pk": self.pk})