diff --git a/gui/slick/interfaces/default/config_general.tmpl b/gui/slick/interfaces/default/config_general.tmpl index 2dcad7c0..afd01d09 100644 --- a/gui/slick/interfaces/default/config_general.tmpl +++ b/gui/slick/interfaces/default/config_general.tmpl @@ -274,6 +274,16 @@ +
+ + +
diff --git a/sickbeard/__init__.py b/sickbeard/__init__.py index b99e3b45..ae6c4d1f 100644 --- a/sickbeard/__init__.py +++ b/sickbeard/__init__.py @@ -487,7 +487,8 @@ def initialize(consoleLogging=True): GUI_NAME, HOME_LAYOUT, HISTORY_LAYOUT, DISPLAY_SHOW_SPECIALS, COMING_EPS_LAYOUT, COMING_EPS_SORT, COMING_EPS_DISPLAY_PAUSED, COMING_EPS_MISSED_RANGE, DATE_PRESET, TIME_PRESET, TIME_PRESET_W_SECONDS, \ METADATA_WDTV, METADATA_TIVO, IGNORE_WORDS, CALENDAR_UNPROTECTED, CREATE_MISSING_SHOW_DIRS, \ ADD_SHOWS_WO_DIR, USE_SUBTITLES, SUBTITLES_LANGUAGES, SUBTITLES_DIR, SUBTITLES_SERVICES_LIST, SUBTITLES_SERVICES_ENABLED, SUBTITLES_HISTORY, SUBTITLES_FINDER_FREQUENCY, subtitlesFinderScheduler, \ - USE_FAILED_DOWNLOADS, DELETE_FAILED, ANON_REDIRECT, LOCALHOST_IP, TMDB_API_KEY + USE_FAILED_DOWNLOADS, DELETE_FAILED, ANON_REDIRECT, LOCALHOST_IP, TMDB_API_KEY, \ + PROXY_SETTING if __INITIALIZED__: return False @@ -541,6 +542,7 @@ def initialize(consoleLogging=True): LOCALHOST_IP = check_setting_str(CFG, 'General', 'localhost_ip', '') ANON_REDIRECT = check_setting_str(CFG, 'General', 'anon_redirect', 'http://dereferer.org/?') + PROXY_SETTING = check_setting_str(CFG, 'General', 'proxy_setting', '') # attempt to help prevent users from breaking links by using a bad url if not ANON_REDIRECT.endswith('?'): ANON_REDIRECT = '' @@ -1300,6 +1302,7 @@ def save_config(): new_config['General']['launch_browser'] = int(LAUNCH_BROWSER) new_config['General']['update_shows_on_start'] = int(UPDATE_SHOWS_ON_START) new_config['General']['sort_article'] = int(SORT_ARTICLE) + new_config['General']['proxy_setting'] = PROXY_SETTING if PROXY_SETTING else '' new_config['General']['use_listview'] = int(USE_LISTVIEW) new_config['General']['metadata_xbmc'] = METADATA_XBMC diff --git a/sickbeard/helpers.py b/sickbeard/helpers.py index 96013ff0..0bb6f6a2 100644 --- a/sickbeard/helpers.py +++ b/sickbeard/helpers.py @@ -184,7 +184,16 @@ Returns a byte-string retrieved from the url provider. url = urlparse.urlunparse(parsed) it = iter(req_headers) - resp = requests.get(url, params=params, data=post_data, headers=dict(zip(it, it))) + + if sickbeard.PROXY_SETTING: + proxies = { + "http": sickbeard.PROXY_SETTING, + "https": sickbeard.PROXY_SETTING, + } + + resp = requests.get(url, params=params, data=post_data, headers=dict(zip(it, it)), proxies=proxies) + else: + resp = requests.get(url, params=params, data=post_data, headers=dict(zip(it, it))) except requests.HTTPError, e: logger.log(u"HTTP error " + str(e.errno) + " while loading URL " + url, logger.WARNING) return None @@ -981,4 +990,4 @@ def real_path(path): """ Returns: the canonicalized absolute pathname. The resulting path will have no symbolic link, '/./' or '/../' components. """ - return ek.ek(os.path.normpath, ek.ek(os.path.normcase, ek.ek(os.path.realpath, path))) \ No newline at end of file + return ek.ek(os.path.normpath, ek.ek(os.path.normcase, ek.ek(os.path.realpath, path))) diff --git a/sickbeard/providers/kat.py b/sickbeard/providers/kat.py index 3e700549..bb2cb841 100644 --- a/sickbeard/providers/kat.py +++ b/sickbeard/providers/kat.py @@ -312,8 +312,16 @@ class KATProvider(generic.TorrentProvider): parsed = list(urlparse.urlparse(url)) parsed[2] = re.sub("/{2,}", "/", parsed[2]) # replace two or more / with one url = urlparse.urlunparse(parsed) + + if sickbeard.PROXY_SETTING: + proxies = { + "http": sickbeard.PROXY_SETTING, + "https": sickbeard.PROXY_SETTING, + } - r = requests.get(url) + r = requests.get(url, proxies=proxies) + else: + r = requests.get(url) except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError), e: logger.log(u"Error loading " + self.name + " URL: " + str(sys.exc_info()) + " - " + ex(e), logger.ERROR) return None diff --git a/sickbeard/providers/thepiratebay.py b/sickbeard/providers/thepiratebay.py index 5c540e40..13f98568 100644 --- a/sickbeard/providers/thepiratebay.py +++ b/sickbeard/providers/thepiratebay.py @@ -306,9 +306,17 @@ class ThePirateBayProvider(generic.TorrentProvider): # We have to fake a search on the proxy site to get data if self.proxy.isEnabled(): headers.update({'referer': self.proxy.getProxyURL()}) - + try: - r = requests.get(url, headers=headers) + if sickbeard.PROXY_SETTING: + proxies = { + "http": sickbeard.PROXY_SETTING, + "https": sickbeard.PROXY_SETTING, + } + + r = requests.get(url, headers=headers, proxies=proxies) + else: + r = requests.get(url, headers=headers) except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError), e: logger.log(u"Error loading " + self.name + " URL: " + str(sys.exc_info()) + " - " + ex(e), logger.ERROR) return None diff --git a/sickbeard/webserve.py b/sickbeard/webserve.py index 61de1d87..e1286e42 100644 --- a/sickbeard/webserve.py +++ b/sickbeard/webserve.py @@ -976,7 +976,7 @@ class ConfigGeneral: def saveGeneral(self, log_dir=None, web_port=None, web_log=None, encryption_version=None, web_ipv6=None, update_shows_on_start=None, launch_browser=None, web_username=None, use_api=None, api_key=None, web_password=None, version_notify=None, enable_https=None, https_cert=None, https_key=None, - sort_article=None, auto_update=None, + sort_article=None, auto_update=None, proxy_settings=None, anon_redirect=None, git_path=None, calendar_unprotected=None, date_preset=None, time_preset=None): results = [] @@ -991,6 +991,7 @@ class ConfigGeneral: sickbeard.LAUNCH_BROWSER = config.checkbox_to_value(launch_browser) sickbeard.SORT_ARTICLE = config.checkbox_to_value(sort_article) sickbeard.ANON_REDIRECT = anon_redirect + sickbeard.PROXY_SETTINGS = proxy_settings sickbeard.GIT_PATH = git_path sickbeard.CALENDAR_UNPROTECTED = config.checkbox_to_value(calendar_unprotected) # sickbeard.LOG_DIR is set in config.change_LOG_DIR() @@ -3727,4 +3728,4 @@ class WebInterface: errorlogs = ErrorLogs() - ui = UI() \ No newline at end of file + ui = UI() diff --git a/tests/tests.py b/tests/tests.py index 0c96669e..ff516715 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -31,16 +31,23 @@ from sickbeard import classes class APICheck(unittest.TestCase): lang = "en" - search_term = 'american' + search_term = 'american dad' results = {} final_results = [] + lINDEXER_API_PARMS = indexerApi(1).api_params.copy() + lINDEXER_API_PARMS['language'] = lang + lINDEXER_API_PARMS['custom_ui'] = classes.AllShowsListUI + t = indexerApi(1).indexer(**lINDEXER_API_PARMS) + test = t[search_term] + for indexer in indexerApi().indexers: lINDEXER_API_PARMS = indexerApi(indexer).api_params.copy() lINDEXER_API_PARMS['language'] = lang lINDEXER_API_PARMS['custom_ui'] = classes.AllShowsListUI t = indexerApi(indexer).indexer(**lINDEXER_API_PARMS) + t[search_term] print("Searching for Show with searchterm: %s on Indexer: %s" % (search_term, indexerApi(indexer).name)) try: