Made few tests

2020-01-22 19:07:38 +01:00
parent 95b967b940
commit bf01a922f0
3 changed files with 39 additions and 12 deletions
--- a/NyaaPy/nyaa.py
+++ b/NyaaPy/nyaa.py
@@ -46,13 +46,11 @@ class Nyaa:
    def get(self, id):
        r = requests.get("{}/view/{}".format(self.URI, id))
        r.raise_for_status()
-        # ! Description not working TODO
-        # with open("test.html", "w") as f:
-        #    f.write(r.text)
+
        return utils.parse_single(request_text=r.text)

    def get_user(self, username):
        r = requests.get("{}/user/{}".format(self.URI, username))
-        soup = BeautifulSoup(r.text, 'html.parser')
+        r.raise_for_status()

-        return utils.parse_nyaa(soup.select('table tr'), limit=None)
+        return utils.parse_nyaa(request_text=r.text, limit=None)