使用API在BigQuery中创建数据集

pip install --upgrade google-api-python-client

from googleapiclient.discovery import build
from google.oauth2 import service_account

# 使用API密钥或OAuth 2.0凭据进行身份验证
credentials = service_account.Credentials.from_service_account_file(
    'path/to/your/service-account-file.json'
)

# 创建BigQuery API客户端
bigquery_service = build('bigquery', 'v2', credentials=credentials)

# 定义数据集ID和项目ID
project_id = 'your-project-id'
dataset_id = 'your_dataset_id'

# 创建数据集资源
dataset_resource = {
    'datasetReference': {
        'projectId': project_id,
        'datasetId': dataset_id
    }
}

# 发送请求创建数据系
try:
    dataset = bigquery_service.datasets().insert(projectId=project_id, body=dataset_resource).execute()
    print(f"Dataset {dataset['datasetReference']['datasetId']} created.")
except Exception as e:
    print(f"Error creating dataset: {e}")