\s*
(.+?)(?=
)',
+ res, re.DOTALL).group(1)
+
+ # Replace HTML links with markdown style versions
+ desc = re.sub(r'
]*>(\s*)([^<]+?)(\s*'
+ r')', r'\2[\3](\1)\4', desc)
+
+ print('Description for "%s":' % torrent_name, color='zebra_1')
+ print(desc, color='zebra_0')
+
+
+def print_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=default_headers)
+ req.add_header('Accept-encoding', 'gzip')
+ f = request.urlopen(req, timeout=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'
\s*([^<]+?)\s* | \s*([^<]+?)\s*', 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'
+
+
+def save_torrents(chosen_links, mags, folder):
+ for link in chosen_links:
+ magnet = mags[int(link)][0]
+ name = re.search(r'dn=([^\&]*)', magnet)
+ torrent_name = parse.unquote(name.group(1)).replace('+', ' ')
+ info_hash = int(re.search(r'btih:([a-f0-9]{40})', magnet).group(1), 16)
+ file = os.path.join(folder, torrent_name + '.torrent')
+
+ try:
+ torrent = get_torrent(info_hash)
+ except HTTPError:
+ print('There is no cached file for this torrent :(', color='ERROR')
+ else:
+ open(file,'wb').write(torrent)
+ print('Saved {:X} in {}'.format(info_hash, file))
+
+
+def save_magnets(chosen_links, mags, folder):
+ for link in chosen_links:
+ magnet = mags[int(link)][0]
+ name = re.search(r'dn=([^\&]*)', magnet)
+ torrent_name = parse.unquote(name.group(1)).replace('+', ' ')
+ info_hash = int(re.search(r'btih:([a-f0-9]{40})', magnet).group(1), 16)
+ file = os.path.join(folder, torrent_name + '.magnet')
+
+ print('Saved {:X} in {}'.format(info_hash, file))
+ with open(file, 'w') as f:
+ f.write(magnet + '\n')
+
+
+def main():
+ config = load_config()
+
+ parser = argparse.ArgumentParser(
+ description='finds and downloads torrents from the Pirate Bay')
+ parser.add_argument('-b', dest='browse',
+ action='store_true',
+ help='display in Browse mode')
+ parser.add_argument('search', metavar='search',
+ nargs='*', help='term to search for')
+ parser.add_argument('-c', dest='category', metavar='category',
+ help='specify a category to search', default='All')
+ parser.add_argument('-s', dest='sort', metavar='sort',
+ help='specify a sort option', default='SeedersDsc')
+ parser.add_argument('-R', dest='recent', action='store_true',
+ help='torrents uploaded in the last 48hours.'
+ '*ignored in searches*')
+ parser.add_argument('-l', dest='list_categories',
+ action='store_true',
+ help='list categories')
+ parser.add_argument('--list_sorts', dest='list_sorts',
+ action='store_true',
+ help='list Sortable Types')
+ parser.add_argument('-L', '--local', dest='database',
+ help='an xml file containing the Pirate Bay database')
+ parser.add_argument('-p', dest='pages', default=1,
+ help='the number of pages to fetch '
+ "(doesn't work with --local)")
+ parser.add_argument('-0', dest='first',
+ action='store_true',
+ help='choose the top result')
+ parser.add_argument('-a', '--download-all',
+ action='store_true',
+ help='download all results')
+ parser.add_argument('-t', '--transmission',
+ action='store_true',
+ help='open magnets with transmission-remote')
+ parser.add_argument('-P', '--port', dest='port',
+ help='transmission-remote rpc port. default is 9091')
+ parser.add_argument('-C', '--custom', dest='command',
+ help='open magnets with a custom command'
+ ' (%%s will be replaced with the url)')
+ parser.add_argument('-M', '--save-magnets',
+ action='store_true',
+ help='save magnets links as files')
+ parser.add_argument('-T', '--save-torrents',
+ action='store_true',
+ help='save torrent files')
+ parser.add_argument('-S', '--save-directory',
+ type=str, metavar='DIRECTORY',
+ help='directory where to save downloaded files'
+ ' (if none is given $PWD will be used)')
+ parser.add_argument('--disable-colors', dest='color',
+ action='store_false',
+ help='disable colored output')
+ args = parser.parse_args()
+
+ if (config.getboolean('Misc', 'colors') and not args.color
+ or not config.getboolean('Misc', 'colors')):
+ global colored_output
+ colored_output = False
+
+ if args.save_directory:
+ config.set('Save', 'directory', args.save_directory)
+
+ transmission_command = ['transmission-remote']
+ if args.port:
+ transmission_command.append(args.port)
+
+ if args.transmission or config.getboolean('Misc', 'transmission'):
+ ret = subprocess.call(transmission_command + ['-l'],
+ stdout=subprocess.DEVNULL,
+ stderr=subprocess.DEVNULL)
+ if ret != 0:
+ print('Transmission is not running.')
+ return
+
+ if args.list_categories:
+ cur_color = 'zebra_0'
+ for key, value in sorted(categories.items()) :
+ cur_color = 'zebra_0' if cur_color == 'zebra_1' else 'zebra_1'
+ print(str(value), '\t', key, sep='', color=cur_color)
+ return
+
+ if args.list_sorts:
+ cur_color = 'zebra_0'
+ for key, value in sorted(sorts.items()):
+ cur_color = 'zebra_0' if cur_color == 'zebra_1' else 'zebra_1'
+ print(str(value), '\t', key, sep='', color=cur_color)
+ return
+
+ if args.database or config.getboolean('LocalDB', 'enabled'):
+ if args.database:
+ path = args.database
+ else:
+ path = config.get('LocalDB', 'path')
+ mags = local(path, args.search)
+ sizes, uploaded = [], []
+
+ else:
+ mags, mirrors = [], {'https://thepiratebay.se'}
+ try:
+ req = request.Request('https://proxybay.co/list.txt',
+ headers=default_headers)
+ f = request.urlopen(req, timeout=default_timeout)
+ except IOError:
+ print('Could not fetch additional mirrors', color='WARN')
+ else:
+ if f.getcode() != 200:
+ raise IOError('The proxy bay responded with an error.')
+ mirrors = mirrors.union([i.decode('utf-8').strip()
+ for i in f.readlines()][3:])
+
+ for mirror in mirrors:
+ try:
+ print('Trying', mirror, end='... ')
+ mags, sizes, uploaded, identifiers = remote(args, mirror)
+ except (URLError, IOError, ValueError, timeout):
+ print('Failed', color='WARN')
+ else:
+ site = mirror
+ print('Ok', color='alt')
+ break
+ else:
+ print('No available mirrors :(', color='WARN')
+ return
+
+ if not mags:
+ print('No results')
+ return
+
+ print_search_results(mags, sizes, uploaded, local=args.database)
+
+ if args.first:
+ print('Choosing first result')
+ choices = [0]
+ elif args.download_all:
+ print('Downloading all results')
+ choices = range(len(mags))
+ else:
+ # New input loop to support different link options
+ while True:
+ print("\nSelect links (Type 'h' for more options"
+ ", 'q' to quit)", end='\b', color='alt')
+ try:
+ l=input(': ')
+ except KeyboardInterrupt :
+ print('\nCancelled.')
+ return
+
+ try:
+ # Very permissive handling
+ # Check for any occurances or d, f, p, t, m, or q
+ cmd_code_match = re.search(r'([hdfpmtq])', l,
+ flags=re.IGNORECASE)
+ if cmd_code_match:
+ code = cmd_code_match.group(0).lower()
+ else:
+ code = None
+
+ # Clean up command codes
+ # Substitute multiple consecutive spaces/commas for single
+ # comma remove anything that isn't an integer or comma.
+ # Turn into list
+ l = re.sub(r'^[hdfp, ]*|[hdfp, ]*$', '', l)
+ l = re.sub('[ ,]+', ',', l)
+ l = re.sub('[^0-9,]', '', l)
+ choices = l.split(',')
+
+ # Act on option, if supplied
+ print('')
+ if code == 'h':
+ print('Options:',
+ ': Download selected torrents',
+ '[m]: Save magnets as files',
+ '[t]: Save .torrent files',
+ '[d]: Get descriptions',
+ '[f]: Get files',
+ '[p] Print search results',
+ '[q] Quit', sep='\n')
+ elif code == 'q':
+ print('Bye.', color='alt')
+ return
+ elif code == 'd':
+ print_descriptions(choices, mags, site, identifiers)
+ elif code == 'f':
+ print_file_lists(choices, mags, site, identifiers)
+ elif code == 'p':
+ print_search_results(mags, sizes, uploaded)
+ elif code == 'm':
+ save_magnets(choices, mags,
+ config.get('Save', 'directory'))
+ elif code == 't':
+ save_torrents(choices, mags,
+ config.get('Save', 'directory'))
+ elif not l:
+ print('No links entered!', color='WARN')
+ else:
+ break
+ except Exception as e:
+ print('Exception:', e, color='ERROR')
+ choices = ()
+
+ save_to_file = False
+
+ if args.save_magnets or config.getboolean('Save', 'magnets'):
+ print('Saving selected magnets...')
+ save_magnets(choices, mags, config.get('Save', 'directory'))
+ save_to_file = True
+
+ if args.save_torrents or config.getboolean('Save', 'torrents'):
+ print('Saving selected torrents...')
+ save_torrents(choices, mags, config.get('Save', 'directory'))
+ save_to_file = True
+
+ if save_to_file:
+ return
+
+ for choice in choices:
+ url = mags[int(choice)][0]
+
+ if args.transmission or config.getboolean('Misc', 'transmission'):
+ subprocess.call(transmission_command + ['-l', '--add', url], shell=False)
+ subprocess.call(transmission_command + ['-l'])
+
+ elif args.command or config.get('Misc', 'openCommand'):
+ command = config.get('Misc', 'openCommand')
+ if args.command:
+ command = args.command
+ subprocess.call(parse_cmd(command, url), shell=False)
+
+ else:
+ webbrowser.open(url)
+
+
+if __name__ == '__main__':
+ main()
|