reorganizes the es_session auth credentials

kellischeuble · Jun 19, 2020 · 93b05d6 · 93b05d6
1 parent 94735bc
commit 93b05d6
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 27 deletions.
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/src/app/queries.py b/src/app/queries.py
@@ -13,17 +13,19 @@
 region = os.environ["REGION"]
 service = "es"
 
-class ESConnection(AWSAuthConnection):
-    def __init__(self, region, **kwargs):
-        super(ESConnection, self).__init__(**kwargs)
-        self._set_auth_region_name(region)
-        self._set_auth_service_name("es")
+session = boto3.Session()
+credentials = session.get_credentials()
 
+awsauth = AssumeRoleAWS4Auth(credentials, region, service)
+
+es = Elasticsearch(
+    hosts=[host],
+    http_auth=awsauth,
+    use_ssl=True,
+    verify_certs=True,
+    connection_class=RequestsHttpConnection,
+)
 
-es = ESConnection(
-    region=region, 
-    host=host,
-    is_secure=False)
 
 # es = Elasticsearch()