Question

我有以下型号：

class Document(models.Model):
    ...

class DocumentAttributes(models.Model):
    document = models.ForeignKey(Document)
    key = models.TextField()
    value = models.TextField()

我想根据属性查询文档。指定的键必须与其中一个值匹配。

可能最好用一个例子：

    self.d1 = document_factory(attributes={'a': '1', 'b': '1'})
    self.d2 = document_factory(attributes={'a': '2', 'b': '2'})
    self.d3 = document_factory(attributes={'a': '2', 'b': '1'})
    self.d4 = document_factory(attributes={'a': '3', 'b': '4'})
    self.d5 = document_factory(attributes={'a': '3', 'b': '2'})
    self.d6 = document_factory(attributes={'a': '1', 'b': '4'})
    self.d7 = document_factory(attributes={'a': '2', 'b': '4'})

 docs = whitelist_keyvalue_in({'a': ['1', '3'], 'b': ['1', '4']}, doc_qs).all()

文档现在应该包含d1，d4，d6。

这是我的实施：

def whitelist_keyvalue_in(json_obj, doc_qs):
    qs = doc_qs
    for key in json_obj:
        values = [json_obj[key]] if isinstance(json_obj[key], basestring) else json_obj[key]
        q_values = Q()
        for v in values:
            q_values |= Q(value=v)
        qs = qs.filter(attributes=DocumentAttributes.objects.filter(key=key).filter(q_values))
    print(qs.query)
    return qs

出于某种原因，这只会返回d1？并且生成的查询并不完美。

你能发现任何错误吗？有没有更好的方法来写这个？

SELECT ... FROM "document_document"
INNER JOIN "document_documentattributes" ON ("document_document"."id" = "document_documentattributes"."document_id")
INNER JOIN "document_documentattributes" T3 ON ("document_document"."id" = T3."document_id")
WHERE
("document_documentattributes"."id" = ( SELECT U0."id"
                       FROM "document_documentattributes" U0
                       WHERE (U0."key" = 'a' AND (U0."value" = '1' OR U0."value" = '3')))
                       AND T3."id" = ( SELECT U0."id"
                                FROM "document_documentattributes" U0
                                WHERE (U0."key" = 'b' AND (U0."value" = '1' OR U0."value" = '4'))))

如果我自己使用原始查询，那么工作正常：

def whitelist_keyvalue_in(json_obj, doc_qs):
    names = {key: 'da{}'.format(k_index) for k_index, key in enumerate(json_obj)}
    raw_sql = "SELECT da0.document_id as id FROM document_documentattributes as da0 "
    for key in json_obj:
        if names[key] == 'da0':
            continue
        raw_sql += ("JOIN document_documentattributes as {0} ON {0}.document_id = da0.document_id "
                    "".format(names[key]))
    for key in json_obj:
        where_and = 'WHERE' if names[key] == 'da0' else ' AND'
        values = [json_obj[key]] if isinstance(json_obj[key], basestring) else json_obj[key]
        values_opts = ' OR '.join("{}.value = '{}'".format(names[key], value) for value in values)
        raw_sql += "{} {}.key = '{}' AND ({})".format(where_and, names[key], key, values_opts)
    return doc_qs.filter(id__in=(d.id for d in doc_qs.raw(raw_sql)))

给出：

SELECT da0.document_id as id 
FROM document_documentattributes as da0 
JOIN document_documentattributes as da1 ON da1.document_id = da0.document_id 
WHERE da0.key = 'a' AND (da0.value = '1' OR da0.value = '3') 
  AND da1.key = 'b' AND (da1.value = '1' OR da1.value = '4')

SELECT ... FROM "document_document" WHERE "document_document"."id" IN (1, 4, 6)

我更愿意避免使用id__in，但却无法弄清楚如何从原始查询集到常规查询集。

如果我必须使用原始sql，有没有办法可以避免两个选择返回正常的查询集？

Answer 1

你的

qs = qs.filter(attributes=DocumentAttributes.objects.filter(key=key).filter(q_values))位于for循环中，每次应用过滤器时，其行为都类似于and条件。因此，最终的查询将是获取a is 1 or 3 and b is 1 or 4所在的文档。这里d1与条件匹配并返回，因为它有a = 1和b = 1。

在每个密钥周围使用Q()和|可以解决此问题。

此行qs = qs.filter(attributes=DocumentAttributes.objects.filter(key=key).filter(q_values))将触发两个查询，因为DocumentAttributes.objects.filter(key=key).filter(q_values)将被评估为在qs查询中使用

Answer 2

好的，终于想出了如何做到正确。生成的原始查询甚至可以很好地查看。

def whitelist_keyvalue_in(attributes, doc_qs):
    qs = doc_qs
    for key, values in attributes.iteritems():
        values = [values] if isinstance(values, basestring) else values
        qs = qs.filter(attributes__key=key, attributes__value__in=values)
    return qs

Django - 同一个表上的多个连接 - 结果不正确？

2 个答案: