def main(*, source_project_id, source_dataset_id, destination_project_id,
destination_dataset_id):
"""Copies all views from the source_project_id.source_dataset_id to the
destination_project_id.destination_dataset_id."""
# Construct a BigQuery client with the source_project_id
source_client = BigQueryClientImpl(project_id=source_project_id)
# Construct a BigQuery client with the destination_project_id
destination_client = BigQueryClientImpl(project_id=destination_project_id)
destination_dataset = bigquery.DatasetReference(destination_project_id,
destination_dataset_id)
tables_in_source_dataset = source_client.list_tables(source_dataset_id)
for table_ref in tables_in_source_dataset:
table = source_client.get_table(
source_client.dataset_ref_for_id(table_ref.dataset_id),
table_ref.table_id)
# Only copy this view if there is a view_query to replicate and the view doesn't already exist in the
# destination dataset
if table.view_query and not destination_client.table_exists(
destination_dataset, table_id=table.table_id):
# Retrieve all of the information about the view
source_client.copy_view(
view=BigQueryView(dataset_id=table_ref.dataset_id,
view_id=table.table_id,
view_query_template=table.view_query),
destination_client=destination_client,
destination_dataset_ref=destination_dataset)
def copy_bq_views(
source_project_id: str,
source_dataset_id: str,
destination_project_id: str,
destination_dataset_id: str,
) -> None:
"""Copies all views from the source_project_id.source_dataset_id to the
destination_project_id.destination_dataset_id."""
# Construct a BigQuery client with the source_project_id
source_client = BigQueryClientImpl(project_id=source_project_id)
# Construct a BigQuery client with the destination_project_id
destination_client = BigQueryClientImpl(project_id=destination_project_id)
destination_dataset = bigquery.DatasetReference(destination_project_id,
destination_dataset_id)
tables_in_source_dataset = source_client.list_tables(source_dataset_id)
for table_ref in tables_in_source_dataset:
table = source_client.get_table(
source_client.dataset_ref_for_id(table_ref.dataset_id),
table_ref.table_id)
view_query = table.view_query
# Only copy this view if there is a view_query to replicate and the view doesn't already exist in the
# destination dataset
if view_query and not destination_client.table_exists(
destination_dataset, table_id=table.table_id):
# Remove any references to the source_project_id from the view_query
updated_view_query = view_query.replace(source_project_id,
"{project_id}")
# Retrieve all of the information about the view
source_client.copy_view(
view=BigQueryView(
project_id=destination_project_id,
dataset_id=destination_dataset_id,
view_id=table.table_id,
description=table.description,
view_query_template=updated_view_query,
),
destination_client=destination_client,
destination_dataset_ref=destination_dataset,
)