KomSuite/KomSuite-NyaaPy
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 3 Wiki Activity

Issue #30 Improve the torrent objects

Browse Source
This commit is contained in:
Dolphin
2020-12-13 21:12:56 +01:00
parent d3b2892c1f
commit 9db8af3a60
7 changed files with 82 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
NyaaPy/nyaa.py
View File

@@ -1,27 +1,29 @@
import requests
from NyaaPy import utils
from NyaaPy import torrent
class Nyaa:
def __init__(self):
self.SITE = utils.TorrentSite.NYAASI
self.URI = "https://nyaa.si"
self.URL = "https://nyaa.si"
def last_uploads(self, number_of_results):
r = requests.get(self.SITE.value)
r = requests.get(self.URL)
# If anything up with nyaa servers let the user know.
r.raise_for_status()
return utils.parse_nyaa(
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):
url = self.SITE.value
url = self.URL
user = kwargs.get('user', None)
category = kwargs.get('category', 0)
@@ -30,7 +32,7 @@ class Nyaa:
page = kwargs.get('page', 0)
if user:
user_uri = "user/{}".format(user)
user_uri = f"user/{user}"
else:
user_uri = ""
@@ -44,24 +46,29 @@ class Nyaa:
r.raise_for_status()
return utils.parse_nyaa(
json_data = utils.parse_nyaa(
request_text=r.text,
limit=None,
site=self.SITE
)
def get(self, id):
r = requests.get("{}/view/{}".format(self.SITE.value, id))
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()
return utils.parse_single(request_text=r.text, site=self.SITE)
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("{}/user/{}".format(self.SITE.value, username))
r = requests.get(f'{self.URL}/user/{username}')
r.raise_for_status()
return utils.parse_nyaa(
json_data = utils.parse_nyaa(
request_text=r.text,
limit=None,
site=self.SITE
)
return torrent.json_to_class(json_data)