Add support for Amazon ES service with IAM authentication

redash/query_runner/amazon_elasticsearch.py

@@ -0,0 +1,56 @@
+from .elasticsearch import ElasticSearch
+from . import register
+
+try:
+    from requests_aws4auth import AWS4Auth
+    enabled = True
+except ImportError:
+    enabled = False
+
+
+class AmazonElasticsearchService(ElasticSearch):
+    @classmethod
+    def name(cls):
+        return "Amazon Elasticsearch Service"
+
+    @classmethod
+    def enabled(cls):
+        return enabled
+
+    @classmethod
+    def type(cls):
+        return "aws_es"
+
+    @classmethod
+    def configuration_schema(cls):
+        return {
+            'type': 'object',
+            'properties': {
+                'server': {
+                    'type': 'string',
+                    'title': 'Endpoint'
+                },
+                'region': {
+                    'type': 'string',
+                },
+                'access_key': {
+                    'type': 'string',
+                    'title': 'Access Key'
+                },
+                'secret_key': {
+                    'type': 'string',
+                    'title': 'Secret Key'
+                }
+            },
+            "secret": ["secret_key"],
+            "order": ["server", "region", "access_key", "secret_key"],
+            "required": ['server', 'region', 'access_key', 'secret_key']
+        }
+
+    def __init__(self, configuration):
+        super(AmazonElasticsearchService, self).__init__(configuration)
+
+        self.auth = AWS4Auth(configuration['access_key'], configuration['secret_key'], configuration['region'], 'es')
+
+
+register(AmazonElasticsearchService)

redash/query_runner/elasticsearch.py

@@ -74,7 +74,6 @@ def enabled(cls):
 
     def __init__(self, configuration):
         super(BaseElasticSearch, self).__init__(configuration)
-
         self.syntax = "json"
 
         if self.DEBUG_ENABLED:

redash/settings/__init__.py

@@ -15,6 +15,7 @@ def all_settings():
 
     return settings
 
+
 REDIS_URL = os.environ.get('REDASH_REDIS_URL', os.environ.get('REDIS_URL', "redis://localhost:6379/0"))
 PROXIES_COUNT = int(os.environ.get('REDASH_PROXIES_COUNT', "1"))
 
@@ -165,6 +166,7 @@ def all_settings():
     'redash.query_runner.url',
     'redash.query_runner.influx_db',
     'redash.query_runner.elasticsearch',
+    'redash.query_runner.amazon_elasticsearch',
     'redash.query_runner.presto',
     'redash.query_runner.databricks',
     'redash.query_runner.hive_ds',

requirements_all_ds.txt

@@ -25,6 +25,7 @@ pymapd>=0.2.1
 qds-sdk>=1.9.6
 ibm-db>=2.0.9
 pydruid==0.4
+requests-aws4auth==0.9
 # certifi is needed to support MongoDB and SSL:
 certifi
 # We don't install snowflake connector by default, as it's causing conflicts with