From 484dba0de852ce8b055e71e8c20e7d4dec0f432e Mon Sep 17 00:00:00 2001 From: SickBeard Custom User Date: Thu, 24 Apr 2014 16:50:35 +0100 Subject: [PATCH] proper settings for proxy --- gui/slick/interfaces/default/config_general.tmpl | 10 ++++++++++ sickbeard/__init__.py | 5 ++++- sickbeard/helpers.py | 13 ++++++++----- sickbeard/providers/kat.py | 13 ++++++++----- sickbeard/providers/thepiratebay.py | 10 +++++++++- sickbeard/webserve.py | 5 +++-- 6 files changed, 42 insertions(+), 14 deletions(-) diff --git a/gui/slick/interfaces/default/config_general.tmpl b/gui/slick/interfaces/default/config_general.tmpl index 2dcad7c0..919e91e8 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..79ff342c 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', 'http://192.168.1.11:8118') # 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 8e3fd579..0bb6f6a2 100644 --- a/sickbeard/helpers.py +++ b/sickbeard/helpers.py @@ -185,12 +185,15 @@ Returns a byte-string retrieved from the url provider. it = iter(req_headers) - proxies = { - "http": "http://192.168.1.11:8118", - "https": "http://192.168.1.11:8118", - } + 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) + 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 diff --git a/sickbeard/providers/kat.py b/sickbeard/providers/kat.py index 84fb1e8f..bb2cb841 100644 --- a/sickbeard/providers/kat.py +++ b/sickbeard/providers/kat.py @@ -313,12 +313,15 @@ class KATProvider(generic.TorrentProvider): parsed[2] = re.sub("/{2,}", "/", parsed[2]) # replace two or more / with one url = urlparse.urlunparse(parsed) - proxies = { - "http": "http://192.168.1.11:8118", - "https": "http://192.168.1.11:8118", - } + if sickbeard.PROXY_SETTING: + proxies = { + "http": sickbeard.PROXY_SETTING, + "https": sickbeard.PROXY_SETTING, + } - r = requests.get(url, proxies=proxies) + 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 82f50495..9dff25a8 100644 --- a/sickbeard/providers/thepiratebay.py +++ b/sickbeard/providers/thepiratebay.py @@ -313,7 +313,15 @@ class ThePirateBayProvider(generic.TorrentProvider): } try: - r = requests.get(url, headers=headers, proxies=proxies) + 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..c797926b 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()