refactors the generator for clarity and to gen code per design

Author: chalmerlowe

scripts/microgenerator/name_utils.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# 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.
+
+"""A utility module for handling name transformations."""
+
+import re
+from typing import Dict
+
+def to_snake_case(name: str) -> str:
+    """Converts a PascalCase name to snake_case."""
+    return re.sub(r"(?<!^)(?=[A-Z])", "_", name).lower()
+
+def generate_service_names(class_name: str) -> Dict[str, str]:
+    """
+    Generates various name formats for a service based on its client class name.
+
+    Args:
+        class_name: The PascalCase name of the service client class 
+                    (e.g., 'DatasetServiceClient').
+
+    Returns:
+        A dictionary containing different name variations.
+    """
+    snake_case_name = to_snake_case(class_name)
+    module_name = snake_case_name.replace("_client", "")
+    service_name = module_name.replace("_service", "")
+
+    return {
+        "service_name": service_name,
+        "service_module_name": module_name,
+        "service_client_class": class_name,
+        "property_name": snake_case_name,  # Direct use of snake_case_name
+    }

scripts/microgenerator/templates/client.py.j2

@@ -0,0 +1,103 @@
+# TODO: Add a header if needed.
+
+# ======== 🦕 HERE THERE BE DINOSAURS 🦖 =========
+# This content is subject to significant change. Not for review yet.
+# Included as a proof of concept for context or testing ONLY.
+# ================================================
+
+# Imports
+import os
+
+from typing import (
+    Dict,
+    Optional,
+    Sequence,
+    Tuple,
+    Union,
+)
+
+
+{% for imp in service_imports %}
+{{ imp }}
+{% endfor %}
+from google.cloud.bigquery_v2.services.centralized_service import _helpers
+
+{% for imp in type_imports %}
+{{ imp }}
+{% endfor %}
+from google.cloud.bigquery_v2.types import dataset_reference
+
+from google.api_core import client_options as client_options_lib
+from google.api_core import gapic_v1
+from google.api_core import retry as retries
+from google.auth import credentials as auth_credentials
+
+# Create type aliases
+try:
+    OptionalRetry = Union[retries.Retry, gapic_v1.method._MethodDefault, None]
+except AttributeError:  # pragma: NO COVER
+    OptionalRetry = Union[retries.Retry, object, None]  # type: ignore
+
+DatasetIdentifier = Union[str, dataset_reference.DatasetReference]
+
+DEFAULT_RETRY: OptionalRetry = gapic_v1.method.DEFAULT
+DEFAULT_TIMEOUT: Union[float, object] = gapic_v1.method.DEFAULT
+DEFAULT_METADATA: Sequence[Tuple[str, Union[str, bytes]]] = ()
+
+
+class BigQueryClient:
+    def __init__(self, credentials=None, client_options=None):
+        self._clients = {}
+        self._credentials = credentials
+        self._client_options = client_options
+
+    # --- *METHOD SECTION ---
+{% for method in methods %}
+    def {{ method.name }}(
+        self,
+        *,
+        request: Optional["{{ method.service_module_name.replace('_service', '') }}.{{ method.name.replace('_', ' ').title().replace(' ', '') }}Request"] = None,
+        retry: OptionalRetry = DEFAULT_RETRY,
+        timeout: Union[float, object] = DEFAULT_TIMEOUT,
+        metadata: Sequence[Tuple[str, Union[str, bytes]]] = DEFAULT_METADATA,
+    ) -> "{{ method.return_type }}":
+        """
+        TODO: Docstring is purposefully blank. microgenerator will add automatically.
+        """
+
+        return self.{{ method.service_module_name }}_client.{{ method.name }}(
+            request=request,
+            retry=retry,
+            timeout=timeout,
+            metadata=metadata,
+        )
+{% endfor %}
+    
+{#- *ServiceClient Properties Section: methods to get/set service clients -#}
+    # --- *SERVICECLIENT PROPERTIES ---
+{% for service in services %}
+    @property
+    def {{ service.property_name }}(self):
+        if "{{ service.service_name }}" not in self._clients:
+            self._clients["{{ service.service_name }}"] = {{ service.service_module_name }}.{{ service.service_client_class }}(
+                credentials=self._credentials, client_options=self._client_options
+            )
+        return self._clients["{{ service.service_name }}"]
+
+    @{{ service.property_name }}.setter
+    def {{ service.property_name }}(self, value):
+        if not isinstance(value, {{ service.service_module_name }}.{{ service.service_client_class }}):
+            raise TypeError(
+                "Expected an instance of {{ service.service_module_name }}.{{ service.service_client_class }}."
+            )
+        self._clients["{{ service.service_name }}"] = value
+{% endfor %}
+
+{#- Helper Section: methods included from partial template -#}
+    {%- include "partials/_client_helpers.j2" %}
+
+
+# ======== 🦕 HERE THERE WERE DINOSAURS 🦖 =========
+# The above content is subject to significant change. Not for review yet.
+# Included as a proof of concept for context or testing ONLY.
+# ================================================

scripts/microgenerator/templates/partials/_client_helpers.j2

@@ -0,0 +1,49 @@
+    {#
+    This is a partial template file intended to be included in other templates.
+    It contains helper methods for the BigQueryClient class.
+    #}
+    
+    # --- HELPER METHODS ---
+    def _parse_dataset_path(self, dataset_path: str) -> Tuple[Optional[str], str]:
+        """
+        Helper to parse project_id and/or dataset_id from a string identifier.
+    
+        Args:
+            dataset_path: A string in the format 'project_id.dataset_id' or
+                'dataset_id'.
+    
+        Returns:
+            A tuple of (project_id, dataset_id).
+        """
+        if "." in dataset_path:
+            # Use rsplit to handle legacy paths like `google.com:my-project.my_dataset`.
+            project_id, dataset_id = dataset_path.rsplit(".", 1)
+            return project_id, dataset_id
+        return self.project, dataset_path
+    
+    def _parse_dataset_id_to_dict(self, dataset_id: "DatasetIdentifier") -> dict:
+        """
+        Helper to create a dictionary from a project_id and dataset_id to pass
+        internally between helper functions.
+    
+        Args:
+            dataset_id: A string or DatasetReference.
+    
+        Returns:
+            A dict of {"project_id": project_id, "dataset_id": dataset_id_str }.
+        """
+        if isinstance(dataset_id, str):
+            project_id, dataset_id_str = self._parse_dataset_path(dataset_id)
+            return {"project_id": project_id, "dataset_id": dataset_id_str}
+        elif isinstance(dataset_id, dataset_reference.DatasetReference):
+            return {
+                "project_id": dataset_id.project_id,
+                "dataset_id": dataset_id.dataset_id,
+            }
+        else:
+            raise TypeError(f"Invalid type for dataset_id: {type(dataset_id)}")
+    
+    def _parse_project_id_to_dict(self, project_id: Optional[str] = None) -> dict:
+        """Helper to create a request dictionary from a project_id."""
+        final_project_id = project_id or self.project
+        return {"project_id": final_project_id}