snowflake-sqlalchemy 1.5.2__py3-none-any.whl

snowflake/sqlalchemy/custom_types.py

@@ -0,0 +1,105 @@
+#
+# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+#
+
+import sqlalchemy.types as sqltypes
+import sqlalchemy.util as util
+
+TEXT = sqltypes.VARCHAR
+CHARACTER = sqltypes.CHAR
+DEC = sqltypes.DECIMAL
+DOUBLE = sqltypes.FLOAT
+FIXED = sqltypes.DECIMAL
+NUMBER = sqltypes.DECIMAL
+BYTEINT = sqltypes.SMALLINT
+STRING = sqltypes.VARCHAR
+TINYINT = sqltypes.SMALLINT
+VARBINARY = sqltypes.BINARY
+
+
+def _process_float(value):
+    if value == float("inf"):
+        return "inf"
+    elif value == float("-inf"):
+        return "-inf"
+    elif value is not None:
+        return float(value)
+    return value
+
+
+class SnowflakeType(sqltypes.TypeEngine):
+    def _default_dialect(self):
+        # Get around circular import
+        return __import__("snowflake.sqlalchemy").sqlalchemy.dialect()
+
+
+class VARIANT(SnowflakeType):
+    __visit_name__ = "VARIANT"
+
+
+class OBJECT(SnowflakeType):
+    __visit_name__ = "OBJECT"
+
+
+class ARRAY(SnowflakeType):
+    __visit_name__ = "ARRAY"
+
+
+class TIMESTAMP_TZ(SnowflakeType):
+    __visit_name__ = "TIMESTAMP_TZ"
+
+
+class TIMESTAMP_LTZ(SnowflakeType):
+    __visit_name__ = "TIMESTAMP_LTZ"
+
+
+class TIMESTAMP_NTZ(SnowflakeType):
+    __visit_name__ = "TIMESTAMP_NTZ"
+
+
+class GEOGRAPHY(SnowflakeType):
+    __visit_name__ = "GEOGRAPHY"
+
+
+class GEOMETRY(SnowflakeType):
+    __visit_name__ = "GEOMETRY"
+
+
+class _CUSTOM_Date(SnowflakeType, sqltypes.Date):
+    def literal_processor(self, dialect):
+        def process(value):
+            if value is not None:
+                return f"'{value.isoformat()}'"
+
+        return process
+
+
+class _CUSTOM_DateTime(SnowflakeType, sqltypes.DateTime):
+    def literal_processor(self, dialect):
+        def process(value):
+            if value is not None:
+                datetime_str = value.isoformat(" ", timespec="microseconds")
+                return f"'{datetime_str}'"
+
+        return process
+
+
+class _CUSTOM_Time(SnowflakeType, sqltypes.Time):
+    def literal_processor(self, dialect):
+        def process(value):
+            if value is not None:
+                time_str = value.isoformat(timespec="microseconds")
+                return f"'{time_str}'"
+
+        return process
+
+
+class _CUSTOM_Float(SnowflakeType, sqltypes.Float):
+    def bind_processor(self, dialect):
+        return _process_float
+
+
+class _CUSTOM_DECIMAL(SnowflakeType, sqltypes.DECIMAL):
+    @util.memoized_property
+    def _type_affinity(self):
+        return sqltypes.INTEGER if self.scale == 0 else sqltypes.DECIMAL

snowflake/sqlalchemy/provision.py

@@ -0,0 +1,12 @@
+#
+# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+#
+from sqlalchemy.testing.provision import set_default_schema_on_connection
+
+
+# This is only for test purpose required by Requirement "default_schema_name_switch"
+@set_default_schema_on_connection.for_db("snowflake")
+def _snowflake_set_default_schema_on_connection(cfg, dbapi_connection, schema_name):
+    cursor = dbapi_connection.cursor()
+    cursor.execute(f"USE SCHEMA {dbapi_connection.database}.{schema_name};")
+    cursor.close()

snowflake/sqlalchemy/requirements.py

@@ -0,0 +1,297 @@
+#
+# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+#
+
+from sqlalchemy.testing import exclusions
+from sqlalchemy.testing.requirements import SuiteRequirements
+
+
+class Requirements(SuiteRequirements):
+    """
+    stay-closed properties
+
+    1. not supported in snowflake
+
+    - autocommit: sqlalchemy's autocommit isolation level concept does not apply to snowflake
+    - isolation_level: snowflake only supports read committed
+      - ref docs: https://docs.snowflake.com/en/sql-reference/transactions.html#label-txn-autocommit
+                  https://docs.sqlalchemy.org/en/14/core/connections.html#setting-transaction-isolation-levels-including-dbapi-autocommit
+    - index_ddl_if_exists: index not supported in snowflake
+    - non_updating_cascade: updating cascade supported
+    - empty_inserts: not supported in snowflake
+    - full_returning: not supported in snowflake
+    - insert_executemany_returning: not supported in snowflake
+    - returning: not supported in snowflake
+    - indexes_with_expressions: index not supported in snowflake
+    - check_constraint_reflection: not supported in snowflake
+    - reflect_tables_no_columns: not supported in snowflake
+    - server_side_cursors: no supported in snowflake
+    - index_reflects_included_columns: index not supported in snowflake
+    - savepoints: not supported in snowflake
+    - two_phase_transactions: not supported in snowflake
+    - async_dialect: no await used
+    - fetch_expression: not supported in snowflake
+    - fetch_percent: not supported in snowflake
+    - fetch_ties: not supported in snowflake
+    - supports_distinct_on: not supported in snowflake
+    - time_timezone: not supported in snowflake
+    - identity_columns_standard: not supported in snowflake, snowflake does not support setting identity with min max
+    - computed_columns: TODO: not supported in snowflake yet, check SNOW-169530 for virtual column
+    - computed_columns_default_persisted: TODO: not supported in snowflake yet, check SNOW-169530 for virtual column
+    - computed_columns_reflect_persisted: TODO: not supported in snowflake yet, check SNOW-169530 for virtual column
+    - computed_columns_virtual: TODO: not supported in snowflake yet, check SNOW-169530 for virtual column
+    - computed_columns_stored: TODO: not supported in snowflake yet, check SNOW-169530 for virtual column
+
+    2. potential service side issue / unclear service behavior
+
+    - foreign_key_constraint_option_reflection_ondelete: TODO: check service side issue or by design?
+    - fk_constraint_option_reflection_ondelete_restrict: TODO: check service side issue or by design?
+    - foreign_key_constraint_option_reflection_onupdate: TODO: check service side issue or by design?
+    - fk_constraint_option_reflection_onupdate_restrict: TODO: check service side issue or by design?
+
+    3. connector missing feature
+
+    - dbapi_lastrowid: TODO, not supported in snowflake python connector, support it in the future
+    - supports_lastrowid: TODO: not supported, check SNOW-11155
+
+    4. sqlalchemy potentially missing feature
+    note: not sure whether these to be supported
+
+    - collate: TODO: order_by_collation
+    - datetime_timezone: TODO: default for datetime type, snowflake uses TIMESTAMP_NTZ which
+                          contains no time zone info consider creating a new column type TIMESTAMP_TZ for the
+                          the time zone info
+      - ref: https://docs.snowflake.com/en/sql-reference/data-types-datetime.html#timestamp-ltz-timestamp-ntz-timestamp-tz
+    """
+
+    @property
+    def table_ddl_if_exists(self):
+        return exclusions.open()
+
+    @property
+    def table_value_constructor(self):
+        return exclusions.open()
+
+    @property
+    def deferrable_fks(self):
+        return exclusions.open()
+
+    @property
+    def boolean_col_expressions(self):
+        return exclusions.open()
+
+    @property
+    def nullsordering(self):
+        return exclusions.open()
+
+    @property
+    def standalone_binds(self):
+        return exclusions.open()
+
+    @property
+    def intersect(self):
+        return exclusions.open()
+
+    @property
+    def except_(self):
+        return exclusions.open()
+
+    @property
+    def window_functions(self):
+        return exclusions.open()
+
+    @property
+    def ctes(self):
+        return exclusions.open()
+
+    @property
+    def ctes_on_dml(self):
+        return exclusions.open()
+
+    @property
+    def tuple_in(self):
+        return exclusions.open()
+
+    @property
+    def emulated_lastrowid(self):
+        return exclusions.open()
+
+    @property
+    def emulated_lastrowid_even_with_sequences(self):
+        return exclusions.open()
+
+    @property
+    def views(self):
+        return exclusions.open()
+
+    @property
+    def cross_schema_fk_reflection(self):
+        return exclusions.open()
+
+    @property
+    def foreign_key_constraint_name_reflection(self):
+        return exclusions.open()
+
+    @property
+    def implicit_default_schema(self):
+        return exclusions.open()
+
+    @property
+    def default_schema_name_switch(self):
+        return exclusions.open()
+
+    @property
+    def reflects_pk_names(self):
+        return exclusions.open()
+
+    @property
+    def comment_reflection(self):
+        return exclusions.open()
+
+    @property
+    def fk_constraint_option_reflection_ondelete_noaction(self):
+        return exclusions.open()
+
+    @property
+    def temp_table_names(self):
+        return exclusions.open()
+
+    @property
+    def temporary_views(self):
+        return exclusions.open()
+
+    @property
+    def unicode_ddl(self):
+        return exclusions.open()
+
+    @property
+    def datetime_literals(self):
+        return exclusions.open()
+
+    @property
+    def timestamp_microseconds(self):
+        return exclusions.open()
+
+    @property
+    def datetime_historic(self):
+        return exclusions.open()
+
+    @property
+    def date_historic(self):
+        return exclusions.open()
+
+    @property
+    def legacy_unconditional_json_extract(self):
+        return exclusions.open()
+
+    @property
+    def precision_numerics_enotation_small(self):
+        return exclusions.open()
+
+    @property
+    def precision_numerics_enotation_large(self):
+        return exclusions.open()
+
+    @property
+    def precision_numerics_many_significant_digits(self):
+        return exclusions.open()
+
+    @property
+    def precision_numerics_retains_significant_digits(self):
+        return exclusions.open()
+
+    @property
+    def infinity_floats(self):
+        return exclusions.open()
+
+    @property
+    def update_from(self):
+        return exclusions.open()
+
+    @property
+    def delete_from(self):
+        return exclusions.open()
+
+    @property
+    def mod_operator_as_percent_sign(self):
+        return exclusions.open()
+
+    @property
+    def percent_schema_names(self):
+        return exclusions.open()
+
+    @property
+    def order_by_label_with_expression(self):
+        return exclusions.open()
+
+    @property
+    def regexp_match(self):
+        return exclusions.open()
+
+    @property
+    def regexp_replace(self):
+        return exclusions.open()
+
+    @property
+    def fetch_first(self):
+        return exclusions.open()
+
+    @property
+    def fetch_no_order_by(self):
+        return exclusions.open()
+
+    @property
+    def fetch_offset_with_options(self):
+        return exclusions.open()
+
+    @property
+    def identity_columns(self):
+        return exclusions.open()
+
+    @property
+    def duplicate_key_raises_integrity_error(self):
+        # Snowflake allows duplicate value for primary key
+        return exclusions.closed()
+
+    @property
+    def ctes_with_update_delete(self):
+        # Snowflake CTE could only be followed by SELECT
+        # https://docs.snowflake.com/en/user-guide/queries-cte.html
+        return exclusions.closed()
+
+    @property
+    def sql_expression_limit_offset(self):
+        # Snowflake only takes non-negative integer constants for offset/limit
+        return exclusions.closed()
+
+    @property
+    def json_type(self):
+        # TODO: need service/connector support
+        # check https://snowflakecomputing.atlassian.net/browse/SNOW-52370
+        return exclusions.closed()
+
+    @property
+    def implements_get_lastrowid(self):
+        # TODO: need connector lastrowid support, check SNOW-11155
+        return exclusions.closed()
+
+    @property
+    def implicit_decimal_binds(self):
+        # Supporting this would require behavior breaking change to implicitly convert str to Decimal when binding
+        # parameters in string forms of decimal values.
+        # Check https://snowflakecomputing.atlassian.net/browse/SNOW-640134 for details on breaking changes discussion.
+        return exclusions.closed()
+
+    @property
+    def datetime_implicit_bound(self):
+        # Supporting this would require behavior breaking change to implicitly convert str to datetime when binding
+        # parameters in string forms of datetime values.
+        # Check https://snowflakecomputing.atlassian.net/browse/SNOW-640134 for details on breaking changes discussion.
+        return exclusions.closed()
+
+    @property
+    def timestamp_microseconds_implicit_bound(self):
+        # Supporting this would require behavior breaking change to implicitly convert str to timestamp when binding
+        # parameters in string forms of timestamp values.
+        # Check https://snowflakecomputing.atlassian.net/browse/SNOW-640134 for details on breaking changes discussion.
+        return exclusions.closed()