Added inline support

Author: jprakash-db

src/databricks/sql/utils.py

@@ -5,10 +5,10 @@
 import decimal
 from abc import ABC, abstractmethod
 from collections import OrderedDict, namedtuple
-from collections.abc import Iterable
+from collections.abc import Mapping
 from decimal import Decimal
 from enum import Enum
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union, Sequence
 import re
 
 import lz4.frame
@@ -458,8 +458,15 @@ def escape_string(self, item):
         return "'{}'".format(item.replace("\\", "\\\\").replace("'", "\\'"))
 
     def escape_sequence(self, item):
-        l = map(str, map(self.escape_item, item))
-        return "(" + ",".join(l) + ")"
+        l = map(self.escape_item, item)
+        return "ARRAY(" + ",".join(l) + ")"
+
+    def escape_mapping(self, item):
+        l = map(
+            self.escape_item,
+            (element for key, value in item.items() for element in (key, value)),
+        )
+        return "MAP(" + ",".join(l) + ")"
 
     def escape_datetime(self, item, format, cutoff=0):
         dt_str = item.strftime(format)
@@ -476,14 +483,16 @@ def escape_item(self, item):
             return self.escape_number(item)
         elif isinstance(item, str):
             return self.escape_string(item)
-        elif isinstance(item, Iterable):
-            return self.escape_sequence(item)
         elif isinstance(item, datetime.datetime):
             return self.escape_datetime(item, self._DATETIME_FORMAT)
         elif isinstance(item, datetime.date):
             return self.escape_datetime(item, self._DATE_FORMAT)
         elif isinstance(item, decimal.Decimal):
             return self.escape_decimal(item)
+        elif isinstance(item, Sequence):
+            return self.escape_sequence(item)
+        elif isinstance(item, Mapping):
+            return self.escape_mapping(item)
         else:
             raise exc.ProgrammingError("Unsupported object {}".format(item))