Add Date32 Type (#307)

* Add date32 type

* Add date32 type

* Fix

* Flake linting fix

* Set required server version to 21_9

* Lint

* Fix test

clickhouse_sqlalchemy/drivers/base.py

@@ -35,6 +35,7 @@
     'UInt16': types.UInt16,
     'UInt8': types.UInt8,
     'Date': types.Date,
+    'Date32': types.Date32,
     'DateTime': types.DateTime,
     'DateTime64': types.DateTime64,
     'Float64': types.Float64,

clickhouse_sqlalchemy/drivers/compilers/typecompiler.py

@@ -60,6 +60,9 @@ def visit_uint256(self, type_, **kw):
     def visit_date(self, type_, **kw):
         return 'Date'
 
+    def visit_date32(self, type_, **kw):
+        return 'Date32'
+
     def visit_datetime(self, type_, **kw):
         if type_.timezone:
             return "DateTime('%s')" % type_.timezone

clickhouse_sqlalchemy/types/__init__.py

@@ -22,6 +22,7 @@
     'Float32',
     'Float64',
     'Date',
+    'Date32',
     'DateTime',
     'DateTime64',
     'Enum',
@@ -60,6 +61,7 @@
 from .common import Float32
 from .common import Float64
 from .common import Date
+from .common import Date32
 from .common import DateTime
 from .common import DateTime64
 from .common import Enum

clickhouse_sqlalchemy/types/common.py

@@ -146,6 +146,10 @@ class Date(types.Date, ClickHouseTypeEngine):
     __visit_name__ = 'date'
 
 
+class Date32(types.Date, ClickHouseTypeEngine):
+    __visit_name__ = 'date32'
+
+
 class DateTime(types.DateTime, ClickHouseTypeEngine):
     __visit_name__ = 'datetime'
 

tests/types/test_date32.py

@@ -0,0 +1,46 @@
+import datetime
+
+from sqlalchemy import Column
+from sqlalchemy.sql.ddl import CreateTable
+
+from clickhouse_sqlalchemy import types, engines, Table
+from tests.testcase import BaseTestCase, CompilationTestCase
+from tests.util import with_native_and_http_sessions
+
+
+class Date32CompilationTestCase(CompilationTestCase):
+    required_server_version = (21, 9, 0)
+
+    def test_create_table(self):
+        table = Table(
+            'test', CompilationTestCase.metadata(),
+            Column('x', types.Date32, primary_key=True),
+            engines.Memory()
+        )
+
+        self.assertEqual(
+            self.compile(CreateTable(table)),
+            'CREATE TABLE test (x Date32) ENGINE = Memory'
+        )
+
+
+@with_native_and_http_sessions
+class Date32TestCase(BaseTestCase):
+    required_server_version = (21, 9, 0)
+
+    table = Table(
+        'test', BaseTestCase.metadata(),
+        Column('x', types.Date32, primary_key=True),
+        engines.Memory()
+    )
+
+    def test_select_insert(self):
+        # Use a date before epoch to validate dates before epoch can be stored.
+        date = datetime.date(1925, 1, 1)
+        with self.create_table(self.table):
+            self.session.execute(self.table.insert(), [{'x': date}])
+            result = self.session.execute(self.table.select()).scalar()
+            if isinstance(result, datetime.date):
+                self.assertEqual(result, date)
+            else:
+                self.assertEqual(result, date.isoformat())