如何检查 BigQuery 中是否存在数据集？

Question

我们可以按如下方式启动 BigQuery 数据集：

dataset_ref = self.client.dataset(dataset_id=self.dataset_id)
dataset = bigquery.Dataset(dataset_ref)

如何检查此数据集是否已经存在？ 当我查看数据集的属性时，它们似乎与存在的集合和不存在的集合重叠。

Answer 1

文档建议使用get_dataset来确定数据集是否存在。

from google.cloud.exceptions import NotFound

dataset_id = "pigs_in_space"

try:
    client.get_dataset(dataset_id)  # Make an API request.
    print("Dataset {} already exists".format(dataset_id))
except NotFound:
    print("Dataset {} is not found".format(dataset_id))

Answer 2

您可以使用 create_dataset function 中的 exists_ok 参数忽略该错误。

# create dataset if not exists

self.client.create_dataset(dataset_id=self.dataset_id, exists_ok=True)

我在 Google 的 bigquery API repo中找到了它。