python - 检测是否从 psycopg2 游标中获取？

Question

假设我执行以下命令。

insert into hello (username) values ('me')

然后我跑了起来

cursor.fetchall()

出现以下错误

psycopg2.ProgrammingError: no results to fetch

如何在不检查查询是“insert”还是“select”的情况下检测是否调用 fetchall()？

谢谢。

Answer 1

看这个属性:

cur.description

执行查询后，如果没有返回任何行，它将被设置为 None，否则将包含数据 - 例如:

(Column(name='id', type_code=20, display_size=None, internal_size=8, precision=None, scale=None, null_ok=None),)

捕获异常并不理想，因为在某些情况下您可能会覆盖真正的异常。