Allow searching a user's snatched torrents

src/ptpapi/scripts/ptp.py

@@ -171,7 +171,14 @@ def search_page(api, args, target, movies, torrents, terms):
     if not movies and not torrents:
         logger.debug('Attempting to search target "%s" with terms %s', target, terms)
         if target == "torrents":
-            movies = api.search(terms)
+            if terms.get("type") == "snatched":
+                if "userid" in terms:
+                    user = ptpapi.User(terms["userid"])
+                else:
+                    user = api.current_user()
+                movies = user.snatched(terms)
+            else:
+                movies = api.search(terms)
             # Check to see if we should scrape the cover view data to save calls
             wanted_fields = set()
             if movie_template is not None:

src/ptpapi/user.py

@@ -21,6 +21,14 @@ def __repr__(self):
     def __str__(self):
         return "<ptpapi.User ID %s>" % self.ID
 
+    def snatched(self, filters=None):
+        if filters is None:
+            filters = {}
+        filters["type"] = "snatched"
+        filters["userid"] = str(self.ID)
+        req = session.base_get("torrents.php", params=filters)
+        return [Movie(data=m) for m in snarf_cover_view_data(req.content)]
+
     def bookmarks(self, search_terms=None):
         """Fetch a list of movies the user has bookmarked