Django 嵌套查询集

Question

我有一个像这样的 Django 数据模型（省略了数据字段）：

class Atom(Model):
    pass

class State(Model):
    atom = ForeignKey(Atom)

class Transition(Model):
    atom = ForeignKey(Atom)
    upstate = ForeignKey(State,related_name='uptrans')
    lostate = ForeignKey(State,related_name='lotrans')

当我查询时，要限制的字段可以在任一模型中，因此在 Transition.objects.filter(...)< 上查询是最简单的/code> 因为其他模型中的所有字段都可以通过外键访问。我们将生成的查询集称为 t。

现在我还想要一个与 t 对应的 Atom 模型的 QuerySet a ，可以像 a = t.values('atom' 那样完成).distinct()。到目前为止，一切都很好。

但是，我还希望 a 中的每个条目都有 一个 属性/字段，用于保存该 Atom 状态的查询集，仍然反映原始选择的标准t，通过 upstate 或 lostate 外键之一。

到目前为止，我已经通过循环 t、添加 values('upstate_id') 和 values('lostate_id') 在 State 上创建了查询集> 到 Python set() 以丢弃重复项，然后使用此列表查询 State。但这样我就无法实现原子内状态的嵌套结构。

如果可能的话，可以使用未评估的 QuerySet 来提供有关如何执行此操作的任何建议，因为我不是将它们传递到模板中，而是传递到生成器（yield 语句）中，这是一个传输大量数据的好方法。

Answer 1

我认为以下函数执行我上面描述的操作，但我不确定按原子进一步过滤原始 QuerySet 的循环是否是正确的方法。

def getAtomsWithStates(t):
    atom_ids = set( t.values_list('atom_id',flat=True) )
    atoms = Atoms.objects.filter(pk__in=atom_ids)
    for atom in atoms:
        upstate_ids = t.filter(atom=atom).values_list('upstate_id',flat=True)
        lostate_ids = t.filter(atom=atom).values_list('lostate_id',flat=True)
        all_ids = set( upstate_ids + lostate_ids )

        # attach the new QuerySet to the entry in the outer one:
        atom.States = State.objects.filter(pk__in=all_ids)

    return atoms

现在我可以像这样执行我需要的嵌套循环：

someAtoms = getAtomsWithStates( Transition.objects.filter(...) )
for atom in someAtoms:
    for state in atom.States:
        print state.field

但是，再一次，可能有一个更智能的解决方案，我肯定会感兴趣。

Answer 2

很高兴您能够理解集合。但是，使用 SQL 的 In 不会重复您的数据。让我们考虑一下。如果我说，“给我一个在此列表中的原子：(1, 2, 3, 3, 3, 4)”，数据库将返回原子 1, 2, 3 和 4。为了简化，我不会要求 Python 执行 set 算术，因为数据库应该能够很好地处理它。有时需要使用 set，但您的场景似乎不是其中之一。

您的替代方案：

states = State.objects.filter(
    Q(pk__in=t.values_list('upstate', flat=True)) |
    Q(pk__in=t.values_list('lostate', flat=True)
)

即便如此，您的模型似乎可以使用一些更改，但我不完全理解您想要完成的任务。请注意，在我的替代方案中，我不对原子做任何事情。我使用 Q 对象来执行 OR 运算，但您也许可以向状态模型添加一个标志来指示它是高还是低。或者您可以使用带有直通表的 M2M 关系。为什么你的转变和你的状态都与原子相关联？您可以从 Transition 中消除 atom 并从 State 中获取 atom，如下所示：

atoms = Atom.objects.filter(
    pk__in=State.objects.filter(
        Q(pk__in=t.values_list('upstate', flat=True)) |
        Q(pk__in=t.values_list('lostate', flat=True)
    ).values_list('atom', flat=True)
)