Merge branch 'main' into authorized_views

Author: daniel-sanche

.github/.OwlBot.lock.yaml

@@ -13,5 +13,5 @@
 # limitations under the License.
 docker:
   image: gcr.io/cloud-devrel-public-resources/owlbot-python:latest
-  digest: sha256:a1c5112b81d645f5bbc4d4bbc99d7dcb5089a52216c0e3fb1203a0eeabadd7d5
-# created: 2025-01-02T23:09:36.975468657Z
+  digest: sha256:8ff1efe878e18bd82a0fb7b70bb86f77e7ab6901fed394440b6135db0ba8d84a
+# created: 2025-01-09T12:01:16.422459506Z

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-    ".": "2.28.0"
+    ".": "2.28.1"
 }

CHANGELOG.md

@@ -4,6 +4,13 @@
 
 [1]: https://pypi.org/project/google-cloud-bigtable/#history
 
+## [2.28.1](https://github.com/googleapis/python-bigtable/compare/v2.28.0...v2.28.1) (2025-01-17)
+
+
+### Bug Fixes
+
+* Allow empty headers for btql routing ([#1072](https://github.com/googleapis/python-bigtable/issues/1072)) ([e7ecfeb](https://github.com/googleapis/python-bigtable/commit/e7ecfeb8984a45c880d9483305964fff347eb4b8))
+
 ## [2.28.0](https://github.com/googleapis/python-bigtable/compare/v2.27.0...v2.28.0) (2025-01-08)
 
 

google/cloud/bigtable/data/_async/client.py

@@ -645,6 +645,8 @@ async def execute_query(
                 will be chained with a RetryExceptionGroup containing GoogleAPIError exceptions
                 from any retries that failed
             google.api_core.exceptions.GoogleAPIError: raised if the request encounters an unrecoverable error
+            google.cloud.bigtable.data.exceptions.ParameterTypeInferenceFailed: Raised if
+                a parameter is passed without an explicit type, and the type cannot be infered
         """
         warnings.warn(
             "ExecuteQuery is in preview and may change in the future.",

google/cloud/bigtable/data/_sync_autogen/client.py

@@ -489,6 +489,8 @@ def execute_query(
                 will be chained with a RetryExceptionGroup containing GoogleAPIError exceptions
                 from any retries that failed
             google.api_core.exceptions.GoogleAPIError: raised if the request encounters an unrecoverable error
+            google.cloud.bigtable.data.exceptions.ParameterTypeInferenceFailed: Raised if
+                a parameter is passed without an explicit type, and the type cannot be infered
         """
         warnings.warn(
             "ExecuteQuery is in preview and may change in the future.",

google/cloud/bigtable/data/execute_query/_async/execute_query_iterator.py

@@ -22,7 +22,6 @@
     Tuple,
     TYPE_CHECKING,
 )
-
 from google.api_core import retry as retries
 
 from google.cloud.bigtable.data.execute_query._byte_cursor import _ByteCursor
@@ -116,7 +115,6 @@ def __init__(
             exception_factory=_retry_exception_factory,
         )
         self._req_metadata = req_metadata
-
         try:
             self._register_instance_task = CrossSync.create_task(
                 self._client._register_instance,

google/cloud/bigtable/data/execute_query/_parameters_formatting.py

@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Dict, Optional
 import datetime
+from typing import Any, Dict, Optional
+
 from google.api_core.datetime_helpers import DatetimeWithNanoseconds
+
 from google.cloud.bigtable.data.exceptions import ParameterTypeInferenceFailed
-from google.cloud.bigtable.data.execute_query.values import ExecuteQueryValueType
 from google.cloud.bigtable.data.execute_query.metadata import SqlType
+from google.cloud.bigtable.data.execute_query.values import ExecuteQueryValueType
 
 
 def _format_execute_query_params(
@@ -48,7 +50,6 @@ def _format_execute_query_params(
     parameter_types = parameter_types or {}
 
     result_values = {}
-
     for key, value in params.items():
         user_provided_type = parameter_types.get(key)
         try:
@@ -109,6 +110,16 @@ def _detect_type(value: ExecuteQueryValueType) -> SqlType.Type:
             "Cannot infer type of None, please provide the type manually."
         )
 
+    if isinstance(value, list):
+        raise ParameterTypeInferenceFailed(
+            "Cannot infer type of ARRAY parameters, please provide the type manually."
+        )
+
+    if isinstance(value, float):
+        raise ParameterTypeInferenceFailed(
+            "Cannot infer type of float, must specify either FLOAT32 or FLOAT64 type manually."
+        )
+
     for field_type, type_dict in _TYPES_TO_TYPE_DICTS:
         if isinstance(value, field_type):
             return type_dict

google/cloud/bigtable/data/execute_query/_query_result_parsing_utils.py

@@ -22,6 +22,7 @@
     SqlType.Bytes: "bytes_value",
     SqlType.String: "string_value",
     SqlType.Int64: "int_value",
+    SqlType.Float32: "float_value",
     SqlType.Float64: "float_value",
     SqlType.Bool: "bool_value",
     SqlType.Timestamp: "timestamp_value",

google/cloud/bigtable/data/execute_query/metadata.py

@@ -21,23 +21,16 @@
 """
 
 from collections import defaultdict
-from typing import (
-    Optional,
-    List,
-    Dict,
-    Set,
-    Type,
-    Union,
-    Tuple,
-    Any,
-)
+import datetime
+from typing import Any, Dict, List, Optional, Set, Tuple, Type, Union
+
+from google.api_core.datetime_helpers import DatetimeWithNanoseconds
+from google.protobuf import timestamp_pb2  # type: ignore
+from google.type import date_pb2  # type: ignore
+
 from google.cloud.bigtable.data.execute_query.values import _NamedList
 from google.cloud.bigtable_v2 import ResultSetMetadata
 from google.cloud.bigtable_v2 import Type as PBType
-from google.type import date_pb2  # type: ignore
-from google.protobuf import timestamp_pb2  # type: ignore
-from google.api_core.datetime_helpers import DatetimeWithNanoseconds
-import datetime
 
 
 class SqlType:
@@ -127,6 +120,8 @@ class Array(Type):
         def __init__(self, element_type: "SqlType.Type"):
             if isinstance(element_type, SqlType.Array):
                 raise ValueError("Arrays of arrays are not supported.")
+            if isinstance(element_type, SqlType.Map):
+                raise ValueError("Arrays of Maps are not supported.")
             self._element_type = element_type
 
         @property
@@ -140,10 +135,21 @@ def from_pb_type(cls, type_pb: Optional[PBType] = None) -> "SqlType.Array":
             return cls(_pb_type_to_metadata_type(type_pb.array_type.element_type))
 
         def _to_value_pb_dict(self, value: Any):
-            raise NotImplementedError("Array is not supported as a query parameter")
+            if value is None:
+                return {}
+
+            return {
+                "array_value": {
+                    "values": [
+                        self.element_type._to_value_pb_dict(entry) for entry in value
+                    ]
+                }
+            }
 
         def _to_type_pb_dict(self) -> Dict[str, Any]:
-            raise NotImplementedError("Array is not supported as a query parameter")
+            return {
+                "array_type": {"element_type": self.element_type._to_type_pb_dict()}
+            }
 
         def __eq__(self, other):
             return super().__eq__(other) and self.element_type == other.element_type
@@ -222,6 +228,13 @@ class Float64(Type):
         value_pb_dict_field_name = "float_value"
         type_field_name = "float64_type"
 
+    class Float32(Type):
+        """Float32 SQL type."""
+
+        expected_type = float
+        value_pb_dict_field_name = "float_value"
+        type_field_name = "float32_type"
+
     class Bool(Type):
         """Bool SQL type."""
 
@@ -376,6 +389,7 @@ def _pb_metadata_to_metadata_types(
     "bytes_type": SqlType.Bytes,
     "string_type": SqlType.String,
     "int64_type": SqlType.Int64,
+    "float32_type": SqlType.Float32,
     "float64_type": SqlType.Float64,
     "bool_type": SqlType.Bool,
     "timestamp_type": SqlType.Timestamp,

google/cloud/bigtable/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "2.28.0"  # {x-release-please-version}
+__version__ = "2.28.1"  # {x-release-please-version}