feat(all): add Poetry support

2024-07-17 21:34:25 +00:00
parent acfeae3467
commit 7afb4823c0
12 changed files with 827 additions and 200 deletions
--- a/nyaapy/nyaa.py
+++ b/nyaapy/nyaa.py
@@ -0,0 +1,94 @@
+import requests
+from nyaapy import utils
+from nyaapy import torrent
+
+
+class Nyaa:
+
+    def __init__(self):
+        self.SITE = utils.TorrentSite.NYAASI
+        self.URL = "https://nyaa.si"
+
+    def last_uploads(self, number_of_results):
+        r = requests.get(self.URL)
+
+        # If anything up with nyaa servers let the user know.
+        r.raise_for_status()
+
+        json_data = utils.parse_nyaa(
+            request_text=r.text, limit=number_of_results + 1, site=self.SITE
+        )
+
+        return torrent.json_to_class(json_data)
+
+    def search(self, keyword, **kwargs):
+        base_url = self.URL
+
+        user = kwargs.get("user", None)
+        category = kwargs.get("category", 0)
+        subcategory = kwargs.get("subcategory", 0)
+        filters = kwargs.get("filters", 0)
+        page = kwargs.get("page", 0)
+        sorting = kwargs.get(
+            "sort", "id"
+        )  # Sorting by id = sorting by date, this is the default.
+        order = kwargs.get("order", "desc")
+
+        user_uri = f"user/{user}" if user else ""
+
+        if page > 0:
+            search_uri = "{}/{}?f={}&c={}_{}&q={}&p={}&s={}&o={}".format(
+                base_url,
+                user_uri,
+                filters,
+                category,
+                subcategory,
+                keyword,
+                page,
+                sorting,
+                order,
+            )
+        else:
+            search_uri = "{}/{}?f={}&c={}_{}&q={}&s={}&o={}".format(
+                base_url,
+                user_uri,
+                filters,
+                category,
+                subcategory,
+                keyword,
+                sorting,
+                order,
+            )
+
+        if not user:
+            search_uri += "&page=rss"
+
+        http_response = requests.get(search_uri)
+        http_response.raise_for_status()
+
+        if user:
+            json_data = utils.parse_nyaa(
+                request_text=http_response.text, limit=None, site=self.SITE
+            )
+        else:
+            json_data = utils.parse_nyaa_rss(
+                request_text=http_response.text, limit=None, site=self.SITE
+            )
+
+        # Convert JSON data to a class object
+        return torrent.json_to_class(json_data)
+
+    def get(self, view_id):
+        r = requests.get(f"{self.URL}/view/{view_id}")
+        r.raise_for_status()
+
+        json_data = utils.parse_single(request_text=r.text, site=self.SITE)
+
+        return torrent.json_to_class(json_data)
+
+    def get_user(self, username):
+        r = requests.get(f"{self.URL}/user/{username}")
+        r.raise_for_status()
+
+        json_data = utils.parse_nyaa(request_text=r.text, limit=None, site=self.SITE)
+        return torrent.json_to_class(json_data)