feat: allow listing votes for admins

Author: db0

examples/list_post_votes.py

@@ -0,0 +1,57 @@
+
+## Get the latest post by user and list its likes. Required admin access
+## python examples/post_likes.py db0
+
+import os
+import argparse
+import json
+from pythorhead import Lemmy
+
+
+arg_parser = argparse.ArgumentParser()
+arg_parser.add_argument('username', action="store")
+arg_parser.add_argument('-d', '--lemmy_domain', action='store', required=False, type=str, help="the domain in which to look for this user")
+arg_parser.add_argument('-u', '--lemmy_username', action='store', required=False, type=str, help="Which user to authenticate as")
+arg_parser.add_argument('-p', '--lemmy_password', action='store', required=False, type=str, help="Which password to authenticate with")
+args = arg_parser.parse_args()
+
+
+
+lemmy_domain = args.lemmy_domain
+if not lemmy_domain:
+    lemmy_domain = os.getenv('LEMMY_DOMAIN', "lemmy.dbzer0.com")
+if not lemmy_domain:
+    raise Exception("You need to provide a lemmy domain via env var or arg")
+
+lemmy_username = args.lemmy_username
+if not lemmy_username:
+    lemmy_username = os.getenv("LEMMY_USERNAME")
+
+lemmy_password = args.lemmy_password
+if not lemmy_password:
+    lemmy_password = os.getenv("LEMMY_PASSWORD")
+
+lemmy = Lemmy(f"https://{lemmy_domain}")
+if lemmy_username and lemmy_password:
+    login = lemmy.log_in(lemmy_username, lemmy_password)
+user = lemmy.get_user(username=args.username, return_user_object = True)    
+if user:
+    uposts = user.get_latest_posts()
+    if not len(uposts):
+        print("User hasn't made any posts")
+    votes = []
+    scores = []
+    more_votes = []
+    page = 1 
+    # TODO: Make this into a built-in method once a post is a class
+    while len(more_votes) >= 50 or page == 1:
+        more_votes = lemmy.post.list_votes(uposts[0]["post"]["id"], page=page).get("post_likes", [])
+        page += 1
+        votes += more_votes
+        for v in more_votes:
+            scores.append(v["score"])
+    #print(json.dumps(votes, indent=4))
+    print(scores)
+    print(f"{len(votes)} Total votes counted")
+else:
+    print("no matching username found")

pythorhead/classes/user.py

@@ -24,6 +24,8 @@ class LemmyUser:
     matrix_user_id: str | None = None
     ban_expires: datetime | None = None
     updated: datetime | None = None
+    comments: list[dict] = None
+    posts: list[dict] = None
     # The owning lemmy instance. We use it to reach the API classes
     lemmy = None
 
@@ -44,13 +46,17 @@ def refresh(self) -> None:
         Args:
             new_data: Dictionary containing updated user data
         """
-        new_data = self.lemmy.user.get(person_id=self.person_id)
+        fresh_data = self.lemmy.user.get(person_id=self.id)
+        user_dict = fresh_data['person_view']['person']
+        user_dict['is_admin'] = fresh_data['person_view']['is_admin']
+        user_dict['comments'] = fresh_data['comments']
+        user_dict['posts'] = fresh_data['posts']
         # Handle datetime conversion
-        if 'ban_expires' in new_data and new_data['ban_expires']:
-            new_data['ban_expires'] = datetime.fromisoformat(new_data['ban_expires'])
+        if 'ban_expires' in user_dict and user_dict['ban_expires']:
+            user_dict['ban_expires'] = datetime.fromisoformat(user_dict['ban_expires'])
 
-        for field_name in new_data:
-            setattr(self, field_name, new_data[field_name])
+        for field_name in user_dict:
+            setattr(self, field_name, user_dict[field_name])
 
 
     def purge(self) -> None:
@@ -101,4 +107,13 @@ def pm(self, content):
         self.lemmy.private_message(
             content=content,
             recipient_id=self.id,
-        )
+        )
+    
+    def get_latest_posts(self):
+        self.refresh()
+        return self.posts
+
+    def get_latest_comments(self):
+        self.refresh()
+        return self.comments
+    

pythorhead/lemmy.py

@@ -99,6 +99,8 @@ def get_user(self, **kwargs) -> LemmyUser | None:
             return
         user_dict = user_json['person_view']['person']
         user_dict['is_admin'] = user_json['person_view']['is_admin']
+        user_dict['comments'] = user_json['comments']
+        user_dict['posts'] = user_json['posts']
         return LemmyUser.from_dict(user_dict, self)
 
     def search(

pythorhead/post.py

@@ -298,6 +298,7 @@ def lock(self, post_id: int, locked: bool) -> Optional[dict]:
             "locked": locked,
         }
         return self._requestor.api(Request.POST, "/post/lock", json=lock_post)
+
     def mark_as_read(self, post_id: int, read: bool) -> Optional[dict]:
         """
 
@@ -433,3 +434,24 @@ def purge(self, id: int, reason: Optional[str]) -> Optional[dict]:
         return self._requestor.api(Request.POST, "/admin/purge/post", json=purge_post)
 
     __call__ = create
+
+
+    def list_votes(self, post_id: int, page: int = 1, limit: int = 50) -> Optional[dict]:
+        """
+
+        List the likes of a post
+
+        Args:
+            post_id (int)
+            page (int)
+            limit (int)
+
+        Returns:
+            Optional[dict]: list of likes if successful
+        """
+        list_votes_params = {
+            "post_id": post_id,
+            "page": page,
+            "limit": limit,
+        }
+        return self._requestor.api(Request.GET, "/post/like/list", params=list_votes_params)