Update driver requirements

Refactor ES initialization
allegroai · Feb 13, 2022 · 8908c7d · 8908c7d
1 parent b9996e2
commit 8908c7d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 18 deletions.
diff --git a/apiserver/database/__init__.py b/apiserver/database/__init__.py
@@ -60,7 +60,7 @@ def initialize(cls):
                 log.info(f"Using override mongodb host {override_hostname}")
             if override_port:
                 log.info(f"Using override mongodb port {override_port}")
-                
+
         for key, alias in get_items(Database).items():
             if key not in db_entries:
                 missing.append(key)

diff --git a/apiserver/elastic/initialize.py b/apiserver/elastic/initialize.py
@@ -82,15 +82,10 @@ def check_elastic_empty() -> bool:
         es_logger.addFilter(log_filter)
         for retry in range(max_retries):
             try:
-                hosts = cluster_conf.get("hosts", None)
-                http_auth = (
-                    es_factory.get_credentials("events")
-                    if cluster_conf.get("secure", True)
-                    else None
-                )
-                args = cluster_conf.get("args", {})
                 es = Elasticsearch(
-                    hosts=hosts, http_auth=http_auth, **args
+                    hosts=cluster_conf.get("hosts", None),
+                    http_auth=es_factory.get_credentials("events", cluster_conf),
+                    **cluster_conf.get("args", {})
                 )
                 return not es.indices.get_template(name="events*")
             except exceptions.NotFoundError as ex:
@@ -118,10 +113,7 @@ def init_es_data():
 
         log.info(f"Applying mappings to ES host: {hosts_config}")
         args = cluster_conf.get("args", {})
-        http_auth = (
-            es_factory.get_credentials(name)
-            if cluster_conf.get("secure", True)
-            else None
-        )
+        http_auth = es_factory.get_credentials(name)
+
         res = apply_mappings_to_cluster(hosts_config, name, es_args=args, http_auth=http_auth)
         log.info(res)
diff --git a/apiserver/requirements.txt b/apiserver/requirements.txt
@@ -3,7 +3,7 @@ bcrypt>=3.1.4
 boltons>=19.1.0
 boto3==1.14.13
 dpath>=1.4.2,<2.0
-elasticsearch>=7.0.0,<8.0.0
+elasticsearch==7.13.3
 fastjsonschema>=2.8
 flask-compress>=1.4.0
 flask-cors>=3.0.5
@@ -16,15 +16,16 @@ jinja2==2.11.3
 jsonmodels>=2.3
 jsonschema>=2.6.0
 luqum>=0.10.0
-mongoengine==0.19.1
+mongoengine==0.23.1
 nested_dict>=1.61
 packaging==20.3
 psutil>=5.6.5
 pyhocon>=0.3.35
 pyjwt<2.0.0
-pymongo[srv]==3.12
+pymongo[srv]==3.12.0
 python-rapidjson>=0.6.3
-redis>=2.10.5
+redis==3.5.3
+redis-py-cluster>=2.1.3
 related>=0.7.2
 requests>=2.13.0
 semantic_version>=2.8.3,<3