Ponysearch/searx/engines/yggtorrent.py

128 lines
3.4 KiB
Python
Raw Normal View History

# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Yggtorrent (Videos, Music, Files)
"""
2020-08-18 18:02:41 +02:00
from lxml import html
from operator import itemgetter
from datetime import datetime
from urllib.parse import quote
from searx.utils import extract_text, get_torrent_size
from searx.network import get as http_get
2020-08-18 18:02:41 +02:00
# about
about = {
2021-02-15 13:38:34 +01:00
"website": 'https://www4.yggtorrent.li/',
"wikidata_id": None,
"official_api_documentation": None,
"use_official_api": False,
"require_api_key": False,
"results": 'HTML',
}
2020-08-18 18:02:41 +02:00
# engine dependent config
categories = ['files']
2020-08-18 18:02:41 +02:00
paging = True
# search-url
2021-02-15 13:38:34 +01:00
url = 'https://www4.yggtorrent.li/'
2020-08-18 18:02:41 +02:00
search_url = url + 'engine/search?name={search_term}&do=search&page={pageno}&category={search_type}'
# yggtorrent specific type-definitions
search_types = {'files': 'all',
'music': '2139',
'videos': '2145'}
cookies = dict()
def init(engine_settings=None):
# initial cookies
resp = http_get(url, allow_redirects=False)
if resp.ok:
for r in resp.history:
cookies.update(r.cookies)
cookies.update(resp.cookies)
# do search-request
def request(query, params):
search_type = search_types.get(params['category'], 'all')
pageno = (params['pageno'] - 1) * 50
params['url'] = search_url.format(search_term=quote(query),
search_type=search_type,
pageno=pageno)
params['cookies'] = cookies
return params
# get response from search-request
def response(resp):
results = []
dom = html.fromstring(resp.text)
search_res = dom.xpath('//section[@id="#torrents"]/div/table/tbody/tr')
# return empty array if nothing is found
if not search_res:
return []
# parse results
for result in search_res:
link = result.xpath('.//a[@id="torrent_name"]')[0]
href = link.attrib.get('href')
title = extract_text(link)
seed = result.xpath('.//td[8]/text()')[0]
leech = result.xpath('.//td[9]/text()')[0]
# convert seed to int if possible
if seed.isdigit():
seed = int(seed)
else:
seed = 0
# convert leech to int if possible
if leech.isdigit():
leech = int(leech)
else:
leech = 0
params = {'url': href,
'title': title,
'seed': seed,
'leech': leech,
'template': 'torrent.html'}
# let's try to calculate the torrent size
try:
filesize_info = result.xpath('.//td[6]/text()')[0]
filesize = filesize_info[:-2]
filesize_multiplier = filesize_info[-2:].lower()
multiplier_french_to_english = {
'to': 'TiB',
'go': 'GiB',
'mo': 'MiB',
'ko': 'KiB'
}
filesize = get_torrent_size(filesize, multiplier_french_to_english[filesize_multiplier])
params['filesize'] = filesize
except:
pass
# extract and convert creation date
try:
date_ts = result.xpath('.//td[5]/div/text()')[0]
date = datetime.fromtimestamp(float(date_ts))
params['publishedDate'] = date
except:
pass
# append result
results.append(params)
# return results sorted by seeder
return sorted(results, key=itemgetter('seed'), reverse=True)