X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=reddit_renderer.py;h=192bc206e75b860661816837d396afc3059ac6b1;hb=477c84bcdefc573278ec8ed8974d5be8baac562d;hp=30c22e0a4c6021dae7b7d64700ca2d8e90643642;hpb=5c39d86ebc075ccb7be98b1dfab8040b72ff9134;p=kiosk.git

diff --git a/reddit_renderer.py b/reddit_renderer.py
index 30c22e0..192bc20 100644
--- a/reddit_renderer.py
+++ b/reddit_renderer.py
@@ -1,10 +1,10 @@
 #!/usr/bin/env python3
 
-import praw  # type: ignore
-import random
+import logging
 from typing import Callable, Dict, Iterable, List, Set
 
-import constants
+import praw  # type: ignore
+
 import file_writer
 import grab_bag
 import page_builder
@@ -13,7 +13,10 @@ import renderer
 import kiosk_secrets as secrets
 
 
-class reddit_renderer(renderer.debuggable_abstaining_renderer):
+logger = logging.getLogger(__file__)
+
+
+class reddit_renderer(renderer.abstaining_renderer):
     """A renderer to pull text content from reddit."""
 
     def __init__(
@@ -25,7 +28,7 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
         font_size: int = 24,
         additional_filters: Iterable[Callable[[str], bool]] = [],
     ):
-        super(reddit_renderer, self).__init__(name_to_timeout_dict, True)
+        super().__init__(name_to_timeout_dict)
         self.subreddit_list = subreddit_list
         self.praw = praw.Reddit(
             client_id=secrets.reddit_client_id,
@@ -41,14 +44,8 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
         self.filters.extend(additional_filters)
         self.deduper: Set[str] = set()
 
-    def debug_prefix(self) -> str:
-        x = ""
-        for subreddit in self.subreddit_list:
-            x += f"{subreddit} "
-        return f"reddit({x.strip()})"
-
     def periodic_render(self, key: str) -> bool:
-        self.debug_print('called for "%s"' % key)
+        logger.debug('called for "%s"' % key)
         if key == "Scrape":
             return self.scrape_reddit()
         elif key == "Shuffle":
@@ -67,23 +64,26 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
                     filtered = filt.__name__
                     break
             if filtered != "":
-                print(f'Filter {filtered} struck down "{title}"')
+                logger.info(
+                    f'Filter {filtered} struck down "{title}"'
+                )
                 continue
             if msg.ups < self.min_votes:
-                print(f'"{title}" doesn\'t have enough upvotes to be interesting')
+                logger.debug(
+                    f'"{title}" doesn\'t have enough upvotes to be interesting'
+                )
                 continue
 
-            try:
-                self.deduper.add(title)
-                content = f"{msg.ups}"
-                if (
+            self.deduper.add(title)
+            content = f"{msg.ups}"
+            if (
                     msg.thumbnail != "self"
                     and msg.thumbnail != "default"
                     and msg.thumbnail != ""
-                ):
-                    content = f'<IMG SRC="{msg.thumbnail}">'
-                self.messages.add(
-                    f"""
+            ):
+                content = f'<IMG SRC="{msg.thumbnail}">'
+            self.messages.add(
+f"""
 <TABLE STYLE="font-size:{self.font_size}pt;">
   <TR>
     <!-- The number of upvotes or item image: -->
@@ -97,9 +97,7 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
     </TD>
   </TR>
 </TABLE>"""
-                )
-            except:
-                self.debug_print("Unexpected exception, skipping message.")
+            )
 
     def scrape_reddit(self) -> bool:
         self.deduper.clear()
@@ -130,7 +128,7 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
                 self.append_message(msg)
             except:
                 pass
-            self.debug_print(f"There are now {self.messages.size()} messages")
+            logger.debug(f"There are now {self.messages.size()} messages")
         return True
 
     def shuffle_messages(self) -> bool:
@@ -147,7 +145,7 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
         layout.set_title("Reddit /r/%s" % x.strip())
         subset = self.messages.subset(4)
         if subset is None:
-            self.debug_print("Not enough messages to pick from.")
+            logger.debug("Not enough messages to pick from.")
             return False
         for msg in subset:
             layout.add_item(msg)
@@ -158,14 +156,14 @@ class reddit_renderer(renderer.debuggable_abstaining_renderer):
 
 class til_reddit_renderer(reddit_renderer):
     def __init__(self, name_to_timeout_dict: Dict[str, int]):
-        super(til_reddit_renderer, self).__init__(
+        super().__init__(
             name_to_timeout_dict, ["todayilearned"], min_votes=100, font_size=20
         )
 
 
 class quotes_reddit_renderer(reddit_renderer):
     def __init__(self, name_to_timeout_dict: Dict[str, int]):
-        super(quotes_reddit_renderer, self).__init__(
+        super().__init__(
             name_to_timeout_dict, ["quotes"], min_votes=100, font_size=20
         )
 
@@ -176,7 +174,7 @@ class showerthoughts_reddit_renderer(reddit_renderer):
         return "gift card" in msg
 
     def __init__(self, name_to_timeout_dict: Dict[str, int]):
-        super(showerthoughts_reddit_renderer, self).__init__(
+        super().__init__(
             name_to_timeout_dict,
             ["showerthoughts"],
             min_votes=150,
@@ -188,7 +186,7 @@ class showerthoughts_reddit_renderer(reddit_renderer):
 
 class seattle_reddit_renderer(reddit_renderer):
     def __init__(self, name_to_timeout_dict: Dict[str, int]):
-        super(seattle_reddit_renderer, self).__init__(
+        super().__init__(
             name_to_timeout_dict,
             ["seattle", "seattleWA", "SeaWA", "bellevue", "kirkland", "CoronavirusWA"],
             min_votes=50,
@@ -197,7 +195,7 @@ class seattle_reddit_renderer(reddit_renderer):
 
 class lifeprotips_reddit_renderer(reddit_renderer):
     def __init__(self, name_to_timeout_dict: Dict[str, int]):
-        super(lifeprotips_reddit_renderer, self).__init__(
+        super().__init__(
             name_to_timeout_dict, ["lifeprotips"], min_votes=50
         )