def get_dataset_locations()

in helpers/bq-list-locations/main.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def get_dataset_locations(project_id):
    """Fetches all unique BigQuery dataset locations for a project."""

    client = bigquery.Client(project=project_id)

    bq_locations = set()

    for dataset_list_item in client.list_datasets():
        dataset = client.get_dataset(dataset_list_item.reference)  # Get full dataset object
        bq_locations.add(dataset.location)  # Access the location from the full object

    return bq_locations