added retries for rarbg search provider

sickrage/core/config/helpers.py

@@ -112,6 +112,8 @@ def change_gui_lang(language):
  # System default language
  gettext.install('messages', sickrage.LOCALE_DIR, codeset='UTF-8', names=["ngettext"])
 
+ sickrage.app.config.gui.gui_lang = language
+
 
 def change_unrar_tool(unrar_tool):
  # Check for failed unrar attempt, and remove it

sickrage/search_providers/torrent/rarbg.py

@@ -127,23 +127,27 @@ class RarbgProvider(TorrentProvider):
 
  search_params['search_string'] = search_string
 
+ max_retries = retries = 5
+ while retries > 0:
+ backoff = 2 ** (max_retries - retries)
+
  # Check if token is still valid before search
  if not self.login():
- continue
+  break
 
- # sleep 5 secs per request
- sleep(5)
+  # sleep per request based on backoff variable
+  sleep(backoff)
 
  resp = self.session.get(self.urls['api'], params=search_params, random_ua=True)
  if not resp or not resp.content:
  sickrage.app.log.debug("No data returned from provider")
- continue
+  break
 
  try:
  data = resp.json()
  except ValueError:
  sickrage.app.log.debug("No data returned from provider")
- continue
+  break
 
  error = data.get('error')
  error_code = data.get('error_code')
@@ -152,18 +156,28 @@ class RarbgProvider(TorrentProvider):
  if error_code == 5:
  # 5 = Too many requests per second
  log_level = sickrage.app.log.INFO
- elif error_code not in (4, 8, 10, 12, 14, 20):
- # 4 = Invalid token. Use get_token for a new one!
+  retries -= 1
+ elif error_code not in (8, 10, 12, 14, 20):
  # 8, 10, 12, 14 = Cant find * in database. Are you sure this * exists?
  # 20 = No results found
  log_level = sickrage.app.log.WARNING
+ retries = 0
+ elif error_code not in (2, 4):
+ # 2, 4 = Invalid token. Use get_token for a new one!
+ log_level = sickrage.app.log.WARNING
+  retries -= 1
  else:
  log_level = sickrage.app.log.DEBUG
+ retries = 0
 
- sickrage.app.log.log(log_level, '{msg} Code: {code}'.format(msg=error, code=error_code))
+  if retries:
  continue
 
+ sickrage.app.log.log(log_level, '{msg} Code: {code}'.format(msg=error, code=error_code))
+ break
+
  results += self.parse(data, mode)
+ break
 
  return results