84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
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):
|
|
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)
|
|
|
|
if user:
|
|
user_uri = f"user/{user}"
|
|
else:
|
|
user_uri = ""
|
|
|
|
if page > 0:
|
|
uri = f"{url}/{user_uri}?f={filters}&c={category}_{subcategory}&q={keyword}&p={page}"
|
|
else:
|
|
uri = f"{url}/{user_uri}?f={filters}&c={category}_{subcategory}&q={keyword}"
|
|
|
|
if not user:
|
|
uri += "&page=rss"
|
|
|
|
http_response = requests.get(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
|
|
)
|
|
|
|
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)
|