Allow 3d categories

couchpotato/core/media/_base/providers/base.py

@@ -244,10 +244,16 @@ class YarrProvider(Provider):
 
         return 0
 
-    def getCatId(self, identifier):
+    def getCatId(self, quality = None):
+        if not quality: quality = {}
+        identifier = quality.get('identifier')
+
+        want_3d = False
+        if quality.get('custom'):
+            want_3d = quality['custom'].get('3d')
 
         for ids, qualities in self.cat_ids:
-            if identifier in qualities:
+            if identifier in qualities or (want_3d and '3d' in qualities):
                 return ids
 
         if self.cat_backup_id:

couchpotato/core/media/_base/providers/nzb/omgwtfnzbs.py

@@ -42,7 +42,7 @@ class Base(NZBProvider, RSS):
         q = '%s %s' % (title, movie['info']['year'])
         params = tryUrlencode({
             'search': q,
-            'catid': ','.join([str(x) for x in self.getCatId(quality['identifier'])]),
+            'catid': ','.join([str(x) for x in self.getCatId(quality)]),
             'user': self.conf('username', default = ''),
             'api': self.conf('api_key', default = ''),
         })

couchpotato/core/media/_base/providers/torrent/ilovetorrents.py

@@ -36,7 +36,7 @@ class Base(TorrentProvider):
 
         page = 0
         total_pages = 1
-        cats = self.getCatId(quality['identifier'])
+        cats = self.getCatId(quality)
 
         while page < total_pages:
 

couchpotato/core/media/_base/providers/torrent/iptorrents.py

@@ -25,14 +25,14 @@ class Base(TorrentProvider):
     cat_backup_id = None
 
     def buildUrl(self, title, media, quality):
-        return self._buildUrl(title.replace(':', ''), quality['identifier'])
+        return self._buildUrl(title.replace(':', ''), quality)
 
-    def _buildUrl(self, query, quality_identifier):
+    def _buildUrl(self, query, quality):
 
-        cat_ids = self.getCatId(quality_identifier)
+        cat_ids = self.getCatId(quality)
 
         if not cat_ids:
-            log.warning('Unable to find category ids for identifier "%s"', quality_identifier)
+            log.warning('Unable to find category ids for identifier "%s"', quality.get('identifier'))
             return None
 
         return self.urls['search'] % ("&".join(("l%d=" % x) for x in cat_ids), tryUrlencode(query).replace('%', '%%'))

couchpotato/core/media/_base/providers/torrent/kickasstorrents.py

@@ -42,7 +42,7 @@ class Base(TorrentMagnetProvider):
 
         if data:
 
-            cat_ids = self.getCatId(quality['identifier'])
+            cat_ids = self.getCatId(quality)
             table_order = ['name', 'size', None, 'age', 'seeds', 'leechers']
 
             try:

couchpotato/core/media/_base/providers/torrent/thepiratebay.py

@@ -39,7 +39,7 @@ class Base(TorrentMagnetProvider):
 
         page = 0
         total_pages = 1
-        cats = self.getCatId(quality['identifier'])
+        cats = self.getCatId(quality)
 
         base_search_url = self.urls['search'] % self.getDomain()
 

couchpotato/core/media/_base/providers/torrent/torrentbytes.py

@@ -36,7 +36,7 @@ class Base(TorrentProvider):
 
     def _searchOnTitle(self, title, movie, quality, results):
 
-        url = self.urls['search'] % (tryUrlencode('%s %s' % (title.replace(':', ''), movie['info']['year'])), self.getCatId(quality['identifier'])[0])
+        url = self.urls['search'] % (tryUrlencode('%s %s' % (title.replace(':', ''), movie['info']['year'])), self.getCatId(quality)[0])
         data = self.getHTMLData(url)
 
         if data: