From 2b320cb6f6b6f29c7c8c2edb26fae94a0af7c474 Mon Sep 17 00:00:00 2001
From: Gosku <gosku26@gmail.com>
Date: Tue, 19 May 2015 11:59:52 -0400
Subject: [PATCH] Add size attribute in providers.generic for nzb providers
---
sickbeard/providers/generic.py | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/sickbeard/providers/generic.py b/sickbeard/providers/generic.py
index 078b47fe..47f6ef08 100644
--- a/sickbeard/providers/generic.py
+++ b/sickbeard/providers/generic.py
@@ -256,6 +256,21 @@ class GenericProvider:
return title, url
+ def _get_size(self, item):
+ """Gets the size from the item"""
+ if self.providerType != GenericProvider.NZB:
+ logger.log(u"Torrent Generic providers doesn't have _get_size() implemented yet", logger.DEBUG)
+ return -1
+ else:
+ size = item.get('links')[1].get('length')
+ if size:
+ size = int(size)
+ return size
+ else:
+ logger.log(u"Size was not found in your provider response", logger.DEBUG)
+ return -1
+
+
def findSearchResults(self, show, episodes, search_mode, manualSearch=False, downCurQuality=False):
self._checkAuth()
@@ -430,6 +445,7 @@ class GenericProvider:
result.release_group = release_group
result.version = version
result.content = None
+ result.size = self._get_size(item)
if len(epObj) == 1:
epNum = epObj[0].episode
--
GitLab