Google Cloud Dataproc Metastore API client library

import os
from google.cloud.metastore_v1.services.dataproc_metastore import DataprocMetastoreClient
from google.cloud.metastore_v1.types import ListServicesRequest

def list_metastore_services(project_id: str, location: str) -> None:
    """Lists Dataproc Metastore services in a given project and location.

    Args:
        project_id: Your Google Cloud project ID.
        location: The Google Cloud location (e.g., 'us-central1').
    """
    # Instantiates a client
    client = DataprocMetastoreClient()

    # The resource name of the location where the services are located.
    # Example: "projects/my-project/locations/us-central1"
    parent = f"projects/{project_id}/locations/{location}"

    # Construct the request
    request = ListServicesRequest(parent=parent)

    # Call the API
    try:
        page_result = client.list_services(request=request)

        print(f"Dataproc Metastore services in {parent}:")
        found_services = False
        for service in page_result:
            print(f"- {service.name} (State: {service.state.name})")
            found_services = True
        if not found_services:
            print("  No Dataproc Metastore services found.")
    except Exception as e:
        print(f"Error listing services: {e}")
        print("Ensure the API is enabled, credentials are set, and the location is valid.")

# To run this quickstart:
# 1. Ensure `gcloud auth application-default login` has been run or `GOOGLE_APPLICATION_CREDENTIALS` is set.
# 2. Set the `GOOGLE_CLOUD_PROJECT` environment variable to your project ID.
# 3. Set the `GOOGLE_CLOUD_LOCATION` environment variable to your desired location (e.g., "us-central1").
# Example usage:
# GOOGLE_CLOUD_PROJECT='your-project-id' GOOGLE_CLOUD_LOCATION='us-central1' python your_script_name.py
if __name__ == "__main__":
    project_id = os.environ.get("GOOGLE_CLOUD_PROJECT", "")
    location = os.environ.get("GOOGLE_CLOUD_LOCATION", "")

    if not project_id:
        print("Please set the GOOGLE_CLOUD_PROJECT environment variable.")
    elif not location:
        print("Please set the GOOGLE_CLOUD_LOCATION environment variable.")
    else:
        list_metastore_services(project_id, location)