如何检查（统计）Peewee查询是否返回数据及返回数据量？

我之前也碰到过类似的问题，其实Peewee本身提供了几个简洁高效的方法来处理这两个需求，不用自己绕弯路。下面结合常见场景给你梳理一下：

1. 统计返回的数据量 #

如果只是想知道查询结果的总数，优先用Query对象自带的count()方法，它会直接生成并执行COUNT(*)的SQL语句，比把所有结果加载到内存再用len()高效得多，尤其是数据量大的时候。

举个例子：

# 假设你的查询是获取名字包含"Bob"的用户
user_query = User.select().where(User.name.contains("Bob"))

# 统计数量
total_records = user_query.count()
print(f"一共查到 {total_records} 条数据")

2. 检查查询是否返回数据 #

如果你只需要判断有没有数据，不需要具体内容，用exists()方法是最优解——它会执行EXISTS子查询，只要找到匹配的记录就立即返回，不用遍历全量数据：

if user_query.exists():
    print("查询到数据啦！")
else:
    print("没有找到匹配的数据")

如果你之后还要用到查询结果，可以结合first()或者get_or_none()来判断：

• first()会返回查询结果的第一条记录，没有数据就返回None
• get_or_none()和get()类似，但找不到数据时不会抛出异常，直接返回None

示例代码：

# 用first()判断并获取第一条数据
first_user = user_query.first()
if first_user:
    print("有数据，第一条是：", first_user.name)
    # 如果需要所有结果，直接转成列表即可
    all_users = list(user_query)
    print(f"总共 {len(all_users)} 条")
else:
    print("无匹配数据")

# 用get_or_none()（适合确定最多只有一条结果的场景）
target_user = user_query.get_or_none()
if target_user:
    print("找到目标用户：", target_user.name)
else:
    print("没找到目标用户")

避坑提醒 #

• 尽量不要用len(query)来统计数量，因为它会把所有查询结果加载到内存中，数据量大的时候会拖慢性能，甚至占用过多内存。
• 如果用get()方法，当没有匹配数据时会抛出DoesNotExist异常，所以一定要配合try-except块使用，或者直接换get_or_none()更省心。

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