soda-postgres 4.3.0__tar.gz → 4.14.0__tar.gz

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: soda-postgres
-Version: 4.3.0
+Version: 4.14.0
 Summary: Soda Postgres V4
 Author-email: "Soda Data N.V." <info@soda.io>
 License: Proprietary
 Requires-Python: >=3.10
-Requires-Dist: soda-core==4.3.0
+Requires-Dist: soda-core==4.14.0
 Requires-Dist: psycopg[binary]>=3.2

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "soda-postgres"
-version = "4.3.0"
+version = "4.14.0"
 description = "Soda Postgres 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.3.0",
+    "soda-core==4.14.0",
     "psycopg[binary]>=3.2",
 ]
 

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/src/soda_postgres/common/data_sources/postgres_data_source.py

@@ -264,11 +264,11 @@ class PostgresSqlDialect(SqlDialect, sqlglot_dialect="postgres"):
 
         select_columns = []
         if include_table_name_column:
-            select_columns.append(COLUMN("relname", table_alias="c", field_alias="table_name"))
+            select_columns.append(COLUMN("relname", table_alias="c").AS("table_name"))
 
         select_columns.extend(
             [
-                COLUMN("attname", table_alias="a", field_alias="column_name"),
+                COLUMN("attname", table_alias="a").AS("column_name"),
                 # Normalize data type into information_schema.columns style.
                 RAW_SQL(
                     """CASE
@@ -345,9 +345,9 @@ class PostgresSqlDialect(SqlDialect, sqlglot_dialect="postgres"):
                     END AS "datetime_precision"
                 """
                 ),
-                COLUMN(current_database_expression, field_alias="table_catalog"),
-                COLUMN("nspname", table_alias="n", field_alias="table_schema"),
-                COLUMN("relname", table_alias="c", field_alias="table_name"),
+                current_database_expression.AS("table_catalog"),
+                COLUMN("nspname", table_alias="n").AS("table_schema"),
+                COLUMN("relname", table_alias="c").AS("table_name"),
                 RAW_SQL(self.relkind_table_type_sql_expression()),
             ]
         )

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/src/soda_postgres/common/data_sources/postgres_data_source_connection.py

@@ -2,13 +2,17 @@ from __future__ import annotations
 
 import logging
 from abc import ABC
+from datetime import timezone, tzinfo
 from pathlib import Path
 from typing import Callable, ClassVar, Dict, Literal, Optional, Union
 
 import psycopg
 from pydantic import Field, IPvAnyAddress, SecretStr, field_validator
-from soda_core.common.data_source_connection import DataSourceConnection
-from soda_core.common.data_source_results import QueryResult, UpdateResult
+from soda_core.common.data_source_connection import (
+    DataSourceConnection,
+    parse_session_timezone,
+)
+from soda_core.common.data_source_results import QueryResult
 from soda_core.common.logging_constants import soda_logger
 from soda_core.model.data_source.data_source import DataSourceBase
 from soda_core.model.data_source.data_source_connection_properties import (
@@ -86,6 +90,14 @@ class PostgresDataSourceConnection(DataSourceConnection):
         connection = psycopg.connect(**connection_kwargs)
         return connection
 
+    def _fetch_session_timezone(self) -> tzinfo:
+        with self.connection.cursor() as cursor:
+            cursor.execute("SHOW timezone")
+            row = cursor.fetchone()
+        if not row:
+            return timezone.utc
+        return parse_session_timezone(row[0])
+
     def execute_query(self, sql: str, log_query: bool = True) -> QueryResult:
         try:
             return super().execute_query(sql, log_query=log_query)
@@ -95,7 +107,7 @@ class PostgresDataSourceConnection(DataSourceConnection):
             self.rollback()
             raise e
 
-    def execute_update(self, sql: str, log_query: bool = True) -> UpdateResult:
+    def execute_update(self, sql: str, log_query: bool = True) -> int:
         try:
             return super().execute_update(sql, log_query=log_query)
         except psycopg.errors.Error as e:  # Catch the error and roll back the transaction

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/src/soda_postgres/statements/postgres_metadata_tables_query.py

@@ -35,9 +35,9 @@ class PostgresMetadataTablesQuery(MetadataTablesQuery):
         select: list = [
             SELECT(
                 [
-                    COLUMN(current_database_expression, field_alias="table_catalog"),
-                    COLUMN("nspname", table_alias="n", field_alias="table_schema"),
-                    COLUMN("relname", table_alias="c", field_alias="table_name"),
+                    current_database_expression.AS("table_catalog"),
+                    COLUMN("nspname", table_alias="n").AS("table_schema"),
+                    COLUMN("relname", table_alias="c").AS("table_name"),
                     RAW_SQL(self.sql_dialect.relkind_table_type_sql_expression()),
                 ]
             ),

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/src/soda_postgres.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: soda-postgres
-Version: 4.3.0
+Version: 4.14.0
 Summary: Soda Postgres V4
 Author-email: "Soda Data N.V." <info@soda.io>
 License: Proprietary
 Requires-Python: >=3.10
-Requires-Dist: soda-core==4.3.0
+Requires-Dist: soda-core==4.14.0
 Requires-Dist: psycopg[binary]>=3.2

{soda_postgres-4.3.0 → soda_postgres-4.14.0}/src/soda_postgres.egg-info/requires.txt

@@ -1,2 +1,2 @@
-soda-core==4.3.0
+soda-core==4.14.0
 psycopg[binary]>=3.2