324 lines
9.7 KiB
Python
324 lines
9.7 KiB
Python
import urllib
|
|
from enum import Enum
|
|
from urllib.parse import urlencode
|
|
|
|
from lxml import etree
|
|
|
|
|
|
class TorrentSite(Enum):
|
|
"""
|
|
Contains torrent sites
|
|
"""
|
|
|
|
NYAASI = "https://nyaa.si"
|
|
SUKEBEINYAASI = "https://sukebei.nyaa.si"
|
|
|
|
# * nyaa.pantsu.cat redirects to nyaa.net
|
|
NYAANET = "https://nyaa.net"
|
|
SUKEBEINYAANET = "https://sukebei.nyaa.net"
|
|
|
|
|
|
def nyaa_categories(b):
|
|
c = b.replace("?c=", "")
|
|
cats = c.split("_")
|
|
|
|
cat = cats[0]
|
|
sub_cat = cats[1]
|
|
|
|
categories = {
|
|
"1": {
|
|
"name": "Anime",
|
|
"sub_cats": {
|
|
"1": "Anime Music Video",
|
|
"2": "English-translated",
|
|
"3": "Non-English-translated",
|
|
"4": "Raw",
|
|
},
|
|
},
|
|
"2": {"name": "Audio", "sub_cats": {"1": "Lossless", "2": "Lossy"}},
|
|
"3": {
|
|
"name": "Literature",
|
|
"sub_cats": {
|
|
"1": "English-translated",
|
|
"2": "Non-English-translated",
|
|
"3": "Raw",
|
|
},
|
|
},
|
|
"4": {
|
|
"name": "Live Action",
|
|
"sub_cats": {
|
|
"1": "English-translated",
|
|
"2": "Idol/Promotional Video",
|
|
"3": "Non-English-translated",
|
|
"4": "Raw",
|
|
},
|
|
},
|
|
"5": {"name": "Pictures", "sub_cats": {"1": "Graphics", "2": "Photos"}},
|
|
"6": {"name": "Software", "sub_cats": {"1": "Applications", "2": "Games"}},
|
|
}
|
|
|
|
try:
|
|
category_name = (
|
|
f"{categories[cat]['name']} - {categories[cat]['sub_cats'][sub_cat]}"
|
|
)
|
|
except KeyError:
|
|
print("Unable to get Nyaa category name")
|
|
return
|
|
|
|
return category_name
|
|
|
|
|
|
def parse_nyaa_rss(request_text, limit, site):
|
|
"""
|
|
Extracts torrent information from a given rss response.
|
|
"""
|
|
root = etree.fromstring(request_text)
|
|
torrents = []
|
|
|
|
for item in root.xpath("channel/item")[:limit]:
|
|
# Decide category.
|
|
if site in [TorrentSite.NYAASI, TorrentSite.NYAANET]:
|
|
category = item.findtext("nyaa:categoryId", namespaces=item.nsmap)
|
|
elif site in [TorrentSite.SUKEBEINYAASI, TorrentSite.SUKEBEINYAANET]:
|
|
category = item.findtext("nyaa:categoryId", namespaces=item.nsmap)
|
|
else:
|
|
raise ValueError("Unknown TorrentSite received!")
|
|
|
|
try:
|
|
is_remake = item.findtext("nyaa:remake", namespaces=item.nsmap) == "Yes"
|
|
is_trusted = item.findtext("nyaa:trusted", namespaces=item.nsmap) == "Yes"
|
|
item_type = (
|
|
"remake" if is_remake else "trusted" if is_trusted else "default"
|
|
)
|
|
|
|
torrent = {
|
|
"id": item.findtext("guid").split("/")[-1],
|
|
"category": category,
|
|
"url": item.findtext("guid"),
|
|
"name": item.findtext("title"),
|
|
"download_url": item.findtext("link"),
|
|
"magnet": magnet_builder(
|
|
item.findtext("nyaa:infoHash", namespaces=item.nsmap),
|
|
item.findtext("title"),
|
|
),
|
|
"size": item.findtext("nyaa:size", namespaces=item.nsmap),
|
|
"date": item.findtext("pubDate"),
|
|
"seeders": item.findtext("nyaa:seeders", namespaces=item.nsmap),
|
|
"leechers": item.findtext("nyaa:leechers", namespaces=item.nsmap),
|
|
"completed_downloads": None,
|
|
"type": item_type,
|
|
}
|
|
torrents.append(torrent)
|
|
except IndexError:
|
|
pass
|
|
|
|
return torrents
|
|
|
|
|
|
def parse_nyaa(request_text, limit, site):
|
|
parser = etree.HTMLParser()
|
|
tree = etree.fromstring(request_text, parser)
|
|
|
|
# Put proper domain here.
|
|
uri = site.value
|
|
|
|
torrents = []
|
|
|
|
# Going through table rows
|
|
for tr in tree.xpath("//tbody//tr")[:limit]:
|
|
block = []
|
|
|
|
for td in tr.xpath("./td"):
|
|
for link in td.xpath("./a"):
|
|
|
|
href = link.attrib.get("href").split("/")[-1]
|
|
|
|
# Only caring about non-comment pages.
|
|
if href[-9:] != "#comments":
|
|
block.append(href)
|
|
|
|
if link.text and link.text.strip():
|
|
block.append(link.text.strip())
|
|
|
|
if td.text is not None and td.text.strip():
|
|
block.append(td.text.strip())
|
|
|
|
# Add type of torrent based on tr class.
|
|
if tr.attrib.get("class") is not None:
|
|
if "danger" in tr.attrib.get("class"):
|
|
block.append("remake")
|
|
elif "success" in tr.attrib.get("class"):
|
|
block.append("trusted")
|
|
else:
|
|
block.append("default")
|
|
else:
|
|
block.append("default")
|
|
|
|
# Decide category.
|
|
if site in [TorrentSite.NYAASI, TorrentSite.NYAANET]:
|
|
category = nyaa_categories(block[0])
|
|
elif site in [TorrentSite.SUKEBEINYAASI, TorrentSite.SUKEBEINYAANET]:
|
|
category = sukebei_categories(block[0])
|
|
else:
|
|
raise ValueError("Unknown TorrentSite received!")
|
|
|
|
# Create torrent object
|
|
try:
|
|
torrent = {
|
|
"id": block[1],
|
|
"category": category,
|
|
"url": "{}/view/{}".format(uri, block[1]),
|
|
"name": block[2],
|
|
"download_url": "{}/download/{}".format(uri, block[3]),
|
|
"magnet": block[4],
|
|
"size": block[5],
|
|
"date": block[6],
|
|
"seeders": block[7],
|
|
"leechers": block[8],
|
|
"completed_downloads": block[9],
|
|
"type": block[10],
|
|
}
|
|
torrents.append(torrent)
|
|
except IndexError:
|
|
pass
|
|
return torrents
|
|
|
|
|
|
def parse_single(request_text, site):
|
|
parser = etree.HTMLParser()
|
|
tree = etree.fromstring(request_text, parser)
|
|
|
|
# Put proper domain here.
|
|
uri = site.value
|
|
|
|
torrent = {}
|
|
data = []
|
|
torrent_files = []
|
|
|
|
# Find basic uploader info & torrent stats
|
|
for row in tree.xpath("//div[@class='row']"):
|
|
for div_text in row.xpath("./div[@class='col-md-5']//text()"):
|
|
d = div_text.strip()
|
|
if d:
|
|
data.append(d)
|
|
|
|
# Find files, we need only text of the li element(s).
|
|
# Sorry about Pycodestyle aka PEP8 (E501) error
|
|
for el in tree.xpath("//div[contains(@class, 'torrent-file-list')]//li/text()"):
|
|
if el.rstrip():
|
|
torrent_files.append(el)
|
|
|
|
torrent["title"] = tree.xpath("//h3[@class='panel-title']/text()")[0].strip()
|
|
torrent["category"] = data[0]
|
|
torrent["uploader"] = data[4]
|
|
torrent["uploader_profile"] = "{}/user/{}".format(uri, data[4])
|
|
torrent["website"] = data[6]
|
|
torrent["size"] = data[8]
|
|
torrent["date"] = data[3]
|
|
torrent["seeders"] = data[5]
|
|
torrent["leechers"] = data[7]
|
|
torrent["completed"] = data[9]
|
|
torrent["hash"] = data[10]
|
|
torrent["files"] = torrent_files
|
|
|
|
torrent["description"] = ""
|
|
for s in tree.xpath("//div[@id='torrent-description']"):
|
|
torrent["description"] += s.text
|
|
|
|
return torrent
|
|
|
|
|
|
def sukebei_categories(b):
|
|
c = b.replace("?c=", "")
|
|
cats = c.split("_")
|
|
|
|
cat = cats[0]
|
|
subcat = cats[1]
|
|
|
|
categories = {
|
|
"1": {
|
|
"name": "Art",
|
|
"subcats": {
|
|
"1": "Anime",
|
|
"2": "Doujinshi",
|
|
"3": "Games",
|
|
"4": "Manga",
|
|
"5": "Pictures",
|
|
},
|
|
},
|
|
"2": {
|
|
"name": "Real Life",
|
|
"subcats": {"1": "Photobooks & Pictures", "2": "Videos"},
|
|
},
|
|
}
|
|
|
|
try:
|
|
category_name = (
|
|
f"{categories[cat]['name']} - {categories[cat]['subcats'][subcat]}"
|
|
)
|
|
except KeyError:
|
|
print("Unable to get Sukebei category name")
|
|
return
|
|
|
|
return category_name
|
|
|
|
|
|
def magnet_builder(info_hash, title):
|
|
"""
|
|
Generates a magnet link using the info_hash and title of a given file.
|
|
"""
|
|
known_trackers = [
|
|
"http://nyaa.tracker.wf:7777/announce",
|
|
"udp://open.stealth.si:80/announce",
|
|
"udp://tracker.opentrackr.org:1337/announce",
|
|
"udp://exodus.desync.com:6969/announce",
|
|
"udp://tracker.torrent.eu.org:451/announce",
|
|
]
|
|
|
|
magnet_link = f"magnet:?xt=urn:btih:{info_hash}&" + urlencode(
|
|
{"dn": title}, quote_via=urllib.parse.quote
|
|
)
|
|
for tracker in known_trackers:
|
|
magnet_link += f"&{urlencode({'tr': tracker})}"
|
|
|
|
return magnet_link
|
|
|
|
|
|
# Pantsu Utils
|
|
def query_builder(q, params):
|
|
available_params = [
|
|
"category",
|
|
"page",
|
|
"limit",
|
|
"userID",
|
|
"fromID",
|
|
"status",
|
|
"maxage",
|
|
"toDate",
|
|
"fromDate",
|
|
"dateType",
|
|
"minSize",
|
|
"maxSize",
|
|
"sizeType",
|
|
"sort",
|
|
"order",
|
|
"lang",
|
|
]
|
|
query = "?q={}".format(q.replace(" ", "+"))
|
|
|
|
for param, value in params.items():
|
|
if param in available_params:
|
|
if param != "category" and param != "status" and param != "lang":
|
|
query += "&{}={}".format(param, value)
|
|
elif param == "category":
|
|
query += "&c={}_{}".format(value[0], value[1])
|
|
|
|
elif param == "status":
|
|
query += "&s={}".format(value)
|
|
|
|
elif param == "lang":
|
|
for lang in value:
|
|
query += "&lang={}".format(lang)
|
|
|
|
return query
|