soda-sqlserver 4.7.0__tar.gz → 4.14.0__tar.gz

{soda_sqlserver-4.7.0 → soda_sqlserver-4.14.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: soda-sqlserver
-Version: 4.7.0
+Version: 4.14.0
 Summary: Soda SQL Server V4
 Author-email: "Soda Data N.V." <info@soda.io>
 License: Proprietary
 Requires-Python: >=3.10
-Requires-Dist: soda-core==4.7.0
+Requires-Dist: soda-core==4.14.0
 Requires-Dist: pyodbc

{soda_sqlserver-4.7.0 → soda_sqlserver-4.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "soda-sqlserver"
-version = "4.7.0"
+version = "4.14.0"
 description = "Soda SQL Server V4"
 requires-python = ">=3.10"
 license = {text = "Proprietary"}
@@ -8,7 +8,7 @@ authors = [
     {name = "Soda Data N.V.", email = "info@soda.io"}
 ]
 dependencies = [
-    "soda-core==4.7.0",
+    "soda-core==4.14.0",
     "pyodbc",
 ]
 

{soda_sqlserver-4.7.0 → soda_sqlserver-4.14.0}/src/soda_sqlserver/common/data_sources/sqlserver_data_source.py

@@ -14,6 +14,7 @@ from soda_core.common.sql_ast import (
     COUNT,
     CREATE_TABLE,
     CREATE_TABLE_AS_SELECT,
+    CREATE_TABLE_COLUMN,
     CREATE_TABLE_IF_NOT_EXISTS,
     CREATE_VIEW,
     DISTINCT,
@@ -162,6 +163,11 @@ class SqlServerSqlDialect(SqlDialect, sqlglot_dialect="tsql"):
     def _build_length_sql(self, length: LENGTH) -> str:
         return f"LEN({self.build_expression_sql(length.expression)})"
 
+    def _build_count_sql(self, count: COUNT) -> str:
+        # T-SQL COUNT returns INT and overflows above 2,147,483,647 rows. COUNT_BIG is the
+        # BIGINT-returning equivalent with identical null/distinct semantics.
+        return f"COUNT_BIG({self.build_expression_sql(count.expression)})"
+
     def sql_expr_timestamp_literal(self, datetime_in_iso8601: str) -> str:
         return f"'{datetime_in_iso8601}'"
 
@@ -324,6 +330,22 @@ class SqlServerSqlDialect(SqlDialect, sqlglot_dialect="tsql"):
             "datetimeoffset",
         ]
 
+    # SQL Server's datetime2 / datetimeoffset / time accept a fractional-seconds
+    # precision of 0..7. Any higher value is rejected at CREATE TABLE time.
+    _MAX_DATETIME_PRECISION = 7
+
+    def _build_create_table_column_type(self, create_table_column: CREATE_TABLE_COLUMN) -> str:
+        # Clamp datetime precision to SQL Server's max. Cross-source flows from sources
+        # with higher native precision (e.g. Snowflake's TIMESTAMP_NTZ defaults to 9)
+        # would otherwise produce e.g. `datetime2(9)` and fail CREATE TABLE.
+        if create_table_column.type.name.lower() in ("datetime2", "datetimeoffset", "time"):
+            if (
+                create_table_column.type.datetime_precision is not None
+                and create_table_column.type.datetime_precision > self._MAX_DATETIME_PRECISION
+            ):
+                create_table_column.type.datetime_precision = self._MAX_DATETIME_PRECISION
+        return super()._build_create_table_column_type(create_table_column)
+
     def default_varchar_length(self) -> Optional[int]:
         return 255
 

{soda_sqlserver-4.7.0 → soda_sqlserver-4.14.0}/src/soda_sqlserver/common/data_sources/sqlserver_data_source_connection.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import logging
 import struct
 from abc import ABC
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timedelta, timezone, tzinfo
 from typing import Any, Literal, Optional, Union
 
 import pyodbc
@@ -196,5 +196,28 @@ class SqlServerDataSourceConnection(DataSourceConnection):
     def _execute_query_get_result_row_column_name(self, column) -> str:
         return column[0]
 
+    def _fetch_session_timezone(self) -> tzinfo:
+        # Use SYSDATETIMEOFFSET() instead of CURRENT_TIMEZONE_ID() so the same query works
+        # across the whole SQL Server family: SQL Server proper, Microsoft Fabric DW, Azure
+        # Synapse Dedicated/Serverless. CURRENT_TIMEZONE_ID is unsupported in Fabric and
+        # Synapse Dedicated; SYSDATETIMEOFFSET is universally supported.
+        # The connection registers ``handle_datetimeoffset`` as a pyodbc output converter for
+        # SQL_TYPE -155, so the returned value is a tz-aware ``datetime`` whose tzinfo is the
+        # exact offset reported by the server.
+        with self.connection.cursor() as cursor:
+            cursor.execute("SELECT SYSDATETIMEOFFSET()")
+            row = cursor.fetchone()
+        if not row or not isinstance(row[0], datetime) or row[0].tzinfo is None:
+            return timezone.utc
+        offset = row[0].tzinfo.utcoffset(row[0])
+        # Normalize zero offset to ``timezone.utc`` (the singleton) so that adapters
+        # routed through ``parse_session_timezone`` and SQL Server agree on the UTC
+        # representation — adapters that report ``UTC`` / ``Etc/UTC`` resolve to the
+        # same ``timezone.utc`` instance, while a raw ``timezone(timedelta(0))`` would
+        # be ``==`` but not ``is`` equivalent.
+        if offset == timedelta(0):
+            return timezone.utc
+        return timezone(offset)
+
     def _get_autocommit_setting(self) -> bool:
         return False  # No need to set autocommit, as it is set to False by default.

{soda_sqlserver-4.7.0 → soda_sqlserver-4.14.0}/src/soda_sqlserver.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: soda-sqlserver
-Version: 4.7.0
+Version: 4.14.0
 Summary: Soda SQL Server V4
 Author-email: "Soda Data N.V." <info@soda.io>
 License: Proprietary
 Requires-Python: >=3.10
-Requires-Dist: soda-core==4.7.0
+Requires-Dist: soda-core==4.14.0
 Requires-Dist: pyodbc

soda_sqlserver-4.14.0/src/soda_sqlserver.egg-info/requires.txt

@@ -0,0 +1,2 @@
+soda-core==4.14.0
+pyodbc

soda_sqlserver-4.7.0/src/soda_sqlserver.egg-info/requires.txt

@@ -1,2 +0,0 @@
-soda-core==4.7.0
-pyodbc