chore: add logging to bigframes.bigquery functions

Author: sycai

bigframes/bigquery/_operations/approx_agg.py

@@ -14,6 +14,7 @@
 
 from __future__ import annotations
 
+from bigframes.core import log_adapter
 import bigframes.operations.aggregations as agg_ops
 import bigframes.series as series
 
@@ -23,6 +24,7 @@
 """
 
 
+@log_adapter.method_logger
 def approx_top_count(
     series: series.Series,
     number: int,

bigframes/bigquery/_operations/array.py

@@ -24,6 +24,7 @@
 
 import bigframes_vendored.constants as constants
 
+from bigframes.core import log_adapter
 import bigframes.core.groupby as groupby
 import bigframes.operations as ops
 import bigframes.operations.aggregations as agg_ops
@@ -33,6 +34,7 @@
     import bigframes.dataframe as dataframe
 
 
+@log_adapter.method_logger
 def array_length(series: series.Series) -> series.Series:
     """Compute the length of each array element in the Series.
 
@@ -68,6 +70,7 @@ def array_length(series: series.Series) -> series.Series:
     return series._apply_unary_op(ops.len_op)
 
 
+@log_adapter.method_logger
 def array_agg(
     obj: groupby.SeriesGroupBy | groupby.DataFrameGroupBy,
 ) -> series.Series | dataframe.DataFrame:
@@ -121,6 +124,7 @@ def array_agg(
         )
 
 
+@log_adapter.method_logger
 def array_to_string(series: series.Series, delimiter: str) -> series.Series:
     """Converts array elements within a Series into delimited strings.
 

bigframes/bigquery/_operations/datetime.py

@@ -14,8 +14,10 @@
 
 from bigframes import operations as ops
 from bigframes import series
+from bigframes.core import log_adapter
 
 
+@log_adapter.method_logger
 def unix_seconds(input: series.Series) -> series.Series:
     """Converts a timestmap series to unix epoch seconds
 
@@ -43,6 +45,7 @@ def unix_seconds(input: series.Series) -> series.Series:
     return input._apply_unary_op(ops.UnixSeconds())
 
 
+@log_adapter.method_logger
 def unix_millis(input: series.Series) -> series.Series:
     """Converts a timestmap series to unix epoch milliseconds
 
@@ -70,6 +73,7 @@ def unix_millis(input: series.Series) -> series.Series:
     return input._apply_unary_op(ops.UnixMillis())
 
 
+@log_adapter.method_logger
 def unix_micros(input: series.Series) -> series.Series:
     """Converts a timestmap series to unix epoch microseconds
 

bigframes/bigquery/_operations/geo.py

@@ -19,6 +19,7 @@
 import shapely  # type: ignore
 
 from bigframes import operations as ops
+from bigframes.core import log_adapter
 import bigframes.geopandas
 import bigframes.series
 
@@ -28,6 +29,7 @@
 """
 
 
+@log_adapter.method_logger
 def st_area(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -103,6 +105,7 @@ def st_area(
     return series
 
 
+@log_adapter.method_logger
 def st_buffer(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     buffer_radius: float,
@@ -172,6 +175,7 @@ def st_buffer(
     return series
 
 
+@log_adapter.method_logger
 def st_centroid(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -229,6 +233,7 @@ def st_centroid(
     return series
 
 
+@log_adapter.method_logger
 def st_convexhull(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -284,6 +289,7 @@ def st_convexhull(
     return series
 
 
+@log_adapter.method_logger
 def st_difference(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     other: Union[
@@ -387,6 +393,7 @@ def st_difference(
     return series._apply_binary_op(other, ops.geo_st_difference_op)
 
 
+@log_adapter.method_logger
 def st_distance(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     other: Union[
@@ -464,6 +471,7 @@ def st_distance(
     )
 
 
+@log_adapter.method_logger
 def st_intersection(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     other: Union[
@@ -563,6 +571,7 @@ def st_intersection(
     return series._apply_binary_op(other, ops.geo_st_intersection_op)
 
 
+@log_adapter.method_logger
 def st_isclosed(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -623,6 +632,7 @@ def st_isclosed(
     return series
 
 
+@log_adapter.method_logger
 def st_length(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     *,

bigframes/bigquery/_operations/json.py

@@ -24,6 +24,7 @@
 from typing import Any, cast, Optional, Sequence, Tuple, Union
 import warnings
 
+from bigframes.core import log_adapter
 import bigframes.core.utils as utils
 import bigframes.dtypes
 import bigframes.exceptions as bfe
@@ -33,6 +34,7 @@
 from . import array
 
 
+@log_adapter.method_logger
 @utils.preview(name="The JSON-related API `json_set`")
 def json_set(
     input: series.Series,
@@ -85,6 +87,7 @@ def json_set(
     return result
 
 
+@log_adapter.method_logger
 def json_extract(
     input: series.Series,
     json_path: str,
@@ -125,6 +128,7 @@ def json_extract(
     return input._apply_unary_op(ops.JSONExtract(json_path=json_path))
 
 
+@log_adapter.method_logger
 def json_extract_array(
     input: series.Series,
     json_path: str = "$",
@@ -184,6 +188,7 @@ def json_extract_array(
     return input._apply_unary_op(ops.JSONExtractArray(json_path=json_path))
 
 
+@log_adapter.method_logger
 def json_extract_string_array(
     input: series.Series,
     json_path: str = "$",
@@ -260,6 +265,7 @@ def json_extract_string_array(
     return array_series
 
 
+@log_adapter.method_logger
 def json_query(
     input: series.Series,
     json_path: str,
@@ -291,6 +297,7 @@ def json_query(
     return input._apply_unary_op(ops.JSONQuery(json_path=json_path))
 
 
+@log_adapter.method_logger
 def json_query_array(
     input: series.Series,
     json_path: str = "$",
@@ -341,6 +348,7 @@ def json_query_array(
     return input._apply_unary_op(ops.JSONQueryArray(json_path=json_path))
 
 
+@log_adapter.method_logger
 def json_value(
     input: series.Series,
     json_path: str = "$",
@@ -375,6 +383,7 @@ def json_value(
     return input._apply_unary_op(ops.JSONValue(json_path=json_path))
 
 
+@log_adapter.method_logger
 def json_value_array(
     input: series.Series,
     json_path: str = "$",
@@ -430,6 +439,7 @@ def json_value_array(
     return input._apply_unary_op(ops.JSONValueArray(json_path=json_path))
 
 
+@log_adapter.method_logger
 @utils.preview(name="The JSON-related API `parse_json`")
 def parse_json(
     input: series.Series,

bigframes/bigquery/_operations/search.py

@@ -20,7 +20,7 @@
 
 import google.cloud.bigquery as bigquery
 
-import bigframes.core.sql
+from bigframes.core import log_adapter
 import bigframes.ml.utils as utils
 
 if typing.TYPE_CHECKING:
@@ -34,6 +34,7 @@
 """
 
 
+@log_adapter.method_logger
 def create_vector_index(
     table_id: str,
     column_name: str,
@@ -89,6 +90,7 @@ def create_vector_index(
     read_gbq_query(sql)
 
 
+@log_adapter.method_logger
 def vector_search(
     base_table: str,
     column_to_search: str,

bigframes/bigquery/_operations/sql.py

@@ -20,14 +20,14 @@
 
 import google.cloud.bigquery
 
+from bigframes.core import log_adapter
 import bigframes.core.compile.sqlglot.sqlglot_ir as sqlglot_ir
-import bigframes.core.sql
-import bigframes.dataframe
 import bigframes.dtypes
 import bigframes.operations
 import bigframes.series
 
 
+@log_adapter.method_logger
 def sql_scalar(
     sql_template: str,
     columns: Sequence[bigframes.series.Series],

bigframes/bigquery/_operations/struct.py

@@ -22,13 +22,15 @@
 
 import typing
 
+from bigframes.core import log_adapter
 import bigframes.operations as ops
 import bigframes.series as series
 
 if typing.TYPE_CHECKING:
     import bigframes.dataframe as dataframe
 
 
+@log_adapter.method_logger
 def struct(value: dataframe.DataFrame) -> series.Series:
     """Takes a DataFrame and converts it into a Series of structs with each
     struct entry corresponding to a DataFrame row and each struct field