Hide read conversations

twtxtmanager.py

@@ -21,6 +21,9 @@ class TwtxtManager:
         self.unread_twts_count = 0
         self.read_twts_count = 0
 
+        self.show_read_conversations = False
+        self.show_unread_missing_conversation_roots = False
+
     def load_twts(self):
         all_tweets = []
 
@@ -77,7 +80,7 @@ class TwtxtManager:
         # TODO Add cyclic dependency check!!
 
         # list of all top level tweets
-        self.conversation_tree = replies_by_subject.pop(None, [])
+        conversation_tree = replies_by_subject.pop(None, [])
 
         # build conversation trees
         for twt in all_tweets:
@@ -107,11 +110,24 @@ class TwtxtManager:
                 entry = self._cache2.get(hash)
                 twt.read = entry and entry.get("read", False)
                 twt.missing = True
-                self.conversation_tree.append(twt)
+                conversation_tree.append(twt)
                 self.missing_conversation_root_count += 1
-        self.conversation_tree.sort()
+        conversation_tree.sort()
         self.twts_count = len(all_tweets)
 
+        if self.show_read_conversations:
+            self.conversation_tree = conversation_tree
+        else:
+            self.conversation_tree = []
+            for twt in conversation_tree:
+                if self.show_unread_missing_conversation_roots:
+                    consider_twt_read = twt.read
+                else:
+                    consider_twt_read = (hasattr(twt, "missing") and twt.missing) or twt.read
+
+                if not consider_twt_read or self._has_unread_replies(twt):
+                    self.conversation_tree.append(twt)
+
 
     def toggle_read(self, twt):
         """
@@ -155,3 +171,16 @@ class TwtxtManager:
         return False
 
 
+    def _has_unread_replies(self, twt):
+        """
+        Determine whether there are any unread replies in the given
+        conversation without taking the read status of the given twt itself
+        into account.
+        """
+
+        for reply in twt.replies:
+            if not reply.read or self._has_unread_replies(reply):
+                return True
+        return False
+
+