Django - 多个连接在同一个表上 - 错误的结果？

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

OK终于想通了如何得到它的权利。生成的原始查询甚至很好看。

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 

Answer 2

你

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查询中使用