support dateadd (#7)

discover-labs · Nov 27, 2022 · 265ce07 · 265ce07
1 parent f46fa29
commit 265ce07
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 54 deletions.
diff --git a/dictum_backend_vertica/__init__.py b/dictum_backend_vertica/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
diff --git a/dictum_backend_vertica/vertica_backend.py b/dictum_backend_vertica/vertica_backend.py
@@ -3,6 +3,7 @@
 from dictum_core.backends.secret import Secret
 from dictum_core.backends.sql_alchemy import SQLAlchemyBackend, SQLAlchemyCompiler
 from sqlalchemy import Integer, cast
+from sqlalchemy.sql import func
 
 
 class VerticaCompiler(SQLAlchemyCompiler):
@@ -31,6 +32,9 @@ def datediff(self, part, start, end):
             )
         return super().datediff(part, start, end)
 
+    def dateadd(self, part, interval, value):
+        return func.timestampadd(part, interval, value)
+
 
 class VerticaBackend(SQLAlchemyBackend):
     type = "vertica"

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dictum-backend-vertica"
-version = "0.1.4"
+version = "0.1.5"
 description = "Vertica backend for Dictum"
 authors = ["Mikhail Akimov"]
 

diff --git a/tests/test_dictum_backend_vertica.py b/tests/test_dictum_backend_vertica.py
@@ -9,7 +9,7 @@
 
 
 def test_version():
-    assert __version__ == "0.1.4"
+    assert __version__ == "0.1.5"
 
 
 def test_entry_point():