From 9612580071fc85e79f272d670bc9c552fddf2b19 Mon Sep 17 00:00:00 2001
From: Alex Ryndin <me@alexryndin.me>
Date: Thu, 25 Mar 2021 17:36:09 +0300
Subject: [PATCH] fix timetz and timestamptz datatypes

---
 sqlalchemy_vertica/base.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/sqlalchemy_vertica/base.py b/sqlalchemy_vertica/base.py
index 7fe1605..f5e5644 100644
--- a/sqlalchemy_vertica/base.py
+++ b/sqlalchemy_vertica/base.py
@@ -13,6 +13,14 @@
     NUMERIC, FLOAT, REAL, DATE, DATETIME, BOOLEAN, BLOB, TIMESTAMP, TIME
 from sqlalchemy.sql import sqltypes
 
+def TIMESTAMP_WITH_TIMEZONE(*args, **kwargs):
+    kwargs['timezone'] = True
+    return TIMESTAMP(*args, **kwargs)
+
+def TIME_WITH_TIMEZONE(*args, **kwargs):
+    kwargs['timezone'] = True
+    return TIME(*args, **kwargs)
+
 ischema_names = {
     'INT': INTEGER,
     'INTEGER': INTEGER,
@@ -34,10 +42,10 @@
     'DOUBLE': DOUBLE_PRECISION,
     'TIMESTAMP': TIMESTAMP,
     'TIMESTAMP WITH TIMEZONE': TIMESTAMP,
-    'TIMESTAMPTZ': TIMESTAMP(timezone=True),
+    'TIMESTAMPTZ': TIMESTAMP_WITH_TIMEZONE,
     'TIME': TIME,
     'TIME WITH TIMEZONE': TIME,
-    'TIMETZ': TIME(timezone=True),
+    'TIMETZ': TIME_WITH_TIMEZONE,
     'INTERVAL': INTERVAL,
     'DATE': DATE,
     'DATETIME': DATETIME,