From 9bfa913912466f8b27f93f26118a57d7f92bc37b Mon Sep 17 00:00:00 2001 From: Kevin Midboe Date: Sat, 3 Jul 2021 18:41:17 +0200 Subject: [PATCH] Setup default parameters for searchTorrentSite --- torrentSearch/http_utils.py | 2 +- torrentSearch/search.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/torrentSearch/http_utils.py b/torrentSearch/http_utils.py index b9789c9..c6caccd 100644 --- a/torrentSearch/http_utils.py +++ b/torrentSearch/http_utils.py @@ -55,4 +55,4 @@ def fetch_url(url): elif hasattr(e, 'code'): logger.error('The server couldn\'t fulfill the request.') logger.error('Error code: ', e.code) - sys.exit() \ No newline at end of file + sys.exit() diff --git a/torrentSearch/search.py b/torrentSearch/search.py index 60d40ef..8809183 100755 --- a/torrentSearch/search.py +++ b/torrentSearch/search.py @@ -45,12 +45,12 @@ def chooseCandidate(torrent_list): size, _, size_id = torrent.size.partition(' ') if intersecting_release_types: - interesting_torrents.append(torrent) + interesting_torrents.append(torrent) return interesting_torrents -def searchTorrentSite(config, query, site, filter, print_result): +def searchTorrentSite(query, site='jackett', filter=None, print_result=False, config=None): """ Selects site based on input and finds torrents for that site based on query @@ -61,6 +61,8 @@ def searchTorrentSite(config, query, site, filter, print_result): :return: json list with results :rtype: str """ + if config is None: + config = getConfig() logger.debug('Searching for query {} at {}'.format(query, site)) if site == 'piratebay':