From 6f21a2cd18ac8976071823e2c9f1acda60f74474 Mon Sep 17 00:00:00 2001
From: supergonkas <duramato@users.noreply.github.com>
Date: Wed, 23 Dec 2015 00:36:14 +0000
Subject: [PATCH] Add costum url suport for ExtraTorrent

---
 sickbeard/providers/extratorrent.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/sickbeard/providers/extratorrent.py b/sickbeard/providers/extratorrent.py
index b54990033..bcebcf3a7 100644
--- a/sickbeard/providers/extratorrent.py
+++ b/sickbeard/providers/extratorrent.py
@@ -1,4 +1,3 @@
-# coding=utf-8
 # Author: duramato <matigonkas@outlook.com>
 # Author: miigotu
 # URL: https://github.com/SickRage/sickrage
@@ -26,7 +25,7 @@ from sickbeard import logger
 from sickbeard import tvcache
 from sickbeard.common import USER_AGENT
 from sickrage.helper.common import try_int
-from sickrage.providers.torrent.TorrentProvider import TorrentProvider
+from sickrage.providers.TorrentProvider import TorrentProvider
 
 
 class ExtraTorrentProvider(TorrentProvider):
@@ -36,7 +35,7 @@ class ExtraTorrentProvider(TorrentProvider):
         self.urls = {
             'index': 'http://extratorrent.cc',
             'rss': 'http://extratorrent.cc/rss.xml',
-        }
+            }
 
         self.url = self.urls['index']
 
@@ -44,6 +43,7 @@ class ExtraTorrentProvider(TorrentProvider):
         self.ratio = None
         self.minseed = None
         self.minleech = None
+        self.custom_url = None
 
         self.cache = ExtraTorrentCache(self)
         self.headers.update({'User-Agent': USER_AGENT})
@@ -63,7 +63,11 @@ class ExtraTorrentProvider(TorrentProvider):
 
                 try:
                     self.search_params.update({'type': ('search', 'rss')[mode == 'RSS'], 'search': search_string})
-                    data = self.get_url(self.urls['rss'], params=self.search_params)
+                    if self.custom_url:
+                        url = self.custom_url + '/rss.xml'
+                        data = self.get_url(url, params=self.search_params)
+                    else:
+                        data = self.get_url(self.urls['rss'], params=self.search_params)
                     if not data:
                         logger.log(u"No data returned from provider", logger.DEBUG)
                         continue
@@ -88,7 +92,7 @@ class ExtraTorrentProvider(TorrentProvider):
 
                     for item in entries:
                         title = item['title'].decode('utf-8')
-                        # info_hash = item['info_hash']
+                       # info_hash = item['info_hash']
                         size = int(item['size'])
                         seeders = try_int(item['seeders'], 0)
                         leechers = try_int(item['leechers'], 0)
-- 
GitLab