rnhmjoj/pirate-get
1
0
Fork 0
mirror of https://github.com/vikstrous/pirate-get synced 2025-01-10 10:04:21 +01:00
Code Issues Releases Wiki Activity

Enhanced output

Browse Source 
Added column titles, justified columns, S/L ratio and file size
This commit is contained in:
repic 2013-11-11 22:39:40 +01:00
parent a515098431
commit 3affc9171d
1 changed files with 46 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
pirate-get.py
View File

@ -60,10 +60,16 @@ def main():
except Exception: except Exception:
raise Exception("Please provide an integer greater than 0 for the number of pages to fetch.") raise Exception("Please provide an integer greater than 0 for the number of pages to fetch.")
# Catch the Ctrl-C exception and exit cleanly
try:
for page in xrange(pages): for page in xrange(pages):
f = urllib2.urlopen('http://thepiratebay.se/search/' + args.q.replace(" ", "+") + '/' + str(page) + '/7/0') f = urllib2.urlopen('http://thepiratebay.se/search/' + args.q.replace(" ", "+") + '/' + str(page) + '/7/0')
res = f.read() res = f.read()
found = re.findall(""""(magnet\:\?xt=[^"]*)|<td align="right">([^<]+)</td>""", res) found = re.findall(""""(magnet\:\?xt=[^"]*)|<td align="right">([^<]+)</td>""", res)
# get sizes as well and substitute the &nbsp; character
sizes = [ match.replace("&nbsp;", " ") for match in re.findall("(?<=Size )[0-9]+\.[0-9]+\&nbsp\;[KMGT]iB",res) ]
state = "seeds" state = "seeds"
curr = ['',0,0] #magnet, seeds, leeches curr = ['',0,0] #magnet, seeds, leeches
for f in found: for f in found:
@ -78,26 +84,43 @@ def main():
state = 'seeds' state = 'seeds'
res_l.append(curr) res_l.append(curr)
curr = ['', 0, 0] curr = ['', 0, 0]
return res_l except KeyboardInterrupt :
print "\nCancelled."
exit()
# return the sizes in a spearate list
return res_l, sizes
args = parser.parse_args() args = parser.parse_args()
if args.database: if args.database:
mags = local(args) mags = local(args)
else: else:
mags = remote(args) mags, sizes = remote(args)
if mags and len(mags) > 0: if mags and len(mags) > 0:
print "S=seeders" # enhanced print output with column titles
print "L=leechers" print "\n%-5s %-6s %-6s %-5s %-11s %s" % ( "LINK", "SEED", "LEECH", "RATIO", "SIZE", "NAME")
for m in range(len(mags)): for m in range(len(mags)):
magnet = mags[m] magnet = mags[m]
name = re.search("dn=([^\&]*)", magnet[0]) name = re.search("dn=([^\&]*)", magnet[0])
print str(m) + '. S:' + str(magnet[1]) + ' L:' + str(magnet[2]) + ' ', urllib.unquote(name.group(1).encode('ascii')).decode('utf-8').replace("+", " ")
# compute the S/L ratio (Higher is better)
ratio = float(magnet[1])/float(magnet[2])
# enhanced print output with justified columns
print "%-5s %-6s %-6s %5.1f %-11s %s" % (m, magnet[1], magnet[2], ratio ,sizes[m], urllib.unquote(name.group(1).encode('ascii')).decode('utf-8').replace("+", " ") )
try:
l = raw_input("Select a link: ") l = raw_input("Select a link: ")
except KeyboardInterrupt :
print "\nCancelled."
exit()
try: try:
choice = int(l) choice = int(l)
except Exception: except Exception:
choice = None choice = None
if not choice == None: if not choice == None:
webbrowser.open(mags[choice][0]) webbrowser.open(mags[choice][0])
else: else: