forked from Ponysearch/Ponysearch
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""Tokyo Toshokan (A BitTorrent Library for Japanese Media)
|
|
|
|
"""
|
|
|
|
import re
|
|
from datetime import datetime
|
|
from urllib.parse import urlencode
|
|
|
|
from lxml import html
|
|
from searx.utils import extract_text, int_or_zero
|
|
|
|
# about
|
|
about = {
|
|
"website": 'https://www.tokyotosho.info/',
|
|
"wikidata_id": None,
|
|
"official_api_documentation": None,
|
|
"use_official_api": False,
|
|
"require_api_key": False,
|
|
"results": 'HTML',
|
|
}
|
|
|
|
# engine dependent config
|
|
categories = ['files']
|
|
paging = True
|
|
|
|
# search-url
|
|
base_url = 'https://www.tokyotosho.info/'
|
|
search_url = base_url + 'search.php?{query}'
|
|
|
|
|
|
# do search-request
|
|
def request(query, params):
|
|
query = urlencode({'page': params['pageno'], 'terms': query})
|
|
params['url'] = search_url.format(query=query)
|
|
return params
|
|
|
|
|
|
# get response from search-request
|
|
def response(resp):
|
|
results = []
|
|
|
|
dom = html.fromstring(resp.text)
|
|
rows = dom.xpath('//table[@class="listing"]//tr[contains(@class, "category_0")]')
|
|
|
|
# check if there are no results or page layout was changed so we cannot parse it
|
|
# currently there are two rows for each result, so total count must be even
|
|
if len(rows) == 0 or len(rows) % 2 != 0:
|
|
return []
|
|
|
|
# regular expression for parsing torrent size strings
|
|
size_re = re.compile(r'[\d.]+(T|G|M)?B', re.IGNORECASE)
|
|
|
|
# processing the results, two rows at a time
|
|
for i in range(0, len(rows), 2):
|
|
# parse the first row
|
|
name_row = rows[i]
|
|
|
|
links = name_row.xpath('./td[@class="desc-top"]/a')
|
|
params = {'template': 'torrent.html', 'url': links[-1].attrib.get('href'), 'title': extract_text(links[-1])}
|
|
# I have not yet seen any torrents without magnet links, but
|
|
# it's better to be prepared to stumble upon one some day
|
|
if len(links) == 2:
|
|
magnet = links[0].attrib.get('href')
|
|
if magnet.startswith('magnet'):
|
|
# okay, we have a valid magnet link, let's add it to the result
|
|
params['magnetlink'] = magnet
|
|
|
|
# no more info in the first row, start parsing the second one
|
|
info_row = rows[i + 1]
|
|
desc = extract_text(info_row.xpath('./td[@class="desc-bot"]')[0])
|
|
for item in desc.split('|'):
|
|
item = item.strip()
|
|
if item.startswith('Size:'):
|
|
try:
|
|
params['filesize'] = size_re.search(item).group()
|
|
except: # pylint: disable=bare-except
|
|
pass
|
|
elif item.startswith('Date:'):
|
|
try:
|
|
# Date: 2016-02-21 21:44 UTC
|
|
date = datetime.strptime(item, 'Date: %Y-%m-%d %H:%M UTC')
|
|
params['publishedDate'] = date
|
|
except: # pylint: disable=bare-except
|
|
pass
|
|
elif item.startswith('Comment:'):
|
|
params['content'] = item
|
|
stats = info_row.xpath('./td[@class="stats"]/span')
|
|
# has the layout not changed yet?
|
|
if len(stats) == 3:
|
|
params['seed'] = int_or_zero(extract_text(stats[0]))
|
|
params['leech'] = int_or_zero(extract_text(stats[1]))
|
|
|
|
results.append(params)
|
|
|
|
return results
|