Fix

Author: foteinigk

transifex/native/__init__.py

@@ -8,6 +8,7 @@ def init(
     fetch_all_langs=False,
     filter_tags=None,
     filter_status=None,
+    fetch_timeout=0,
 ):
     """Initialize the framework.
 
@@ -25,6 +26,8 @@ def init(
     :param bool fetch_all_langs: force pull all remote languages
     :param str filter_tags: fetch only content with tags
     :param str filter_status: fetch only content with specific translation status
+    :param int fetch_timeout: maximum time in seconds to wait when fetching
+    translations or locales from CDS
     """
     if not tx.initialized:
         tx.init(
@@ -38,6 +41,7 @@ def init(
             fetch_all_langs=fetch_all_langs,
             filter_tags=filter_tags,
             filter_status=filter_status,
+            fetch_timeout=fetch_timeout,
         )
 
 

transifex/native/cds.py

@@ -2,7 +2,6 @@
 import sys
 import time
 from urllib.parse import urlencode
-from transifex.native.django import settings as native_settings
 
 import requests
 from transifex.native.consts import (KEY_CHARACTER_LIMIT,
@@ -58,7 +57,7 @@ class CDSHandler(object):
 
     def __init__(self, configured_languages, token, secret=None,
                  host=TRANSIFEX_CDS_HOST, fetch_all_langs=False,
-                 filter_tags=None, filter_status=None):
+                 filter_tags=None, filter_status=None, fetch_timeout=0):
         """Constructor.
 
         :param list configured_languages: a list of language codes for the
@@ -74,6 +73,7 @@ def __init__(self, configured_languages, token, secret=None,
         self.secret = secret
         self.host = host or TRANSIFEX_CDS_HOST
         self.etags = EtagStore()
+        self.fetch_timeout = fetch_timeout
 
     def fetch_languages(self):
         """Fetch the languages defined in the CDS for the specific project.
@@ -368,7 +368,7 @@ def retry_get_request(self, *args, **kwargs):
         retries_5xx = 0
         last_response_status = 202
         start_ts = time.time()
-        max_total_seconds = native_settings.TRANSIFEX_FETCH_TEMEOUT
+        max_total_seconds = self.fetch_timeout
 
         while True:
             response = requests.get(*args, **kwargs)

transifex/native/core.py

@@ -42,6 +42,7 @@ def init(
         missing_policy=None, error_policy=None, cache=None,
         fetch_all_langs=False, filter_tags=None,
         filter_status=None,
+        fetch_timeout=0,
     ):
         """Create an instance of the core framework class.
 
@@ -62,6 +63,8 @@ def init(
         :param bool fetch_all_langs: force pull all remote languages
         :param str filter_tags: fetch only content with tags
         :param str filter_status: fetch only content with specific translation status
+        :param int fetch_timeout: maximum time in seconds to wait when fetching
+            translations or locales from CDS
         """
         self._languages = languages
         self._cache = cache or MemoryCache()
@@ -72,6 +75,7 @@ def init(
             fetch_all_langs=fetch_all_langs,
             filter_tags=filter_tags,
             filter_status=filter_status,
+            fetch_timeout=fetch_timeout,
         )
         self.initialized = True
 

transifex/native/django/apps.py

@@ -94,6 +94,7 @@ def ready(self):
             fetch_all_langs=native_settings.TRANSIFEX_FETCH_ALL_LANGUAGES,
             filter_tags=native_settings.TRANSIFEX_FILTER_TAGS,
             filter_status=native_settings.TRANSIFEX_FILTER_STATUS,
+            fetch_timeout=native_settings.TRANSIFEX_FETCH_TIMEOUT,
         )
 
         if fetch_translations: