Nyaa.si & sukebei.nyaa.si LXML fully ready

NyaaPy/pantsu.py

@@ -1,18 +1,24 @@
 import requests
 from NyaaPy import utils
 
+
 class Pantsu:
 
     def __init__(self):
         self.BASE_URL = "https://nyaa.pantsu.cat/api"
-    
+        self.SITE = utils.TorrentSite.NYAANET
+
     def last_uploads(self, number_of_results):
-       r = requests.get(self.URI)
-       soup = BeautifulSoup(r.text, 'html.parser')
-       rows = soup.select('table tr')
-
-       return utils.parse_nyaa(rows, limit=number_of_results + 1)
+        r = requests.get(self.SITE.value)
+        r.raise_for_status()
+        with open("test.html", "w") as f:
+            f.write(r.text)
 
+        return utils.parse_nyaa(
+            request_text=r.text,
+            limit=number_of_results + 1,
+            site=self.SITE
+        )
 
     # Torrents - GET
     def search(self, keyword, **kwargs):
@@ -23,10 +29,11 @@ class Pantsu:
     def view(self, item_id):
         request = requests.get("{}/view/{}".format(self.BASE_URL, item_id))
 
+        request.raise_for_status()
+
         return request.json()
 
     # Torrents - POST
-
     def upload(self):
         return "Work in progress!"
 
@@ -34,7 +41,6 @@ class Pantsu:
         return "Work in progress!"
 
     # Users
-
     def login(self, username, password):
         login = requests.post("{}/login/".format(
             self.BASE_URL), data={'username': username, 'password': password})