Python+Sqlite3中如何让if not语句优先于else执行?
解决SQLite SUM查询中返回NULL导致的判断问题
我懂你遇到的问题了!其实问题出在SQLite的聚合函数特性上,不是你的Python判断逻辑写错了~
问题根源
当你使用SUM()这类聚合函数时,哪怕没有符合WHERE条件的行,SQLite也会返回一行结果,这行的SUM值是NULL。所以c.fetchone()得到的不是None,而是一个包含None的元组(比如(None,))。这时候if not data:判断的是元组是否为空,显然这个元组不为空,所以永远会走else分支。
解决方案
这里有两种可靠的解决方式,推荐第二种,既安全又简洁:
方案1:在Python中检查返回值
直接判断元组里的第一个元素是否为None:
query = "SELECT SUM(column1) FROM test WHERE date BETWEEN ? AND ?" # 用参数化查询替代字符串拼接,避免SQL注入风险 c.execute(query, (blah_start, blah_end)) data = c.fetchone() if data[0] is None: amountsum = 0 else: amountsum = data[0] # 或者更简洁的写法 amountsum = data[0] if data[0] is not None else 0
方案2:在SQL中用COALESCE函数处理NULL
COALESCE函数会返回第一个非NULL的值,我们可以让SQL直接返回0而不是NULL:
# 用COALESCE把SUM的NULL结果替换成0 query = "SELECT COALESCE(SUM(column1), 0) FROM test WHERE date BETWEEN ? AND ?" # 同样用参数化查询,安全又规范 c.execute(query, (blah_start, blah_end)) data = c.fetchone() amountsum = data[0] # 这里永远不会得到None,直接赋值即可
额外提醒:一定要避免用字符串拼接方式构建SQL语句,这会带来SQL注入的安全风险,使用参数化查询(把变量放到execute的第二个参数里)才是正确的做法!
内容的提问来源于stack exchange,提问作者Oliver Almonte
相关产品推荐
相关产品推荐

