in common/py_libs/bq_helper.py [0:0]
def create_dataset(bq_client: bigquery.Client,
dataset_name: str,
location: str,
suppress_success_logging: bool = False) -> None:
"""Creates a BigQuery dataset."""
dataset_ref = bigquery.Dataset(dataset_name)
dataset_ref.location = location
try:
bq_client.create_dataset(dataset_ref, timeout=30)
if not suppress_success_logging:
logging.info("✅ Dataset %s has been created in %s.",
dataset_name,
location)
except Conflict:
logging.warning("⚠️ Dataset %s already exists in %s. Skipping it.",
dataset_name,
location)
except Exception:
logging.error("⛔️ Failed to create dataset %s in %s.", dataset_name,
location,
exc_info=True)
raise