From 58968650090975a96510c57342a21bc99ced7a7d Mon Sep 17 00:00:00 2001 From: Adler Santos Date: Tue, 5 Apr 2022 12:17:40 -0400 Subject: [PATCH] fix: Fix generating BQ views for IDC dataset (#324) --- .../idc/pipelines/_images/generate_bq_views/script.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/datasets/idc/pipelines/_images/generate_bq_views/script.py b/datasets/idc/pipelines/_images/generate_bq_views/script.py index 1fc8d085a..6d1ad15df 100644 --- a/datasets/idc/pipelines/_images/generate_bq_views/script.py +++ b/datasets/idc/pipelines/_images/generate_bq_views/script.py @@ -51,14 +51,17 @@ def main( source_view = client.get_table( f"{source_project}.{dataset}.{table.table_id}" ) - create_or_update_view(client, source_view, target_project) + create_or_update_view(client, source_view, source_project, target_project) source_views.append(table.table_id) sync_views(client, dataset, source_views, target_project) def create_or_update_view( - client: bigquery.Client, source_view: bigquery.Table, target_project: str + client: bigquery.Client, + source_view: bigquery.Table, + source_project: str, + target_project: str, ) -> None: try: target_view = client.get_table( @@ -71,7 +74,7 @@ def create_or_update_view( f"{target_project}.{source_view.dataset_id}.{source_view.table_id}" ) _view.description = source_view.description - _view.view_query = source_view.view_query + _view.view_query = source_view.view_query.replace(source_project, target_project) # Create the view if it doesn't exist. Otherwise, update it. if not target_view: