KevinMidboe/torrent_search
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 1 Wiki Activity

Now we have a option to filter on high quality releases with the -f flag.

Browse Source
This commit is contained in:
Kevin
2018-02-26 18:19:54 +01:00
parent bf9ed49a86
commit 4996f20bbc
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
torrentSearch/search.py
View File

@@ -4,7 +4,7 @@
"""Torrent Search. """Torrent Search.
Usage: Usage:
search.py <query> [-s <site>] [-p | --print] [--debug | --warning | --error] search.py <query> [-s <site>] [-f] [-p | --print] [--debug | --warning | --error]
search.py (-h | --help) search.py (-h | --help)
search.py --version search.py --version
@@ -12,6 +12,7 @@ Options:
-h --help Show this screen -h --help Show this screen
-s [site] Site to index [default: piratebay] (piratebay|jackett) -s [site] Site to index [default: piratebay] (piratebay|jackett)
-p --print Print result to console -p --print Print result to console
-f Filter response on release type
--version Show version --version Show version
--debug Print all debug logs --debug Print all debug logs
--warning Print only logged warnings --warning Print only logged warnings
@@ -34,6 +35,8 @@ from torrentSearch.jackett import Jackett
from torrentSearch.piratebay import Piratebay from torrentSearch.piratebay import Piratebay
from torrentSearch.utils import ColorizeFilter from torrentSearch.utils import ColorizeFilter
from pprint import pprint
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) BASE_DIR = os.path.dirname(os.path.abspath(__file__))
ch = logging.StreamHandler() ch = logging.StreamHandler()
@@ -73,7 +76,7 @@ def main():
logger.error('"{}" is a invalid site. Select from: {}'.format(arguments['-s'], config['DEFAULT']['SITE_OPTIONS'])) logger.error('"{}" is a invalid site. Select from: {}'.format(arguments['-s'], config['DEFAULT']['SITE_OPTIONS']))
sys.exit() sys.exit()
searchTorrentSite(config, arguments['<query>'], site, arguments['--print']) searchTorrentSite(config, arguments['<query>'], site, arguments['-f'], arguments['--print'])
def getConfig(): def getConfig():
@@ -120,18 +123,13 @@ def chooseCandidate(torrent_list):
intersecting_release_types = set(torrent.find_release_type()) & set(match_release_type) intersecting_release_types = set(torrent.find_release_type()) & set(match_release_type)
size, _, size_id = torrent.size.partition(' ') size, _, size_id = torrent.size.partition(' ')
# if intersecting_release_types and int(torrent.seed_count) > 0 and float(size) > 4 and size_id == 'GiB':
if intersecting_release_types: if intersecting_release_types:
interesting_torrents.append(torrent.get_all_attr()) interesting_torrents.append(torrent)
# print('{} : {} : {} {}'.format(torrent.name, torrent.size, torrent.seed_count, torrent.magnet))
# interesting_torrents.append(torrent)
# else:
# print('Denied match! %s : %s : %s' % (torrent.name, torrent.size, torrent.seed_count))
return interesting_torrents return interesting_torrents
def searchTorrentSite(config, query, site, print_result): def searchTorrentSite(config, query, site, filter, print_result):
""" """
Selects site based on input and finds torrents for that site based on query Selects site based on input and finds torrents for that site based on query
@@ -153,10 +151,14 @@ def searchTorrentSite(config, query, site, print_result):
config['JACKETT']['PATH'], config['JACKETT']['LIMIT'], config.getboolean('JACKETT', 'SSL')) config['JACKETT']['PATH'], config['JACKETT']['LIMIT'], config.getboolean('JACKETT', 'SSL'))
torrents_found = jackett.search(query) torrents_found = jackett.search(query)
if (filter):
torrents_found = chooseCandidate(torrents_found)
jsonList = createJSONList(torrents_found) jsonList = createJSONList(torrents_found)
if (print_result): if (print_result):
print(jsonList) print(jsonList)
return jsonList return jsonList
def signal_handler(signal, frame): def signal_handler(signal, frame):