如何在 ponyorm 中进行行计数? Python

how to make a rowcount in ponyorm? Python

我正在使用这个 Python ORM 来管理我的应用程序中的数据库:ponyorm.com

我刚刚更改了 table 中的一个属性,我会向我计算行数,他 return 1 表示 TRUE,0 表示 FALSE。

例如:仅使用 sqlite3 我会这样做:

user = conn.execute('SELECT * FROM users')

count = user.rowcount

if count == 1:
    print('Return %d lines' %count)
else:
    print('Bad....return %d lines', %count)

使用 rowcount 属性通常不是计算行数的正确方法。根据有关 rowcount 属性的 SQLite 文档,

Although the Cursor class of the sqlite3 module implements this attribute, the database engine’s own support for the determination of “rows affected”/”rows selected” is quirky.

如果您使用 SQL,获取行数的标准方法是使用 COUNT(*) 函数。在SQLite中可以通过以下方式实现:

cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM users')
rowcount = cursor.fetchone()[0]

使用 PonyORM,您可以通过三种替代方式进行计数。

rowcount = select(u for u in User).count()  # approach 1
rowcount = User.select().count()  # approach 2
rowcount = count(u for u in User)  # approach 3

上面的所有行都应该产生相同的查询。您可以选择您认为最直观的那条线。

如果您不想统计所有行,而只想统计特定行,您可以在查询中添加条件。例如,要计算价格大于 100 的产品数量,您可以编写以下任何一行:

rowcount = select(p for p in Product if p.price > 100).count()  # approach 1
rowcount = Product.select(lambda p: p.price > 100).count()  # approach 2
rowcount = count(p for p in Product if p.price > 100)  # approach 3

此外,您可能不想计算行数,而是计算特定列中不同值的数量。例如,不同用户国家/地区的数量。这可以通过以下方式完成:

user_countries_count = select(count(u.country) for u in User).get()

希望我回答了你的问题。