google-cloud-bigquery 3.42.3__tar.gz → 3.43.0__tar.gz

{google_cloud_bigquery-3.42.3/google_cloud_bigquery.egg-info → google_cloud_bigquery-3.43.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-bigquery
-Version: 3.42.3
+Version: 3.43.0
 Summary: Google BigQuery API client library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache 2.0
@@ -19,7 +19,7 @@ Classifier: Topic :: Internet
 Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE
-Requires-Dist: google-api-core[grpc]<3.0.0,>=2.25.0
+Requires-Dist: google-api-core[grpc]<3.0.0,>=2.28.0
 Requires-Dist: google-auth[pyopenssl]<3.0.0,>=2.14.1
 Requires-Dist: google-cloud-core<3.0.0,>=2.4.1
 Requires-Dist: google-resumable-media<3.0.0,>=2.0.0
@@ -85,6 +85,26 @@ processing power of Google's infrastructure.
 .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/bigquery/latest/summary_overview
 .. _Product Documentation: https://cloud.google.com/bigquery/docs/reference/v2/
 
+
+Related Libraries
+-----------------
+
+For higher-level data analysis and DataFrame operations, consider using one of the following connector libraries:
+
+.. list-table::
+   :header-rows: 1
+
+   * - Library
+     - Description
+     - Documentation
+   * - `BigQuery DataFrames (bigframes) <https://dataframes.bigquery.dev/>`_
+     - Flexible pandas-like and scikit-learn-like API powered by BigQuery engine for large-scale data analysis and ML.
+     - `BigFrames Docs <https://dataframes.bigquery.dev/>`_
+   * - `pandas-gbq <https://pandas-gbq.readthedocs.io/>`_
+     - Convenient integration allowing pandas DataFrames to load from and write to BigQuery.
+     - `pandas-gbq Docs <https://pandas-gbq.readthedocs.io/>`_
+
+
 Quick Start
 -----------
 

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/README.rst

@@ -21,6 +21,26 @@ processing power of Google's infrastructure.
 .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/bigquery/latest/summary_overview
 .. _Product Documentation: https://cloud.google.com/bigquery/docs/reference/v2/
 
+
+Related Libraries
+-----------------
+
+For higher-level data analysis and DataFrame operations, consider using one of the following connector libraries:
+
+.. list-table::
+   :header-rows: 1
+
+   * - Library
+     - Description
+     - Documentation
+   * - `BigQuery DataFrames (bigframes) <https://dataframes.bigquery.dev/>`_
+     - Flexible pandas-like and scikit-learn-like API powered by BigQuery engine for large-scale data analysis and ML.
+     - `BigFrames Docs <https://dataframes.bigquery.dev/>`_
+   * - `pandas-gbq <https://pandas-gbq.readthedocs.io/>`_
+     - Convenient integration allowing pandas DataFrames to load from and write to BigQuery.
+     - `pandas-gbq Docs <https://pandas-gbq.readthedocs.io/>`_
+
+
 Quick Start
 -----------
 

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google/cloud/bigquery/_helpers.py

@@ -295,7 +295,7 @@ class CellDataParser:
             return _date_from_iso8601_date(value)
 
     def time_to_py(self, value, field):
-        """Coerce 'value' to a datetime date, if set or not nullable"""
+        """Coerce 'value' to a datetime time, if set or not nullable"""
         if _not_null(value, field):
             if len(value) == 8:  # HH:MM:SS
                 fmt = _TIMEONLY_WO_MICROS

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google/cloud/bigquery/_versions_helpers.py

@@ -16,10 +16,8 @@
 from typing import Any
 
 import packaging.version
-
 from google.cloud.bigquery import exceptions
 
-
 _MIN_PYARROW_VERSION = packaging.version.Version("3.0.0")
 _MIN_BQ_STORAGE_VERSION = packaging.version.Version("2.0.0")
 _BQ_STORAGE_OPTIONAL_READ_SESSION_VERSION = packaging.version.Version("2.6.0")
@@ -27,6 +25,7 @@ _MIN_PANDAS_VERSION = packaging.version.Version("1.1.0")
 
 _MIN_PANDAS_VERSION_RANGE = packaging.version.Version("1.5.0")
 _MIN_PYARROW_VERSION_RANGE = packaging.version.Version("10.0.1")
+_MIN_PANDAS_GBQ_DELEGATION_VERSION = packaging.version.Version("1.0.0")
 
 
 class PyarrowVersions:
@@ -247,3 +246,52 @@ SUPPORTS_RANGE_PYARROW = (
     and PYARROW_VERSIONS.try_import() is not None
     and PYARROW_VERSIONS.installed_version >= _MIN_PYARROW_VERSION_RANGE
 )
+
+
+class PandasGBQVersions:
+    """Version and delegation comparisons for pandas-gbq package."""
+
+    def __init__(self):
+        self._installed_version = None
+        self._delegation_api_version = None
+
+    @property
+    def installed_version(self) -> packaging.version.Version:
+        """Return the parsed version of pandas-gbq."""
+        if self._installed_version is not None:
+            return self._installed_version
+
+        try:
+            import pandas_gbq  # type: ignore
+
+            self._installed_version = packaging.version.parse(
+                getattr(pandas_gbq, "__version__", "0.0.0")
+            )
+        except Exception:
+            self._installed_version = packaging.version.parse("0.0.0")
+        return self._installed_version
+
+    @property
+    def delegation_api_version(self) -> packaging.version.Version:
+        """Return the delegation API version of pandas-gbq if installed, otherwise 0.0.0."""
+        if self._delegation_api_version is not None:
+            return self._delegation_api_version
+
+        try:
+            import pandas_gbq  # type: ignore
+
+            raw_version = getattr(
+                pandas_gbq, "_internal_delegation_api_version", "0.0.0"
+            )
+            self._delegation_api_version = packaging.version.parse(str(raw_version))
+        except Exception:
+            self._delegation_api_version = packaging.version.parse("0.0.0")
+        return self._delegation_api_version
+
+    @property
+    def is_delegation_supported(self) -> bool:
+        """True if the installed pandas-gbq version supports query delegation API."""
+        return self.delegation_api_version >= _MIN_PANDAS_GBQ_DELEGATION_VERSION
+
+
+PANDAS_GBQ_VERSIONS = PandasGBQVersions()

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google/cloud/bigquery/dbapi/connection.py

@@ -78,10 +78,14 @@ class Connection(object):
         """
         self._closed = True
 
-        if self._owns_client:
+        for cursor_ in list(self._cursors_created):
+            if not cursor_._closed:
+                cursor_.close()
+
+        if self._owns_client and self._client is not None:
             self._client.close()
 
-        if self._owns_bqstorage_client:
+        if self._owns_bqstorage_client and self._bqstorage_client is not None:
             # There is no close() on the BQ Storage client itself.
             transport = self._bqstorage_client.transport
             transport.close()
@@ -95,10 +99,6 @@ class Connection(object):
             if channel is not None:
                 channel.close()
 
-        for cursor_ in self._cursors_created:
-            if not cursor_._closed:
-                cursor_.close()
-
     def commit(self):
         """No-op, but for consistency raise an error if connection is closed."""
 

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google/cloud/bigquery/query.py

@@ -1020,7 +1020,6 @@ class RangeQueryParameter(_AbstractQueryParameter):
     ):
         self.name = name
         self.range_element_type = self._parse_range_element_type(range_element_type)
-        print(self.range_element_type.type_._type)
         self.start = start
         self.end = end
 

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google/cloud/bigquery/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "3.42.3"
+__version__ = "3.43.0"

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0/google_cloud_bigquery.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-bigquery
-Version: 3.42.3
+Version: 3.43.0
 Summary: Google BigQuery API client library
 Author-email: Google LLC <googleapis-packages@google.com>
 License: Apache 2.0
@@ -19,7 +19,7 @@ Classifier: Topic :: Internet
 Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE
-Requires-Dist: google-api-core[grpc]<3.0.0,>=2.25.0
+Requires-Dist: google-api-core[grpc]<3.0.0,>=2.28.0
 Requires-Dist: google-auth[pyopenssl]<3.0.0,>=2.14.1
 Requires-Dist: google-cloud-core<3.0.0,>=2.4.1
 Requires-Dist: google-resumable-media<3.0.0,>=2.0.0
@@ -85,6 +85,26 @@ processing power of Google's infrastructure.
 .. _Client Library Documentation: https://cloud.google.com/python/docs/reference/bigquery/latest/summary_overview
 .. _Product Documentation: https://cloud.google.com/bigquery/docs/reference/v2/
 
+
+Related Libraries
+-----------------
+
+For higher-level data analysis and DataFrame operations, consider using one of the following connector libraries:
+
+.. list-table::
+   :header-rows: 1
+
+   * - Library
+     - Description
+     - Documentation
+   * - `BigQuery DataFrames (bigframes) <https://dataframes.bigquery.dev/>`_
+     - Flexible pandas-like and scikit-learn-like API powered by BigQuery engine for large-scale data analysis and ML.
+     - `BigFrames Docs <https://dataframes.bigquery.dev/>`_
+   * - `pandas-gbq <https://pandas-gbq.readthedocs.io/>`_
+     - Convenient integration allowing pandas DataFrames to load from and write to BigQuery.
+     - `pandas-gbq Docs <https://pandas-gbq.readthedocs.io/>`_
+
+
 Quick Start
 -----------
 

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/google_cloud_bigquery.egg-info/requires.txt

@@ -1,4 +1,4 @@
-google-api-core[grpc]<3.0.0,>=2.25.0
+google-api-core[grpc]<3.0.0,>=2.28.0
 google-auth[pyopenssl]<3.0.0,>=2.14.1
 google-cloud-core<3.0.0,>=2.4.1
 google-resumable-media<3.0.0,>=2.0.0

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/pyproject.toml

@@ -41,7 +41,7 @@ classifiers = [
   "Topic :: Internet",
 ]
 dependencies = [
-  "google-api-core[grpc] >= 2.25.0, < 3.0.0",
+  "google-api-core[grpc] >= 2.28.0, < 3.0.0",
   "google-auth[pyopenssl] >= 2.14.1, < 3.0.0",
   "google-cloud-core >= 2.4.1, < 3.0.0",
   "google-resumable-media >= 2.0.0, < 3.0.0",

{google_cloud_bigquery-3.42.3 → google_cloud_bigquery-3.43.0}/tests/system/test_client.py

@@ -204,12 +204,16 @@ class TestBigQuery(unittest.TestCase):
             tag_key = key_values.pop()
 
             # Delete tag values first
-            [
-                tag_values_client.delete_tag_value(name=tag_value.name).result()
-                for tag_value in key_values
-            ]
+            for tag_value in key_values:
+                try:
+                    tag_values_client.delete_tag_value(name=tag_value.name).result()
+                except NotFound:
+                    pass
 
-            tag_keys_client.delete_tag_key(name=tag_key.name).result()
+            try:
+                tag_keys_client.delete_tag_key(name=tag_key.name).result()
+            except NotFound:
+                pass
 
     def test_get_service_account_email(self):
         client = Config.CLIENT
@@ -2182,58 +2186,101 @@ class TestBigQuery(unittest.TestCase):
 
     def test_dbapi_connection_does_not_leak_sockets(self):
         pytest.importorskip("google.cloud.bigquery_storage")
-        current_process = psutil.Process()
-        conn_start = current_process.net_connections()
-        conn_count_start = len(conn_start)
 
-        with helpers.patch_tracked_requests():
-            # Provide no explicit clients, so that the connection will create and own them.
-            connection = dbapi.connect()
-            cursor = connection.cursor()
+        import google.api_core.grpc_helpers
+        import requests
 
-            cursor.execute(
-                """
-                SELECT id, `by`, timestamp
-                FROM `bigquery-public-data.hacker_news.full`
-                ORDER BY `id` ASC
-                LIMIT 100000
-            """
-            )
-            rows = cursor.fetchall()
-            self.assertEqual(len(rows), 100000)
+        # Track HTTP Sessions
+        created_sessions = set()
+        closed_sessions = set()
 
-            connection.close()
-        import gc
+        original_session_init = requests.Session.__init__
 
-        gc.collect()
-        for _ in range(30):  # Wait up to 3 seconds
-            conn_end = current_process.net_connections()
-            conn_count_end = len(conn_end)
-            if conn_count_end <= conn_count_start:
-                break
-            time.sleep(0.1)
+        def patched_session_init(self, *args, **kwargs):
+            original_session_init(self, *args, **kwargs)
+            created_sessions.add(self)
+
+            original_close = self.close
+
+            def patched_close(*s_args, **s_kwargs):
+                original_close(*s_args, **s_kwargs)
+                closed_sessions.add(self)
+
+            self.close = patched_close
+
+        # Track gRPC Channels
+        created_channels = set()
+        closed_channels = set()
+
+        original_create_channel = google.api_core.grpc_helpers.create_channel
+
+        def patched_create_channel(*args, **kwargs):
+            channel = original_create_channel(*args, **kwargs)
+            created_channels.add(channel)
+
+            original_channel_close = channel.close
+
+            def patched_channel_close(*c_args, **c_kwargs):
+                original_channel_close(*c_args, **c_kwargs)
+                closed_channels.add(channel)
+
+            channel.close = patched_channel_close
+            return channel
+
+        # Apply patches
+        requests.Session.__init__ = patched_session_init
+        google.api_core.grpc_helpers.create_channel = patched_create_channel
 
         try:
-            self.assertLessEqual(conn_count_end, conn_count_start)
-        except AssertionError as e:
-            # Due to flakiness in this test (likely caused by OS cleanup delays or
-            # non-deterministic garbage collection of sockets), we want to capture
-            # the detailed state of connections in future failing runs to help
-            # decrease false positives and identify the root cause.
-            conn_debug = [
-                f"Status: {c.status}, Laddr: {c.laddr}, Raddr: {c.raddr}"
-                for c in current_process.net_connections()
-            ]
-            debug_msg = "\n".join(conn_debug)
-
-            raise AssertionError(
-                f"{e}\n\n"
-                f"--- Socket Leak Debug Info ---\n"
-                f"Start Count: {conn_count_start}\n"
-                f"End Count: {conn_count_end}\n"
-                f"Current Connections:\n{debug_msg}"
+            with helpers.patch_tracked_requests():
+                # Provide no explicit clients, so that the connection will create and own them.
+                connection = dbapi.connect()
+                cursor = connection.cursor()
+
+                cursor.execute(
+                    """
+                    SELECT id, `by`, timestamp
+                    FROM `bigquery-public-data.hacker_news.full`
+                    ORDER BY `id` ASC
+                    LIMIT 100000
+                """
+                )
+                rows = cursor.fetchall()
+                self.assertEqual(len(rows), 100000)
+
+                connection.close()
+
+            # Assertions
+            self.assertEqual(
+                created_sessions,
+                closed_sessions,
+                f"HTTP Sessions leak detected! Leaked: {created_sessions - closed_sessions}",
+            )
+
+            self.assertEqual(
+                created_channels,
+                closed_channels,
+                f"gRPC Channels leak detected! Leaked: {created_channels - closed_channels}",
             )
 
+        finally:
+            # Revert patches
+            requests.Session.__init__ = original_session_init
+            google.api_core.grpc_helpers.create_channel = original_create_channel
+
+            # Clean up any unclosed sessions/channels to avoid leaking in the test runner
+            for s in created_sessions - closed_sessions:
+                try:
+                    s.close()
+                except Exception:
+                    pass
+
+            for c in created_channels - closed_channels:
+                try:
+                    c.close()
+                except Exception:
+                    pass
+
     def _load_table_for_dml(self, rows, dataset_id, table_id):
         from google.cloud._testing import _NamedTemporaryFile
         from google.cloud.bigquery.job import (