brynq-sdk-elastic 2.2.1__tar.gz → 2.2.2__tar.gz

{brynq_sdk_elastic-2.2.1 → brynq_sdk_elastic-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_elastic
-Version: 2.2.1
+Version: 2.2.2
 Summary: elastic wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_elastic-2.2.1 → brynq_sdk_elastic-2.2.2}/brynq_sdk/elastic/elastic.py

@@ -17,6 +17,7 @@ class Elastic:
         try:
             self.verify = False
             self.disabled = disabled
+            self.timeout = 60 # seconds before request to elastic fails
             elasticsearch_host = os.getenv("ELASTIC_HOST")
             elasticsearch_port = os.getenv("ELASTIC_PORT")
             kibana_port = os.getenv("KIBANA_PORT")
@@ -75,7 +76,7 @@ class Elastic:
             return 'Healthy connection established with elasticsearch!'
 
         try:
-            health = requests.get(url=f'{self.elasticsearch_host}/_cat/health?', headers=self.elastic_headers, verify=self.verify).status_code
+            health = requests.get(url=f'{self.elasticsearch_host}/_cat/health?', headers=self.elastic_headers, verify=self.verify, timeout=self.timeout).status_code
             if health != 200:
                 raise ConnectionError(f"Elasticsearch cluster health check failed with status code: {health}")
             else:
@@ -185,7 +186,7 @@ class Elastic:
 
         # initial request
         params = {"size": size, "scroll": "10m"}
-        response = requests.get(url=url, headers=self.elastic_headers, params=params, verify=self.verify).json()
+        response = requests.get(url=url, headers=self.elastic_headers, params=params, verify=self.verify, timeout=self.timeout).json()
 
         # next requests until finished
         scroll_id = response['_scroll_id']
@@ -356,7 +357,7 @@ class Elastic:
 
         url = f'{self.elasticsearch_host}/{index_name}/_doc/'
         body = json.dumps(document)
-        response = requests.post(url=url, data=body, headers=self.elastic_headers, verify=self.verify)
+        response = requests.post(url=url, data=body, headers=self.elastic_headers, verify=self.verify, timeout=self.timeout)
         return response
 
     def get_document(self, index_name: str, document_id: str) -> requests.Response:
@@ -370,7 +371,7 @@ class Elastic:
             return None
 
         url = f'{self.elasticsearch_host}/{index_name}/_doc/{document_id}'
-        response = requests.get(url=url, headers=self.elastic_headers, verify=self.verify)
+        response = requests.get(url=url, headers=self.elastic_headers, verify=self.verify, timeout=self.timeout)
         return response
 
     def delete_document(self, index_name: str, document_id: str) -> requests.Response:
@@ -384,7 +385,7 @@ class Elastic:
             return None
 
         url = f'{self.elasticsearch_host}/{index_name}/_doc/{document_id}'
-        response = requests.delete(url=url, headers=self.elastic_headers, verify=self.verify)
+        response = requests.delete(url=url, headers=self.elastic_headers, verify=self.verify, timeout=self.timeout)
         return response
 
     def task_execution_log(self, information: dict) -> requests.Response:
@@ -399,7 +400,7 @@ class Elastic:
         # Add new document
         url = f'{self.elasticsearch_host}/task_execution_log_{self.client_user}/_doc/'
         body = json.dumps(information)
-        response = requests.post(url=url, data=body, headers=self.elastic_headers, verify=self.verify)
+        response = requests.post(url=url, data=body, headers=self.elastic_headers, verify=self.verify, timeout=self.timeout)
         return response
 
     @staticmethod

{brynq_sdk_elastic-2.2.1 → brynq_sdk_elastic-2.2.2}/brynq_sdk_elastic.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-elastic
-Version: 2.2.1
+Version: 2.2.2
 Summary: elastic wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_elastic-2.2.1 → brynq_sdk_elastic-2.2.2}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup
 
 setup(
     name='brynq_sdk_elastic',
-    version='2.2.1',
+    version='2.2.2',
     description='elastic wrapper from BrynQ',
     long_description='elastic wrapper from BrynQ',
     author='BrynQ',