如何在Python中查询Snowflake最大日期并捕获表不存在错误？

核心思路 #

针对你的需求，需要实现两个关键目标：精准捕获表不存在的特定错误、正确获取查询结果。以下是优化后的代码及详细说明：

完整实现代码 #

import snowflake.connector
from snowflake.connector.errors import SnowflakeException
import pandas as pd

# 保持上下文管理确保连接自动释放
with read_snowflake(user=snf_user, warehouse=snf_warehouse, database=snf_db, schema=snf_schema) as snf_conn:
    max_date = '2023/01/01'  # 默认 fallback 日期
    cur = snf_conn.cursor()
    
    try:
        # 执行最大日期查询
        cur.execute("SELECT MAX(SN_DATE) MAX_DT FROM db.tbl.N_SNAPSHOT")
        # 获取单行结果（查询仅返回单个值）
        result_row = cur.fetchone()
        
        # 处理非空结果：表存在且有数据
        if result_row and result_row[0] is not None:
            max_date = result_row[0].strftime('%Y/%m/%d')
    
    except SnowflakeException as e:
        # 精准匹配「表不存在」错误（错误代码90014对应OBJECT_DOES_NOT_EXIST）
        if e.errno == 90014:
            print("目标表不存在，使用默认日期")
        else:
            # 其他Snowflake异常按需处理，比如抛出或记录日志
            print(f"Snowflake执行错误: {e}")
    
    finally:
        cur.close()  # 确保游标资源释放

print(f"最终使用日期: {max_date}")

关键细节说明 #

1. 捕获特定错误
   Snowflake的「对象不存在」错误对应代码90014，通过导入SnowflakeException并判断e.errno，可以精准捕获表不存在的场景，避免泛用Exception导致的误处理。

2. 正确获取查询结果
   你之前打印cur得到奇怪数字是因为直接打印游标对象本身，而非查询结果。使用fetchone()可以获取单行查询结果（你的查询仅返回一个值），再通过result_row[0]提取最大日期值。

3. 处理边界情况

   • 如果表存在但无数据，result_row[0]会是None，此时仍保留默认日期；
   • 用with语句管理连接，确保连接自动关闭，避免资源泄漏；
   • finally块强制关闭游标，进一步保证资源释放。

可选：用Pandas简化结果处理 #

如果习惯用Pandas操作数据，也可以这样实现：

with read_snowflake(user=snf_user, warehouse=snf_warehouse, database=snf_db, schema=snf_schema) as snf_conn:
    max_date = '2023/01/01'
    try:
        max_date_df = pd.read_sql("SELECT MAX(SN_DATE) MAX_DT FROM db.tbl.N_SNAPSHOT", snf_conn)
        # 检查结果非空且日期值有效
        if not max_date_df.empty and pd.notna(max_date_df['MAX_DT'].iloc[0]):
            max_date = max_date_df['MAX_DT'].iloc[0].strftime('%Y/%m/%d')
    except SnowflakeException as e:
        if e.errno == 90014:
            print("表不存在，使用默认日期")
        else:
            raise  # 其他错误抛出，按需调整

内容的提问来源于stack exchange，提问作者n8.