Autoincrement testing

PeterDKay · PeterDKay · commit d4171f5ded94 · 2024-01-11T18:23:11.000Z
diff --git a/src/databricks/sqlalchemy/dialect/base.py b/src/databricks/sqlalchemy/dialect/base.py
@@ -97,20 +97,19 @@ def visit_create_table(self, create, **kw):
         liquid_cluster_columns = []
         for create_column in create.columns:
             column = create_column.element
-            # column is of type <class 'sqlalchemy.sql.schema.Column'>
-            print(column.autoincrement)
-            # Add somewhere 'if column.autoincrement:' And append correct syntax
             try:
                 processed = self.process(
                     create_column, first_pk=column.primary_key and not first_pk
                 )
-                print(processed)
+                if column.autoincrement:   # If doesn't work try 'is True' and == 'True'
+                    processed = "`".join(processed.split("`")[:-1]) + "`" + "BIGINT GENERATED ALWAYS AS IDENTITY"
                 if processed is not None:
                     text += separator
                     separator = ", \n"
                     text += "\t" + processed
                 if column.primary_key:
                     first_pk = True
+                print(processed)
             except exc.CompileError as ce:
                 util.raise_(
                     exc.CompileError(
diff --git a/src/databricks/sqlalchemy/dialect/compiler.py b/src/databricks/sqlalchemy/dialect/compiler.py
@@ -4,8 +4,7 @@
 class DatabricksTypeCompiler(compiler.GenericTypeCompiler):
     """Originally forked from pyhive"""
 
-    def visit_INTEGER(self, type_, **kwargs):
-        print(kwargs)
+    def visit_INTEGER(self, type_):
         return "INT"
 
     def visit_NUMERIC(self, type_):
