Snowpark Python存储过程：如何从session对象获取可读日志信息？

问题场景 #

我用Python基于Snowpark API编写存储过程，需要记录每次DML/查询操作的执行情况。之前通过构造SQL字符串调用session.call(SQLString)时，能直接记录构造好的SQL，比如JavaScript里的示例：

const EXAMPLE_QUERY = `CALL EXAMPLE_PROCEDURE('${ARG1}', '${ARG2}', '${ARG3}');`;
let exampleStatement = snowflake.createStatement({sqlText: EXAMPLE_QUERY});
log(EXAMPLE_QUERY, "info", `Exit code: ${exitCode}`);

但现在用Snowpark的session对象方法（比如session.table()、select()、distinct()）时，没有手动构造SQL字符串，不知道该从session或Query对象中提取什么内容存入日志。不需要完整SQL，但要能清晰了解数据操作情况，当前代码最后一行的log(get_tables_query, ...)无法正常工作。

可行解决方案 #

1. 提取Snowpark自动生成的SQL语句 #

Snowpark的Query/DataFrame对象提供了内部方法_sql()，可以直接获取对应的SQL字符串，和手动构造的SQL格式一致，适合需要完整SQL日志的场景：

# 替换原有的log行
log(get_tables_query._sql(), 'info', f'Unique tables returned: {len(example_output)}')

这样日志里会记录类似SELECT DISTINCT "EXAMPLE_COLUMN" FROM "EXAMPLE_DB"."EXAMPLE_SCHEMA"."EXAMPLE_TABLE"的完整SQL语句，和之前JS的方式效果一致。

2. 记录操作元数据（无SQL依赖） #

如果不想依赖内部方法，可手动拼接操作的关键元数据，同样能清晰反映操作情况：

operation_meta = {
    "operation_type": "DISTINCT 查询",
    "target_table": example_table,
    "selected_columns": ["EXAMPLE_COLUMN"],
    "result_row_count": len(example_output)
}
# 将元数据转为字符串存入日志
log(str(operation_meta), 'info', '查询完成')

日志中会显示结构化的操作信息，即使没有SQL也能快速了解操作内容和结果。

3. 手动构造简洁操作描述 #

如果不需要完整SQL，可基于Snowpark的操作逻辑手动拼接简洁的描述字符串，和之前构造SQL的思路类似：

operation_desc = f"从表 {example_table} 中查询 EXAMPLE_COLUMN 的唯一值"
log(operation_desc, 'info', f'返回唯一值数量: {len(example_output)}')

这种方式简单直接，日志可读性强，适合不需要精确SQL的场景。

注意事项 #

• _sql()是Snowpark的内部方法，虽然当前版本可用，但后续版本更新可能有变动，若追求稳定性，优先选择后两种方案。
• 对于DML操作（如update、delete），可通过WriteOperation对象的相关属性获取操作表、条件，或结合row_count()方法记录影响行数。

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