我尝试使用peewee来获取和格式化来自sqlite数据库的一些数据,使用groupconcat和Case。但我面临着这些功能的问题。在
首先让我们从我想要实现的目标开始:
为了更好地指出问题,我简化了我的表结构:1个包含两个列的简单表:name(Char),is_controlled(Boolean)。 SQL请求计算所需的结果:
SELECT
SUM(is_controlled),
GROUP_CONCAT(CASE WHEN is_controlled = 1 THEN name ELSE NULL END, ':') as controlled,
GROUP_CONCAT(CASE WHEN is_controlled = 0 THEN name ELSE NULL END, ':') as not_controlled
FROM component;
输出(这是我对peewee的期望):
^{pr2}$下面是一个允许测试我的问题的脚本:
from peewee import *
db = SqliteDatabase('test.db')
class Component(Model):
name = CharField()
is_controlled = BooleanField()
class Meta:
database = db
raw_data = [
{'name': 'comp1', 'is_controlled': True},
{'name': 'comp2', 'is_controlled': False},
{'name': 'comp3', 'is_controlled': True},
]
db.connect()
# Populate database
db.create_tables([Component])
for item in raw_data:
Component.get_or_create(**item)
res = Component.select(
fn.Sum(Component.is_controlled).alias('controlled_count'),
fn.GROUP_CONCAT(Case(None, [((Component.is_controlled == True), Component.name)], None), ':').alias('controlled'),
fn.GROUP_CONCAT(Case(None, [((Component.is_controlled == False), Component.name)], None), ':').alias('not_controlled')
)
print res[0].controlled_count
print res[0].controlled
print res[0].not_controlled
db.close()
如您所见,数据结构很简单(我最大限度地简化了示例)。输出是:
2
:comp3:
:
我检查了peewee生成的SQL查询(使用资源sql())看起来是这样的:
sql = 'SELECT Sum("t1"."is_controlled") AS "controlled_count", GROUP_CONCAT(CASE WHEN ("t1"."is_controlled" = ?) THEN ? END, "t1"."name") AS "controlled", GROUP_CONCAT(CASE WHEN ("t1"."is_controlled" = ?) THEN ? END, "t1"."name") AS "not_controlled" FROM "component" AS "t1"'
params = [True, ':', False, ':']
我们可以看到,peewee生成的SQL请求中缺少ELSE NULL部分。我试过几件事,比如调整Case函数的参数,但是我不能让它正常工作。在
如何正确使用peewee与使用SQL有相同的结果?在
(我使用的是python2.7.15和peewee3.6.4anssqlite 3.19.4)
Case
函数的签名提供了一个线索:在代码内部,它检查:
^{pr2}$所以,当你传递
None
时,它与“空/未指定”的情况是无法区分的,Peewee会忽略它。在作为一种解决方法,您可以指定
SQL('NULL')
作为默认值。或者您可以使用一个空字符串,尽管我不确定您是否依赖于带有null的组concat行为,因此这可能不起作用?在相关问题 更多 >
编程相关推荐