From 8560a2f56b84cd2718106af5b1bfd9346b1f0272 Mon Sep 17 00:00:00 2001 From: Lance Allen Date: Sat, 21 May 2016 23:45:19 -0700 Subject: [PATCH] updating the terminology for arguments to make more sense --- thepirate.py | 21 ++++++++------------- 1 file changed, 8 insertions(+), 13 deletions(-) diff --git a/thepirate.py b/thepirate.py index 55ea835..1da8789 100755 --- a/thepirate.py +++ b/thepirate.py @@ -35,7 +35,11 @@ tpb = "https://thepiratebay.se" # Torrent server IP; can be any machine running transmission-daemon # 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 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('--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) @@ -97,7 +101,6 @@ def Get_Search_URL(): 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_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 @@ -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 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 args.arg_take_top and tpb_torrent_links: 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 tpb_search_results.update({number:link}) #Append results to tpb_search_results dictionary print "({}) {}".format(number, path.basename(link)) - if tpb_search_results: #If dict is not empty, continue with script print "\n(98) Search again" 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 """ - tpb_magnet_links = [] - tpb_torrent_page = requests.get(tpb_torrent_url, verify=False) tpb_torrent_page_soup = bs4.BeautifulSoup(tpb_torrent_page.content, "html.parser") - + for link in tpb_torrent_page_soup.find_all('a'): if str(link.get('href')).startswith('magnet:?xt'): tpb_magnet_links.append(link.get('href')) - + tpb_magnet_link = tpb_magnet_links[0] - print "\n[+] Adding magnet link for torrent:\n\n{}".format(tpb_torrent_url) - transmissionrpc.Client(rpcserver).add_torrent(tpb_magnet_link) - print "\n[.] Done!\n" - exit(0) if __name__ == "__main__":