updating the terminology for arguments to make more sense

master
Lance Allen 8 years ago
parent 157c6b1d68
commit 8560a2f56b

@ -35,7 +35,11 @@ tpb = "https://thepiratebay.se"
# Torrent server IP; can be any machine running transmission-daemon # Torrent server IP; can be any machine running transmission-daemon
# with a firewall inbound allowed to TCP/9091 (transmissionrpc) # with a firewall inbound allowed to TCP/9091 (transmissionrpc)
rpcserver = 'localhost' rpcserver = ''
if not rpcserver:
print "[!] Transmission server not set! Update the 'rpcserver' variable in the script"
exit(10)
# Squelch HTTPS insecure warnings # Squelch HTTPS insecure warnings
requests.packages.urllib3.disable_warnings() requests.packages.urllib3.disable_warnings()
@ -49,7 +53,7 @@ parser.add_argument('--search', '-s', dest='arg_search_string', help='The string
parser.add_argument('--top', '-t', dest='arg_take_top', action='store_true', help='Automatically grab the torrent with most seeds', required=False) parser.add_argument('--top', '-t', dest='arg_take_top', action='store_true', help='Automatically grab the torrent with most seeds', required=False)
parser.add_argument('--file', '-l', dest='arg_magnet_link', help='Direct link to magnet link or torrent file', required=False) parser.add_argument('--magnet', '-m', dest='arg_magnet_link', help='Direct link to magnet link or torrent file', required=False)
parser.add_argument('--url', '-u', dest='arg_torrent_page', help='URL of the torrent file', required=False) parser.add_argument('--url', '-u', dest='arg_torrent_page', help='URL of the torrent file', required=False)
@ -97,7 +101,6 @@ def Get_Search_URL():
tpb_search_string = raw_input("[+] What would you like to search?\n>>> ") tpb_search_string = raw_input("[+] What would you like to search?\n>>> ")
tpb_search_url = "{}/search/{}/0/7/0".format(tpb, tpb_search_string) #/0/7/0 tells TPB to sort descending by seeds tpb_search_url = "{}/search/{}/0/7/0".format(tpb, tpb_search_string) #/0/7/0 tells TPB to sort descending by seeds
tpb_torrent_page_source = requests.get(tpb_search_url, verify=False).text #Use requests lib to fetch page source for bs4 parsing tpb_torrent_page_source = requests.get(tpb_search_url, verify=False).text #Use requests lib to fetch page source for bs4 parsing
Get_Torrent_Links(tpb_torrent_page_source) #Run Get_Torrent_Links function, passing page source for BS4 parsing Get_Torrent_Links(tpb_torrent_page_source) #Run Get_Torrent_Links function, passing page source for BS4 parsing
@ -118,7 +121,6 @@ def Get_Torrent_Links(source):
if link.get('href').startswith('/torrent'): #Only get links with /torrent as they're valid torrent pages if link.get('href').startswith('/torrent'): #Only get links with /torrent as they're valid torrent pages
tpb_torrent_links.append(link.get('href')) #Set the results to tpb_torrent_links array tpb_torrent_links.append(link.get('href')) #Set the results to tpb_torrent_links array
#If -t is supplied, bypass this section of code and go on to download the top torrent #If -t is supplied, bypass this section of code and go on to download the top torrent
if args.arg_take_top and tpb_torrent_links: if args.arg_take_top and tpb_torrent_links:
Download_Torrent_From_URL("{}/{}".format(tpb, tpb_torrent_links[0])) Download_Torrent_From_URL("{}/{}".format(tpb, tpb_torrent_links[0]))
@ -127,7 +129,6 @@ def Get_Torrent_Links(source):
for number,link in enumerate(tpb_torrent_links): #Enumerate the array so the numbers start at 0 for number,link in enumerate(tpb_torrent_links): #Enumerate the array so the numbers start at 0
tpb_search_results.update({number:link}) #Append results to tpb_search_results dictionary tpb_search_results.update({number:link}) #Append results to tpb_search_results dictionary
print "({}) {}".format(number, path.basename(link)) print "({}) {}".format(number, path.basename(link))
if tpb_search_results: #If dict is not empty, continue with script if tpb_search_results: #If dict is not empty, continue with script
print "\n(98) Search again" print "\n(98) Search again"
print "(99) Exit" print "(99) Exit"
@ -178,24 +179,18 @@ def Download_Torrent_From_URL(tpb_torrent_url):
""" """
Grabs the first magnet link and adds it to the queue via RPC to rpcserver Grabs the first magnet link and adds it to the queue via RPC to rpcserver
""" """
tpb_magnet_links = [] tpb_magnet_links = []
tpb_torrent_page = requests.get(tpb_torrent_url, verify=False) tpb_torrent_page = requests.get(tpb_torrent_url, verify=False)
tpb_torrent_page_soup = bs4.BeautifulSoup(tpb_torrent_page.content, "html.parser") tpb_torrent_page_soup = bs4.BeautifulSoup(tpb_torrent_page.content, "html.parser")
for link in tpb_torrent_page_soup.find_all('a'): for link in tpb_torrent_page_soup.find_all('a'):
if str(link.get('href')).startswith('magnet:?xt'): if str(link.get('href')).startswith('magnet:?xt'):
tpb_magnet_links.append(link.get('href')) tpb_magnet_links.append(link.get('href'))
tpb_magnet_link = tpb_magnet_links[0] tpb_magnet_link = tpb_magnet_links[0]
print "\n[+] Adding magnet link for torrent:\n\n{}".format(tpb_torrent_url) print "\n[+] Adding magnet link for torrent:\n\n{}".format(tpb_torrent_url)
transmissionrpc.Client(rpcserver).add_torrent(tpb_magnet_link) transmissionrpc.Client(rpcserver).add_torrent(tpb_magnet_link)
print "\n[.] Done!\n" print "\n[.] Done!\n"
exit(0) exit(0)
if __name__ == "__main__": if __name__ == "__main__":

Loading…
Cancel
Save