Compute BigQuery Project Size in Bytes

bq_size.py

def query_size(client, dataset_id: str) -> float:
    query_job = client.query(f'SELECT sum(size_bytes) as size FROM {dataset_id}.__TABLES__')
    results = query_job.result()
    for row in results:
        print(f'adding to size: {row.size}')
        if row.size is None:
          return 0
        return float(row.size)


def sum_up_size():
  total_size = 0
  client = bigquery.Client()

  datasets = list(client.list_datasets())
  project = client.project

  if datasets:
      print("Datasets in project {}:".format(project))
      for dataset in datasets:
          print("\t{}".format(dataset.dataset_id))
          total_size = total_size + query_size(client, dataset.dataset_id)
          print(f'current size: {total_size}')
  else:
      print("{} project does not contain any datasets.".format(project))

sum_up_size()