diff --git a/couchpotato/core/media/_base/providers/nzb/newznab.py b/couchpotato/core/media/_base/providers/nzb/newznab.py index 84b0e45..0f28db8 100644 --- a/couchpotato/core/media/_base/providers/nzb/newznab.py +++ b/couchpotato/core/media/_base/providers/nzb/newznab.py @@ -128,6 +128,7 @@ class Base(NZBProvider, RSS): api_keys = splitString(self.conf('api_key'), clean = False) extra_score = splitString(self.conf('extra_score'), clean = False) custom_tags = splitString(self.conf('custom_tag'), clean = False) + custom_categories = splitString(self.conf('custom_categories'), clean = False) list = [] for nr in range(len(hosts)): @@ -144,12 +145,16 @@ class Base(NZBProvider, RSS): try: custom_tag = custom_tags[nr] except: custom_tag = '' + try: custom_category = custom_categories[nr].replace(" ", ",") + except: custom_category = '' + list.append({ 'use': uses[nr], 'host': host, 'api_key': key, 'extra_score': score, - 'custom_tag': custom_tag + 'custom_tag': custom_tag, + 'custom_category' : custom_category }) return list @@ -266,6 +271,13 @@ config = [{ 'description': 'Add custom tags, for example add rls=1 to get only scene releases from nzbs.org', }, { + 'name': 'custom_categories', + 'advanced': True, + 'label': 'Custom Categories', + 'default': '2000,2000,2000,2000,2000,2000', + 'description': 'Specify categories to search in seperated by a single space, defaults to all movies. EG: "2030 2040 2060" would only search in HD, SD, and 3D movie categories', + }, + { 'name': 'api_key', 'default': ',,,,,', 'label': 'Api Key', diff --git a/couchpotato/core/media/movie/providers/nzb/newznab.py b/couchpotato/core/media/movie/providers/nzb/newznab.py index fc94acb..3392b91 100644 --- a/couchpotato/core/media/movie/providers/nzb/newznab.py +++ b/couchpotato/core/media/movie/providers/nzb/newznab.py @@ -23,4 +23,7 @@ class Newznab(MovieProvider, Base): if len(host.get('custom_tag', '')) > 0: query = '%s&%s' % (query, host.get('custom_tag')) + if len(host['custom_category']) > 0: + query = '%s&cat=%s' % (query, host['custom_category']) + return query