Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 11 additions & 2 deletions src/databricks/sqlalchemy/_ddl.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import logging
import re

from sqlalchemy.sql import compiler, sqltypes
import logging

logger = logging.getLogger(__name__)

Expand Down Expand Up @@ -68,7 +69,15 @@ def get_column_specification(self, column, **kwargs):
See comments in test_suite.py. We may implement implicit IDENTITY using this
feature in the future, similar to the Microsoft SQL Server dialect.
"""
if column is column.table._autoincrement_column or column.autoincrement is True:
is_autoincrement = bool(column.autoincrement)
if (
column.table is not None # column.table can be None
and not is_autoincrement
):
if column is column.table._autoincrement_column:
is_autoincrement = True

if is_autoincrement:
logger.warning(
"Databricks dialect ignores SQLAlchemy's autoincrement semantics. Use explicit Identity() instead."
)
Expand Down