Add logs in the __init__ file

Author: sycai

bigframes/bigquery/__init__.py

@@ -16,6 +16,8 @@
 such as array functions:
 https://cloud.google.com/bigquery/docs/reference/standard-sql/array_functions. """
 
+import sys
+
 from bigframes.bigquery._operations.approx_agg import approx_top_count
 from bigframes.bigquery._operations.array import (
     array_agg,
@@ -52,43 +54,43 @@
 from bigframes.bigquery._operations.search import create_vector_index, vector_search
 from bigframes.bigquery._operations.sql import sql_scalar
 from bigframes.bigquery._operations.struct import struct
+from bigframes.core import log_adapter
 
-__all__ = [
-    # approximate aggregate ops
-    "approx_top_count",
-    # array ops
-    "array_agg",
-    "array_length",
-    "array_to_string",
-    # datetime ops
-    "unix_micros",
-    "unix_millis",
-    "unix_seconds",
-    # geo ops
-    "st_area",
-    "st_buffer",
-    "st_centroid",
-    "st_convexhull",
-    "st_difference",
-    "st_distance",
-    "st_intersection",
-    "st_isclosed",
-    "st_length",
-    # json ops
-    "json_extract",
-    "json_extract_array",
-    "json_extract_string_array",
-    "json_query",
-    "json_query_array",
-    "json_set",
-    "json_value",
-    "json_value_array",
-    "parse_json",
-    # search ops
-    "create_vector_index",
-    "vector_search",
-    # sql ops
-    "sql_scalar",
-    # struct ops
-    "struct",
+_functions = [
+    approx_top_count,
+    array_agg,
+    array_length,
+    array_to_string,
+    unix_micros,
+    unix_millis,
+    unix_seconds,
+    st_area,
+    st_buffer,
+    st_centroid,
+    st_convexhull,
+    st_difference,
+    st_distance,
+    st_intersection,
+    st_isclosed,
+    st_length,
+    json_extract,
+    json_extract_array,
+    json_extract_string_array,
+    json_query,
+    json_query_array,
+    json_set,
+    json_value,
+    json_value_array,
+    parse_json,
+    create_vector_index,
+    vector_search,
+    sql_scalar,
+    struct,
 ]
+
+__all__ = [f.__name__ for f in _functions]
+
+_module = sys.modules[__name__]
+for f in _functions:
+    _decorated_object = log_adapter.method_logger(f, custom_base_name="bigquery")
+    setattr(_module, f.__name__, _decorated_object)

bigframes/bigquery/_operations/approx_agg.py

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

bigframes/bigquery/_operations/array.py

@@ -24,7 +24,6 @@
 
 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
@@ -34,7 +33,6 @@
     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.
 
@@ -70,7 +68,6 @@ 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:
@@ -124,7 +121,6 @@ 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,10 +14,8 @@
 
 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
 
@@ -45,7 +43,6 @@ 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
 
@@ -73,7 +70,6 @@ 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,7 +19,6 @@
 import shapely  # type: ignore
 
 from bigframes import operations as ops
-from bigframes.core import log_adapter
 import bigframes.geopandas
 import bigframes.series
 
@@ -29,7 +28,6 @@
 """
 
 
-@log_adapter.method_logger
 def st_area(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -105,7 +103,6 @@ def st_area(
     return series
 
 
-@log_adapter.method_logger
 def st_buffer(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     buffer_radius: float,
@@ -175,7 +172,6 @@ def st_buffer(
     return series
 
 
-@log_adapter.method_logger
 def st_centroid(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -233,7 +229,6 @@ def st_centroid(
     return series
 
 
-@log_adapter.method_logger
 def st_convexhull(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
 ) -> bigframes.series.Series:
@@ -289,7 +284,6 @@ def st_convexhull(
     return series
 
 
-@log_adapter.method_logger
 def st_difference(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     other: Union[
@@ -393,7 +387,6 @@ 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[
@@ -471,7 +464,6 @@ def st_distance(
     )
 
 
-@log_adapter.method_logger
 def st_intersection(
     series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
     other: Union[
@@ -571,7 +563,6 @@ 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:
@@ -632,7 +623,6 @@ 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,7 +24,6 @@
 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
@@ -34,7 +33,6 @@
 from . import array
 
 
-@log_adapter.method_logger
 @utils.preview(name="The JSON-related API `json_set`")
 def json_set(
     input: series.Series,
@@ -87,7 +85,6 @@ def json_set(
     return result
 
 
-@log_adapter.method_logger
 def json_extract(
     input: series.Series,
     json_path: str,
@@ -128,7 +125,6 @@ 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 = "$",
@@ -188,7 +184,6 @@ 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 = "$",
@@ -265,7 +260,6 @@ def json_extract_string_array(
     return array_series
 
 
-@log_adapter.method_logger
 def json_query(
     input: series.Series,
     json_path: str,
@@ -297,7 +291,6 @@ 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 = "$",
@@ -348,7 +341,6 @@ 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 = "$",
@@ -383,7 +375,6 @@ 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 = "$",
@@ -439,7 +430,6 @@ 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,6 @@
 
 import google.cloud.bigquery as bigquery
 
-from bigframes.core import log_adapter
 import bigframes.ml.utils as utils
 
 if typing.TYPE_CHECKING:
@@ -34,7 +33,6 @@
 """
 
 
-@log_adapter.method_logger
 def create_vector_index(
     table_id: str,
     column_name: str,
@@ -90,7 +88,6 @@ 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,12 @@
 
 import google.cloud.bigquery
 
-from bigframes.core import log_adapter
 import bigframes.core.compile.sqlglot.sqlglot_ir as sqlglot_ir
 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,15 +22,13 @@
 
 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