Add option to lowercase column names

redash/query_runner/snowflake.py

@@ -1,5 +1,6 @@
 try:
     import snowflake.connector
+
     enabled = True
 except ImportError:
     enabled = False
@@ -39,14 +40,19 @@ def configuration_schema(cls):
             "type": "object",
             "properties": {
                 "account": {"type": "string"},
-                "region": {"type": "string", "default": "us-west"},
                 "user": {"type": "string"},
                 "password": {"type": "string"},
                 "warehouse": {"type": "string"},
                 "database": {"type": "string"},
+                "region": {"type": "string", "default": "us-west"},
+                "lower_case_columns": {
+                    "type": "boolean",
+                    "title": "Lower Case Column Names in Results",
+                    "default": False,
+                },
                 "host": {"type": "string"},
             },
-            "order": ["account", "region", "user", "password", "warehouse", "database", "host"],
+            "order": ["account", "user", "password", "warehouse", "database", "region", "host"],
             "required": ["user", "password", "account", "database", "warehouse"],
             "secret": ["password"],
             "extra_options": [
@@ -82,19 +88,29 @@ def _get_connection(self):
                 host = "{}.snowflakecomputing.com".format(account)
 
 
+
         connection = snowflake.connector.connect(
-            user = self.configuration["user"],
-            password = self.configuration["password"],
-            account = account,
-            region = region,
-            host = host
+            user=self.configuration["user"],
+            password=self.configuration["password"],
+            account=account,
+            region=region,
+            host = host,
         )
 
         return connection
 
+    def _column_name(self, column_name):
+        if self.configuration.get("lower_case_columns", False):
+            return column_name.lower()
+
+        return column_name
+
     def _parse_results(self, cursor):
         columns = self.fetch_columns(
-            [(i[0], self.determine_type(i[1], i[5])) for i in cursor.description]
+            [
+                (self._column_name(i[0]), self.determine_type(i[1], i[5]))
+                for i in cursor.description
+            ]
         )
         rows = [
             dict(zip((column["name"] for column in columns), row)) for row in cursor
@@ -137,10 +153,10 @@ def _run_query_without_warehouse(self, query):
             cursor.close()
             connection.close()
 
-        return data, error    
-    
+        return data, error
+
     def _database_name_includes_schema(self):
-        return '.' in self.configuration.get('database')
+        return "." in self.configuration.get("database")
 
     def get_schema(self, get_stats=False):
         if self._database_name_includes_schema():
@@ -166,4 +182,4 @@ def get_schema(self, get_stats=False):
         return list(schema.values())
 
 
-register(Snowflake)
+register(Snowflake)