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

display number of seeders and leechers in results

Browse Source 
fixes #1
This commit is contained in:
Viktor Stanchev 2013-02-26 16:48:02 -05:00
parent a507ce3b91
commit 6444e98b59
1 changed files with 33 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pirate-get.py
View File

@ -32,7 +32,7 @@ class MyHTMLParser(HTMLParser):
else: else:
self.state = 'looking' self.state = 'looking'
if self.state == 'magnet': if self.state == 'magnet':
self.results.append('magnet:?xt=urn:btih:' + urllib.quote(data) + '&dn=' + urllib.quote(self.title)) self.results.append(['magnet:?xt=urn:btih:' + urllib.quote(data) + '&dn=' + urllib.quote(self.title), '?', '?'])
self.state = 'looking' self.state = 'looking'
@ -49,9 +49,27 @@ def main():
htmlparser.feed(xml_str) htmlparser.feed(xml_str)
return htmlparser.results return htmlparser.results
#todo: redo this with html parser instead of regex
def remote(args): def remote(args):
f = urllib2.urlopen('http://thepiratebay.se/search/' + args.q.replace(" ", "+") + '/0/7/0') f = urllib2.urlopen('http://thepiratebay.se/search/' + args.q.replace(" ", "+") + '/0/7/0')
return re.findall(""""(magnet\:\?xt=[^"]*)""", f.read()) res = f.read()
found = re.findall(""""(magnet\:\?xt=[^"]*)|<td align="right">([^<]+)</td>""", res)
state = "seeds"
curr = ['',0,0] #magnet, seeds, leeches
res_l = []
for f in found:
if f[1] == '':
curr[0] = f[0]
else:
if state == 'seeds':
curr[1] = f[1]
state = 'leeches'
else:
curr[2] = f[1]
state = 'seeds'
res_l.append(curr)
curr = ['', 0, 0]
return res_l
args = parser.parse_args() args = parser.parse_args()
if args.database: if args.database:
@ -59,12 +77,22 @@ def main():
else: else:
mags = remote(args) mags = remote(args)
print "S=seeders"
print "L=leechers"
if mags: if mags:
for m in range(len(mags)): for m in range(len(mags)):
name = re.search("dn=([^\&]*)", mags[m]) magnet = mags[m]
print str(m) + ".", urllib.unquote(name.group(1).encode('ascii')).decode('utf-8').replace("+", " ") 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("+", " ")
l = raw_input("Select a link: ") l = raw_input("Select a link: ")
webbrowser.open(mags[int(l)]) try:
choice = int(l)
except Exception:
choice = None
if not choice == None:
webbrowser.open(mags[choice])
else:
print "Cancelled."
else: else:
print "no results" print "no results"