如何通过Python API查询Google BigQuery中表的分区类型

我调用get_table方法获取表描述时返回了None，现有代码如下：

def partition(dataset1, dataset2):
    try:
        client.get_dataset(dataset2)
        print("Dataset {} already exists".format(dataset2))
    except NotFound:
         print("Dataset {} not found".format(dataset2))
         createDataset(dataset2)
    table = client.get_table(dataset1) # 获取源数据集
    partition_column = table.time_partitioning.field # 获取分区列名

我想知道如何查询该表所使用的分区类型，相关API中提到了Table.partitioning_type属性。

解决方案 #

1. 排查get_table返回None的问题

• 确认dataset1格式正确，需为项目ID.数据集ID.表ID的完整标识；若已设置默认项目和数据集，也要确保目标表真实存在。
• 建议添加异常捕获，明确表不存在的场景：

2. 查询分区类型的代码实现

修改代码，通过partitioning_type属性获取分区类型，同时兼容不同分区类型的详细信息获取：

def partition(dataset1, dataset2):
    try:
        client.get_dataset(dataset2)
        print("Dataset {} already exists".format(dataset2))
    except NotFound:
         print("Dataset {} not found".format(dataset2))
         createDataset(dataset2)
    try:
        table = client.get_table(dataset1) # 获取源表
        if table:
            # 获取分区类型
            partition_type = table.partitioning_type
            print(f"表的分区类型：{partition_type}")
            
            # 根据分区类型获取对应配置
            if partition_type == "TIME":
                partition_column = table.time_partitioning.field
                print(f"时间分区字段：{partition_column}")
            elif partition_type == "RANGE":
                range_info = table.range_partitioning
                print(f"范围分区字段：{range_info.field}，分区范围规则：{range_info.range}")
            elif partition_type == "HASH":
                hash_info = table.hash_partitioning
                print(f"哈希分区字段：{hash_info.fields}，分区数量：{hash_info.num_partitions}")
            else:
                print("该表未设置分区")
    except NotFound:
        print(f"表 {dataset1} 不存在")

说明

• partitioning_type会返回以下值之一：TIME（时间分区）、RANGE（范围分区）、HASH（哈希分区），未分区表则返回None。
• 不同分区类型对应专属属性：时间分区用time_partitioning，范围分区用range_partitioning，哈希分区用hash_partitioning，可按需提取详细配置。

内容的提问来源于stack exchange，提问作者Max Daniel