diff --git a/sickbeard/providers/kat.py b/sickbeard/providers/kat.py index 763091ddcebc6e4d7be4a733aa32ee5d551accd5..bdefc78cdb8b2c583fa123e57d10990880e38146 100644 --- a/sickbeard/providers/kat.py +++ b/sickbeard/providers/kat.py @@ -214,6 +214,13 @@ class KATProvider(generic.TorrentProvider): return [search_string] + def _get_size(self, item): + title, url, id, seeders, leechers, size, pubdate = item + if not size: + return -1 + + return size + def _doSearch(self, search_params, search_mode='eponly', epcount=0, age=0, epObj=None): results = [] @@ -243,7 +250,7 @@ class KATProvider(generic.TorrentProvider): seeders = int(item['torrent_seeds']) leechers = int(item['torrent_peers']) size = int(item['torrent_contentlength']) - except (AttributeError, TypeError): + except (AttributeError, TypeError, KeyError): continue if mode != 'RSS' and (seeders < self.minseed or leechers < self.minleech): diff --git a/sickbeard/providers/omgwtfnzbs.py b/sickbeard/providers/omgwtfnzbs.py index d1cb1453c361194d63995acf93b7706b4509cb5c..6d4265474752ce6f6f4dda48e67f5cf437f26c46 100644 --- a/sickbeard/providers/omgwtfnzbs.py +++ b/sickbeard/providers/omgwtfnzbs.py @@ -101,6 +101,14 @@ class OmgwtfnzbsProvider(generic.NZBProvider): def _get_title_and_url(self, item): return (item['release'], item['getnzb']) + def _get_size(self, item): + try: + size = int(item['sizebytes']) + except (ValueError, TypeError, AttributeError, KeyError): + return -1 + + return size + def _doSearch(self, search, search_mode='eponly', epcount=0, retention=0, epObj=None): self._checkAuth()