Skip to content

Commit

Permalink
[fix] nyaa engine - paging support & filesize (GiB)
Browse files Browse the repository at this point in the history
BTW: pylint engine

Closes: searxng#3290
Signed-off-by: Markus Heiser <markus.heiser@darmarit.de>
  • Loading branch information
return42 committed Mar 7, 2024
1 parent d97b84b commit 4830493
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 16 deletions.
44 changes: 28 additions & 16 deletions searx/engines/nyaa.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,18 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Nyaa.si (Anime Bittorrent tracker)
# lint: pylint
"""Nyaa.si (Anime Bittorrent tracker)
"""

from lxml import html
from urllib.parse import urlencode
from searx.utils import extract_text, get_torrent_size, int_or_zero

from lxml import html
from searx.utils import (
eval_xpath_getindex,
extract_text,
get_torrent_size,
int_or_zero,
)

# about
about = {
Expand All @@ -23,7 +30,6 @@

# search-url
base_url = 'https://nyaa.si/'
search_url = base_url + '?page=search&{query}&offset={offset}'

# xpath queries
xpath_results = '//table[contains(@class, "torrent-list")]//tr[not(th)]'
Expand All @@ -38,8 +44,14 @@

# do search-request
def request(query, params):
query = urlencode({'term': query})
params['url'] = search_url.format(query=query, offset=params['pageno'])
args = urlencode(
{
'q': query,
'p': params['pageno'],
}
)
params['url'] = base_url + '?' + args #
logger.debug("query_url --> %s", params['url'])
return params


Expand All @@ -56,10 +68,10 @@ def response(resp):
torrent_link = ""

# category in which our torrent belongs
try:
category = result.xpath(xpath_category)[0].attrib.get('title')
except:
pass

category = eval_xpath_getindex(result, xpath_category, 0, '')
if category:
category = category.attrib.get('title')

# torrent title
page_a = result.xpath(xpath_title)[0]
Expand Down Expand Up @@ -87,12 +99,12 @@ def response(resp):
downloads = int_or_zero(result.xpath(xpath_downloads))

# let's try to calculate the torrent size
try:

filesize = None
filesize_info = eval_xpath_getindex(result, xpath_filesize, 0, '')
if filesize_info:
filesize_info = result.xpath(xpath_filesize)[0]
filesize, filesize_multiplier = filesize_info.split()
filesize = get_torrent_size(filesize, filesize_multiplier)
except:
pass
filesize = get_torrent_size(*filesize_info.split())

# content string contains all information not included into template
content = 'Category: "{category}". Downloaded {downloads} times.'
Expand Down
1 change: 1 addition & 0 deletions searx/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@
'GB': 1024 * 1024 * 1024,
'MB': 1024 * 1024,
'TiB': 1000 * 1000 * 1000 * 1000,
'GiB': 1000 * 1000 * 1000,
'MiB': 1000 * 1000,
'KiB': 1000,
}
Expand Down

0 comments on commit 4830493

Please sign in to comment.