variable TimeOut

cuenca-mx · Feb 13, 2020 · 864dca1 · 864dca1
1 parent 3c13e4a
commit 864dca1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/hub/__init__.py b/hub/__init__.py
@@ -14,4 +14,4 @@
     init_workers,
 )
 
-logging.basicConfig(level=logging.INFO, format='Hub: %(message)s')
+logging.basicConfig(level=logging.INFO, format='%(message)s')
diff --git a/hub/kinesis/producer.py b/hub/kinesis/producer.py
@@ -5,9 +5,13 @@
 import uuid
 from typing import Dict
 
+from coverage.annotate import os
+
 from hub.client import kinesis_client as client
 from hub.kinesis.helpers import create_stream, dict_to_json, stream_is_active
 
+KINESIS_TIME_OUT = int(os.getenv('KINESIS_TIME_OUT', '5'))
+
 
 class Producer:
     def __init__(self, stream_name: str):
@@ -70,7 +74,9 @@ def wait_for_data(self, uid: str, datetime_insert: dt.datetime) -> Dict:
             except client.exceptions.ProvisionedThroughputExceededException:
                 time.sleep(1)
 
-            if (dt.datetime.utcnow() - start_time).total_seconds() >= 15:
+            if (
+                dt.datetime.utcnow() - start_time
+            ).total_seconds() >= KINESIS_TIME_OUT:
                 raise TimeoutError
 
     @staticmethod

diff --git a/hub/version.py b/hub/version.py
@@ -1 +1 @@
-__version__ = '1.0.20'
+__version__ = '1.0.21'