Added handling for temporary connection loss

pointsbot/bot.py

@@ -19,7 +19,11 @@ MOD_SOLVED_PAT = re.compile('/[Ss]olved')
 def run():
     cfg = config.load()
     levels = cfg.levels
+    db = database.Database(cfg.database_path)
 
+    # Run indefinitely, reconnecting any time a connection is lost
+    while True:
+        try:
             reddit = praw.Reddit(client_id=cfg.client_id,
                                  client_secret=cfg.client_secret,
                                  username=cfg.username,
@@ -33,11 +37,18 @@ def run():
             is_mod = bool(subreddit.moderator(redditor=reddit.user.me()))
             print_level(1, f'Is mod? {is_mod}')
 
-    db = database.Database(cfg.database_path)
+            monitor_comments(subreddit, db)
+        # Ignoring other potential exceptions for now, since we may not be able
+        # to recover from them as well as from this one
+        except prawcore.exceptions.ServerError as e:
+            print('Lost connection to Reddit; attempting to reconnect....')
 
-    # Monitor new comments for confirmed solutions
+
-    # Passing pause_after=0 will bypass the internal exponential delay; instead,
+def monitor_comments(subreddit, db):
-    # have to check if any comments are returned with each query
+    """Monitor new comments in the subreddit, looking for confirmed solutions.
+    """
+    # Passing pause_after=0 will bypass the internal exponential delay, but have
+    # to check if any comments are returned with each query
     for comm in subreddit.stream.comments(skip_existing=True, pause_after=0):
         if comm is None:
             continue