Fix bug to support sqlalchemy v1.4

src/snowflake/sqlalchemy/sql/custom_schema/custom_table_base.py

@@ -24,8 +24,13 @@ def __init__(
         **kw: Any,
     ) -> None:
         if self.__table_prefix__ != "":
-            kw.update(prefixes=kw.get("prefixes", []) + [self.__table_prefix__])
-        super().__init__(name, metadata, *args, **kw)
+            prefixes = kw.get("prefixes", []) + [self.__table_prefix__]
+            kw.update(prefixes=prefixes)
+        if kw["alternative_initializer"] and hasattr(super(), "_init"):
+            super()._init(name, metadata, *args, **kw)
+        else:
+            super().__init__(name, metadata, *args, **kw)
+
         if not kw.get("autoload_with", False):
             self._validate_table()
 

src/snowflake/sqlalchemy/sql/custom_schema/dynamic_table.py

@@ -5,6 +5,7 @@
 import typing
 from typing import Any
 
+from sqlalchemy.exc import ArgumentError
 from sqlalchemy.sql.schema import MetaData, SchemaItem
 
 from snowflake.sqlalchemy.custom_commands import NoneType
@@ -49,6 +50,16 @@ def __init__(
             return
         super().__init__(name, metadata, *args, **kw)
 
+    def _init(
+        self,
+        name: str,
+        metadata: MetaData,
+        *args: SchemaItem,
+        **kw: Any,
+    ) -> None:
+        kw["alternative_initializer"] = True
+        super().__init__(name, metadata, *args, **kw)
+
     def _validate_table(self):
         missing_attributes = []
         if self.target_lag is NoneType:
@@ -58,7 +69,7 @@ def _validate_table(self):
         if self.as_query is NoneType:
             missing_attributes.append("AsQuery")
         if missing_attributes:
-            raise exc.ArgumentError(
+            raise ArgumentError(
                 "DYNAMIC TABLE must have the following arguments: %s"
                 % ", ".join(missing_attributes)
             )
@@ -70,5 +81,7 @@ def __repr__(self) -> str:
             + [repr(self.metadata)]
             + [repr(x) for x in self.columns]
             + [repr(self.target_lag)]
+            + [repr(self.warehouse)]
+            + [repr(self.as_query)]
             + [f"{k}={repr(getattr(self, k))}" for k in ["schema"]]
         )

src/snowflake/sqlalchemy/sql/custom_schema/table_from_query.py

@@ -4,7 +4,6 @@
 import typing
 from typing import Any, Optional
 
-import sqlalchemy
 from sqlalchemy.sql import Selectable
 from sqlalchemy.sql.schema import Column, MetaData, SchemaItem
 from sqlalchemy.util import NoneType
@@ -53,7 +52,7 @@ def __has_defined_columns(self, items: typing.List[SchemaItem]) -> bool:
     def __create_columns_from_selectable(
         self, selectable: Selectable
     ) -> Optional[typing.List[Column]]:
-        if not isinstance(selectable, sqlalchemy.Selectable):
+        if not isinstance(selectable, Selectable):
             return
         columns: typing.List[Column] = []
         for _, c in selectable.exported_columns.items():