PostgreSQL EXISTS查询返回值无法被Python IF语句识别

问题根源分析 #

• 第一个代码块中，fetchall()返回空结果时是空列表[]，而非空字符串""，你的elif条件永远不会触发；另外直接比较整个结果列表不够简洁，应该提取布尔值本身做判断。
• 第二个代码块中，传递参数时sqldata = (phonestring)不是合法元组（Python中单个元素的元组必须加逗号），会导致psycopg解析参数出错，进而返回错误结果；同时判断逻辑可以更直接，无需构造列表再取元素。

修正后的代码示例 #

方法1：使用fetchall()的正确写法 #

def track_exists():
    conn = psycopg.connect("dbname=postgres user=postgres password=***")
    cur = conn.cursor()
    sqltest = "SELECT EXISTS(SELECT 1 FROM CUSTOMERS WHERE CUSTOMERS.PHONE = '608-888-8888')"
    cur.execute(sqltest)
    result = cur.fetchall()
    # 提取结果中的布尔值判断，同时处理空结果情况
    if result and result[0][0]:
        print("working")
        dosomething()
    else:
        print("not working")
    # 关闭游标和连接，避免资源泄漏
    cur.close()
    conn.close()

方法2：使用fetchone()的正确写法（更推荐） #

phonestring = phone_str
print(phonestring)
conn = psycopg.connect("dbname=postgres user=postgres password=***")
cur = conn.cursor()
sqltest = "SELECT EXISTS(SELECT 1 FROM CUSTOMERS WHERE CUSTOMERS.PHONE = (%s))"
# 单个元素的元组必须加逗号，否则会被当成字符串拆分处理
sqldata = (phonestring,)
cur.execute(sqltest, sqldata)
result = cur.fetchone()
# 直接取元组的第一个元素判断布尔值
if result and result[0]:
    print("Customer Exists")
else:
    print("New customer")
# 关闭资源
cur.close()
conn.close()

关键注意事项 #

• 参数传递必须是元组：单个参数时要写成(value,)，否则psycopg会将字符串拆分为单个字符作为参数，导致SQL执行错误。
• 提取布尔值而非直接比较结构：fetchone()返回单个元组（如(True,)），fetchall()返回元组列表（如[(True,)]），直接取元组的第一个元素result[0]来判断布尔值更可靠。
• 务必关闭游标和连接：避免数据库连接资源泄漏。

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