mirror of
https://github.com/vikstrous/pirate-get
synced 2025-01-10 10:04:21 +01:00
128 lines
4.6 KiB
Python
128 lines
4.6 KiB
Python
import builtins
|
|
import re
|
|
import os
|
|
import gzip
|
|
import colorama
|
|
import urllib.parse as parse
|
|
import urllib.request as request
|
|
from io import BytesIO
|
|
|
|
import pirate.data
|
|
|
|
|
|
def print(*args, **kwargs):
|
|
if kwargs.get('color', False) and pirate.data.colored_output:
|
|
colorama.init()
|
|
color_dict = {
|
|
'default': '',
|
|
'header': colorama.Back.BLACK + colorama.Fore.WHITE,
|
|
'alt': colorama.Fore.YELLOW,
|
|
'zebra_0': '',
|
|
'zebra_1': colorama.Fore.BLUE,
|
|
'WARN': colorama.Fore.MAGENTA,
|
|
'ERROR': colorama.Fore.RED}
|
|
|
|
c = color_dict[kwargs.pop('color')]
|
|
args = (c + args[0],) + args[1:] + (colorama.Style.RESET_ALL,)
|
|
kwargs.pop('color', None)
|
|
return builtins.print(*args, **kwargs)
|
|
else:
|
|
kwargs.pop('color', None)
|
|
return builtins.print(*args, **kwargs)
|
|
|
|
|
|
def search_results(mags, sizes, uploaded, local):
|
|
columns = int(os.popen('stty size', 'r').read().split()[1])
|
|
cur_color = 'zebra_0'
|
|
|
|
if local:
|
|
print('{:>4} {:{length}}'.format(
|
|
'LINK', 'NAME', length=columns - 8),
|
|
color='header')
|
|
else:
|
|
print('{:>4} {:>5} {:>5} {:>5} {:9} {:11} {:{length}}'.format(
|
|
'LINK', 'SEED', 'LEECH', 'RATIO',
|
|
'SIZE', 'UPLOAD', 'NAME', length=columns - 52),
|
|
color='header')
|
|
|
|
for m, magnet in enumerate(mags):
|
|
# Alternate between colors
|
|
cur_color = 'zebra_0' if cur_color == 'zebra_1' else 'zebra_1'
|
|
|
|
name = re.search(r'dn=([^\&]*)', magnet[0])
|
|
torrent_name = parse.unquote(name.group(1)).replace('+', ' ')
|
|
|
|
if local:
|
|
line = '{:5} {:{length}}'
|
|
content = [m, torrent_name[:columns]]
|
|
else:
|
|
no_seeders, no_leechers = map(int, magnet[1:])
|
|
size, unit = (float(sizes[m][0]),
|
|
sizes[m][1]) if sizes else (0, '???')
|
|
date = uploaded[m]
|
|
|
|
# compute the S/L ratio (Higher is better)
|
|
try:
|
|
ratio = no_seeders / no_leechers
|
|
except ZeroDivisionError:
|
|
ratio = float('inf')
|
|
|
|
line = ('{:4} {:5} {:5} {:5.1f} {:5.1f}'
|
|
' {:3} {:<11} {:{length}}')
|
|
content = [m, no_seeders, no_leechers, ratio,
|
|
size, unit, date, torrent_name[:columns - 52]]
|
|
|
|
# enhanced print output with justified columns
|
|
print(line.format(*content, length=columns - 52), color=cur_color)
|
|
|
|
|
|
def descriptions(chosen_links, mags, site, identifiers):
|
|
for link in chosen_links:
|
|
link = int(link)
|
|
path = '/torrent/%s/' % identifiers[link]
|
|
req = request.Request(site + path, headers=pirate.data.default_headers)
|
|
req.add_header('Accept-encoding', 'gzip')
|
|
f = request.urlopen(req, timeout=pirate.data.default_timeout)
|
|
|
|
if f.info().get('Content-Encoding') == 'gzip':
|
|
f = gzip.GzipFile(fileobj=BytesIO(f.read()))
|
|
|
|
res = f.read().decode('utf-8')
|
|
name = re.search(r'dn=([^\&]*)', mags[link][0])
|
|
torrent_name = parse.unquote(name.group(1)).replace('+', ' ')
|
|
desc = re.search(r'<div class="nfo">\s*<pre>(.+?)(?=</pre>)',
|
|
res, re.DOTALL).group(1)
|
|
|
|
# Replace HTML links with markdown style versions
|
|
desc = re.sub(r'<a href="\s*([^"]+?)\s*"[^>]*>(\s*)([^<]+?)(\s*'
|
|
r')</a>', r'\2[\3](\1)\4', desc)
|
|
|
|
print('Description for "%s":' % torrent_name, color='zebra_1')
|
|
print(desc, color='zebra_0')
|
|
|
|
|
|
def file_lists(chosen_links, mags, site, identifiers):
|
|
for link in chosen_links:
|
|
path = '/ajax_details_filelist.php'
|
|
query = '?id=' + identifiers[int(link)]
|
|
req = request.Request(site + path + query,
|
|
headers=pirate.data.default_headers)
|
|
req.add_header('Accept-encoding', 'gzip')
|
|
f = request.urlopen(req, timeout=pirate.data.default_timeout)
|
|
|
|
if f.info().get('Content-Encoding') == 'gzip':
|
|
f = gzip.GzipFile(fileobj=BytesIO(f.read()))
|
|
|
|
res = f.read().decode('utf-8').replace(' ', ' ')
|
|
files = re.findall(r'<td align="left">\s*([^<]+?)\s*</td><td ali'
|
|
r'gn="right">\s*([^<]+?)\s*</tr>', res)
|
|
name = re.search(r'dn=([^\&]*)', mags[int(link)][0])
|
|
torrent_name = parse.unquote(name.group(1)).replace('+', ' ')
|
|
|
|
print('Files in "%s":' % torrent_name, color='zebra_1')
|
|
cur_color = 'zebra_0'
|
|
|
|
for f in files:
|
|
print('{0[0]:>11} {0[1]}'.format(f), color=cur_color)
|
|
cur_color = 'zebra_0' if (cur_color == 'zebra_1') else 'zebra_1'
|