From 9095f35545ff811c8cf07ad7256f9670d22f1d31 Mon Sep 17 00:00:00 2001
From: Sonny Bakker <sonny871@hotmail.com>
Date: Sun, 17 Jan 2021 21:44:54 +0100
Subject: [PATCH] Update object representations

Fixes #71
---
 src/newsreader/news/collection/feed.py    | 6 ++++++
 src/newsreader/news/collection/reddit.py  | 6 ++++++
 src/newsreader/news/collection/twitter.py | 6 ++++++
 3 files changed, 18 insertions(+)

diff --git a/src/newsreader/news/collection/feed.py b/src/newsreader/news/collection/feed.py
index 379f18e..368f53c 100644
--- a/src/newsreader/news/collection/feed.py
+++ b/src/newsreader/news/collection/feed.py
@@ -92,6 +92,9 @@ class FeedBuilder(PostBuilder):
         content_details = "\n ".join([item.get("value") for item in content_items])
         return self.sanitize_fragment(content_details)
 
+    def __str__(self):
+        return f"{self.stream.rule.pk}: FeedBuilder"
+
 
 class FeedStream(PostStream):
     rule_type = RuleTypeChoices.feed
@@ -108,6 +111,9 @@ class FeedStream(PostStream):
             message = "Could not parse feed"
             raise StreamParseException(response=response, message=message) from e
 
+    def __str__(self):
+        return f"{self.rule.pk}: FeedStream"
+
 
 class FeedClient(PostClient):
     stream = FeedStream
diff --git a/src/newsreader/news/collection/reddit.py b/src/newsreader/news/collection/reddit.py
index bbd94e3..cc1608a 100644
--- a/src/newsreader/news/collection/reddit.py
+++ b/src/newsreader/news/collection/reddit.py
@@ -290,6 +290,9 @@ class RedditBuilder(PostBuilder):
             title=title,
         )
 
+    def __str__(self):
+        return f"{self.stream.rule.pk}: RedditBuilder"
+
 
 class RedditStream(PostStream):
     rule_type = RuleTypeChoices.subreddit
@@ -315,6 +318,9 @@ class RedditStream(PostStream):
                 response=response, message="Failed parsing json"
             ) from e
 
+    def __str__(self):
+        return f"{self.rule.pk}: RedditStream"
+
 
 class RedditClient(PostClient):
     stream = RedditStream
diff --git a/src/newsreader/news/collection/twitter.py b/src/newsreader/news/collection/twitter.py
index 5ae0b06..e56ec13 100644
--- a/src/newsreader/news/collection/twitter.py
+++ b/src/newsreader/news/collection/twitter.py
@@ -175,6 +175,9 @@ class TwitterBuilder(PostBuilder):
 
         return formatted_entities
 
+    def __str__(self):
+        return f"{self.stream.rule.pk}: TwitterBuilder"
+
 
 class TwitterStream(PostStream):
     rule_type = RuleTypeChoices.twitter_timeline
@@ -199,6 +202,9 @@ class TwitterStream(PostStream):
                 response=response, message="Failed parsing json"
             ) from e
 
+    def __str__(self):
+        return f"{self.rule.pk}: TwitterStream"
+
 
 class TwitterClient(PostClient):
     stream = TwitterStream