Athena: bring back the option to disable query annotations.

karthikrk1 · Jun 28, 2017 · e1ac5bb · e1ac5bb
1 parent c2e84c9
commit e1ac5bb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/redash/query_runner/athena.py b/redash/query_runner/athena.py
@@ -2,10 +2,11 @@
 import logging
 
 from redash.query_runner import *
+from redash.settings import parse_boolean
 from redash.utils import JSONEncoder
 
 logger = logging.getLogger(__name__)
-
+ANNOTATE_QUERY = parse_boolean(os.environ.get('ATHENA_ANNOTATE_QUERY', 'true'))
 
 try:
     import pyathena
@@ -87,6 +88,10 @@ def configuration_schema(cls):
     def enabled(cls):
         return enabled
 
+    @classmethod
+    def annotate_query(cls):
+        return ANNOTATE_QUERY
+
     @classmethod
     def type(cls):
         return "athena"