Merge pull request #29 from jordiyeh/master

Add IDENTITY support when creating DDL

Author: lemeryfertitta

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup
 
-version = '0.4.1'
+version = '0.4.2'
 setup(
     name='sqlalchemy-vertica-python',
     version=version,

sqla_vertica_python/vertica_python.py

@@ -3,6 +3,17 @@
 from sqlalchemy.dialects.postgresql.base import PGDialect
 from sqlalchemy.dialects.postgresql import INTERVAL
 from sqlalchemy.engine import reflection
+from sqlalchemy.schema import CreateColumn
+from sqlalchemy.ext.compiler import compiles
+
+
+# From Postgresql 10 IDENTITY columns section in sqlalchemy/dialects/postgresql/base.py
+@compiles(CreateColumn, 'vertica')
+def use_identity(element, compiler, **kw):
+    text = compiler.visit_create_column(element, **kw)
+    text = text.replace("SERIAL", "IDENTITY(1,1)")
+    return text
+
 
 class VerticaDialect(PGDialect):
     """ Vertica Dialect using a vertica-python connection and PGDialect """