Commit de no perder cambios

src/controller/buffers/blueski/__init__.py

@@ -10,5 +10,5 @@ from .timeline import (
     UserTimeline,
     SearchBuffer,
 )
-from .user import FollowersBuffer, FollowingBuffer, BlocksBuffer
+from .user import FollowersBuffer, FollowingBuffer, BlocksBuffer, PostUserListBuffer
 from .chat import ConversationListBuffer, ChatBuffer as ChatMessageBuffer

src/controller/buffers/blueski/user.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import logging
+import output
 from .base import BaseBuffer
 from wxUI.buffers.blueski import panels as BlueskiPanels
 from sessions.blueski import compose
@@ -12,6 +13,7 @@ class UserBuffer(BaseBuffer):
         kwargs["compose_func"] = "compose_user"
         super(UserBuffer, self).__init__(*args, **kwargs)
         self.type = "user"
+        self.next_cursor = None
         
     def create_buffer(self, parent, name):
         self.buffer = BlueskiPanels.UserPanel(parent, name)
@@ -36,6 +38,7 @@ class UserBuffer(BaseBuffer):
             else:
                 res = getattr(self.session, api_method)(limit=count)
             items = res.get("items", [])
+            self.next_cursor = res.get("cursor")
             
             # Clear existing items for these lists to start fresh? 
             # Or append? Standard lists in TWBlue usually append.
@@ -47,6 +50,31 @@ class UserBuffer(BaseBuffer):
             log.exception(f"Error fetching user list for {self.name}")
             return 0
 
+    def get_more_items(self):
+        api_method = self.kwargs.get("api_method")
+        if not api_method or not self.next_cursor:
+            return
+
+        count = self.session.settings["general"].get("max_posts_per_call", 50)
+        actor = (
+            self.kwargs.get("actor")
+            or self.kwargs.get("did")
+            or self.kwargs.get("handle")
+            or self.kwargs.get("id")
+        )
+        try:
+            if api_method in ("get_followers", "get_follows"):
+                res = getattr(self.session, api_method)(actor=actor, limit=count, cursor=self.next_cursor)
+            else:
+                res = getattr(self.session, api_method)(limit=count, cursor=self.next_cursor)
+            items = res.get("items", [])
+            self.next_cursor = res.get("cursor")
+            added = self.process_items(items, play_sound=False)
+            if added:
+                output.speak(_(u"%s items retrieved") % (str(added)), True)
+        except Exception:
+            log.exception(f"Error fetching more user list items for {self.name}")
+
 class FollowersBuffer(UserBuffer):
     def __init__(self, *args, **kwargs):
         kwargs["api_method"] = "get_followers"
@@ -109,3 +137,51 @@ class BlocksBuffer(UserBuffer):
     def __init__(self, *args, **kwargs):
         kwargs["api_method"] = "get_blocks"
         super(BlocksBuffer, self).__init__(*args, **kwargs)
+
+
+class PostUserListBuffer(UserBuffer):
+    def __init__(self, *args, **kwargs):
+        self.post_uri = kwargs.get("post_uri")
+        self.api_method = kwargs.get("api_method")
+        super(PostUserListBuffer, self).__init__(*args, **kwargs)
+        self.type = "post_user_list"
+
+    def start_stream(self, mandatory=False, play_sound=True):
+        if not self.api_method or not self.post_uri:
+            return 0
+        count = self.session.settings["general"].get("max_posts_per_call", 50)
+        try:
+            res = getattr(self.session, self.api_method)(self.post_uri, limit=count)
+            items = res.get("items", [])
+            self.next_cursor = res.get("cursor")
+            return self.process_items(items, play_sound)
+        except Exception:
+            log.exception("Error fetching post user list for %s", self.name)
+            return 0
+
+    def get_more_items(self):
+        if not self.api_method or not self.post_uri or not self.next_cursor:
+            return
+        count = self.session.settings["general"].get("max_posts_per_call", 50)
+        try:
+            res = getattr(self.session, self.api_method)(self.post_uri, limit=count, cursor=self.next_cursor)
+            items = res.get("items", [])
+            self.next_cursor = res.get("cursor")
+            added = self.process_items(items, play_sound=False)
+            if added:
+                output.speak(_(u"%s items retrieved") % (str(added)), True)
+        except Exception:
+            log.exception("Error fetching more post user list items for %s", self.name)
+
+    def remove_buffer(self, force=False):
+        if not force:
+            from wxUI import commonMessageDialogs
+            import widgetUtils
+            dlg = commonMessageDialogs.remove_buffer()
+            if dlg != widgetUtils.YES:
+                return False
+        try:
+            self.session.db.pop(self.name, None)
+        except Exception:
+            pass
+        return True