Private GIT

Skip to content
Snippets Groups Projects
Commit 0042f60b authored by labrys's avatar labrys
Browse files

Fix file size conversion for torrentleech

parent ddd89d4d
No related branches found
No related tags found
No related merge requests found
......@@ -23,10 +23,9 @@ import urllib
from sickbeard import logger
from sickbeard import tvcache
from sickbeard.bs4_parser import BS4Parser
from sickrage.helper.common import try_int, convert_size
from sickrage.providers.torrent.TorrentProvider import TorrentProvider
from sickrage.helper.common import try_int
class TorrentLeechProvider(TorrentProvider):
......@@ -116,7 +115,9 @@ class TorrentLeechProvider(TorrentProvider):
seeders = try_int(result.find('td', class_='seeders').get_text(strip=True))
leechers = try_int(result.find('td', class_='leechers').get_text(strip=True))
size = self._convertSize(result.find_all('td')[labels.index('Size')].get_text(strip=True))
torrent_size = result.find_all('td')[labels.index('Size')].get_text()
size = convert_size(torrent_size) or -1
except (AttributeError, TypeError, KeyError, ValueError):
continue
......@@ -145,26 +146,6 @@ class TorrentLeechProvider(TorrentProvider):
def seed_ratio(self):
return self.ratio
@staticmethod
def _convertSize(size):
modifier = size[-2:].upper()
size = size[:-2].strip()
try:
size = float(size)
if modifier in 'KB':
size = size * 1024 ** 1
elif modifier in 'MB':
size = size * 1024 ** 2
elif modifier in 'GB':
size = size * 1024 ** 3
elif modifier in 'TB':
size = size * 1024**4
else:
raise
except Exception:
size = -1
return long(size)
class TorrentLeechCache(tvcache.TVCache):
def __init__(self, provider_obj):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment