chore: restores accidentally deleted subdirectory

Author: chalmerlowe

packages/bigframes/bigframes/__init__.py

@@ -0,0 +1,63 @@
+# Copyright 2023 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""BigQuery DataFrames provides a DataFrame API scaled by the BigQuery engine."""
+
+import warnings
+
+# Suppress Python version support warnings from google-cloud libraries.
+# These are particularly noisy in Colab which still uses Python 3.10.
+warnings.filterwarnings(
+    "ignore",
+    category=FutureWarning,
+    message=".*Google will stop supporting.*Python.*",
+)
+
+from bigframes._config import option_context, options  # noqa: E402
+from bigframes._config.bigquery_options import BigQueryOptions  # noqa: E402
+from bigframes.core.global_session import (  # noqa: E402
+    close_session,
+    get_global_session,
+)
+import bigframes.enums as enums  # noqa: E402
+import bigframes.exceptions as exceptions  # noqa: E402
+from bigframes.session import connect, Session  # noqa: E402
+from bigframes.version import __version__  # noqa: E402
+
+_MAGIC_NAMES = ["bqsql"]
+
+
+def load_ipython_extension(ipython):
+    """Called by IPython when this module is loaded as an IPython extension."""
+    # Requires IPython to be installed for import to succeed
+    from bigframes._magics import _cell_magic
+
+    for magic_name in _MAGIC_NAMES:
+        ipython.register_magic_function(
+            _cell_magic, magic_kind="cell", magic_name=magic_name
+        )
+
+
+__all__ = [
+    "options",
+    "BigQueryOptions",
+    "get_global_session",
+    "close_session",
+    "enums",
+    "exceptions",
+    "connect",
+    "Session",
+    "__version__",
+    "option_context",
+]

packages/bigframes/bigframes/_config/__init__.py

@@ -0,0 +1,40 @@
+# Copyright 2023 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Configuration for BigQuery DataFrames. Do not depend on other parts of BigQuery
+DataFrames from this package.
+"""
+
+from bigframes._config.bigquery_options import BigQueryOptions
+from bigframes._config.compute_options import ComputeOptions
+from bigframes._config.display_options import DisplayOptions
+from bigframes._config.experiment_options import ExperimentOptions
+from bigframes._config.global_options import option_context, Options
+import bigframes._config.global_options as global_options
+from bigframes._config.sampling_options import SamplingOptions
+
+options = global_options.options
+"""Global options for the default session."""
+
+__all__ = (
+    "Options",
+    "options",
+    "option_context",
+    "BigQueryOptions",
+    "ComputeOptions",
+    "DisplayOptions",
+    "ExperimentOptions",
+    "SamplingOptions",
+)

packages/bigframes/bigframes/_config/auth.py

@@ -0,0 +1,57 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import threading
+from typing import Optional
+
+import google.auth.credentials
+import google.auth.transport.requests
+import pydata_google_auth
+
+_SCOPES = ["https://www.googleapis.com/auth/cloud-platform"]
+
+# Put the lock here rather than in BigQueryOptions so that BigQueryOptions
+# remains deepcopy-able.
+_AUTH_LOCK = threading.Lock()
+_cached_credentials: Optional[google.auth.credentials.Credentials] = None
+_cached_project_default: Optional[str] = None
+
+
+def get_default_credentials_with_project() -> (
+    tuple[google.auth.credentials.Credentials, Optional[str]]
+):
+    global _AUTH_LOCK, _cached_credentials, _cached_project_default
+
+    with _AUTH_LOCK:
+        if _cached_credentials is not None:
+            return _cached_credentials, _cached_project_default
+
+        _cached_credentials, _cached_project_default = pydata_google_auth.default(
+            scopes=_SCOPES, use_local_webserver=False
+        )
+
+        # Ensure an access token is available.
+        _cached_credentials.refresh(google.auth.transport.requests.Request())
+
+    return _cached_credentials, _cached_project_default
+
+
+def reset_default_credentials_and_project():
+    global _AUTH_LOCK, _cached_credentials, _cached_project_default
+
+    with _AUTH_LOCK:
+        _cached_credentials = None
+        _cached_project_default = None