Truncate table in Google Cloud BigQuery

bigquery_truncate_table.py

from google.cloud import bigquery
from google.cloud.exceptions import NotFound
from google.api_core.exceptions import Conflict

from shared.loaders.bigquery_constants import BIGQUERY_PROJECT_NAME

print(f'Contando registros en {BIGQUERY_PROJECT_NAME}.{bigquery_dataset_name}.{bigquery_table_name}')

query_count = f'SELECT COUNT(*) as record_count FROM  {BIGQUERY_PROJECT_NAME}.{bigquery_dataset_name}.{bigquery_table_name}'
query_count_job = bigquery_client.query(query_count) 
dataframe_count = query_count_job.result().to_dataframe()
    
record_count = dataframe_count.iloc[0]['record_count']

if record_count > 0:
  print(f'Truncating rows {BIGQUERY_PROJECT_NAME}.{bigquery_dataset_name}.{bigquery_table_name}')

  delete_rows_job = bigquery_client.query(f'TRUNCATE TABLE {BIGQUERY_PROJECT_NAME}.{bigquery_dataset_name}.{bigquery_table_name}')
  delete_rows_job.result()