add decimals

Author: const-in

setup.py

@@ -20,7 +20,7 @@
 
 setup(
     name='clickhouse-sqlalchemy',
-    version='0.0.3.7',
+    version='0.0.3.8',
 
     description='Simple ClickHouse SQLAlchemy Dialect',
     long_description=long_description,

src/drivers/base.py

@@ -330,6 +330,9 @@ def visit_enum16(self, type_, **kw):
     def visit_DATETIME(self, type_, **kw):
         return 'DateTime'
 
+    def visit_numeric(self, type_, **kw):
+        return 'Decimal(%s, %s)' % (type_.precision, type_.scale)
+
 
 class ClickHouseExecutionContextBase(default.DefaultExecutionContext):
     @sa_util.memoized_property

src/drivers/escaper.py

@@ -75,4 +75,4 @@ def escape_item(self, item):
         if isinstance(item, (list, tuple)):
             return [self.escape_item(x) for x in item]
         else:
-            raise Exception("Unsupported object {}".format(item))
+            raise Exception("Unsupported object {} ({})".format(item, type(item)))

src/drivers/http/transport.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from functools import partial
+from decimal import Decimal
 
 import pytz
 
@@ -71,6 +72,8 @@ def execute(self, query, params=None):
                 convs.append(self.converters[type_])
             elif type_.startswith("DateTime("):  # datetime with timezone
                 convs.append(partial(self.converters['DateTime'], tz=type_[10:-2]))
+            elif type_.startswith("Decimal"):
+                convs.append(Decimal)
             else:
                 convs.append(None)
 

src/types.py

@@ -83,3 +83,7 @@ def __init__(self, *enums, **kw):
 
 class Enum16(Enum8):
     __visit_name__ = 'enum16'
+
+
+class Decimal(types.Numeric):
+    __visit_name__ = 'numeric'