Django |从查询获得的有序字典有关系

Question

我一个模型中有一个工作（但丑陋的）方法：

@classmethod 
def get_tasks(cls, user): 
    tasks = cls.objects\ 
     .filter(agent=user)\ 
     .prefetch_related('parent', 'agent') 
    ordered_tasks = {} 
    for task in tasks: 
     if task.parent.pk not in ordered_tasks: 
      ordered_tasks[task.parent.pk] = { 
       'parent': task.parent, 
       'tasks': [] 
      } 
     ordered_tasks[task.parent.pk]['tasks'].append(task) 
    return ordered_tasks 

它的工作好，但我敢肯定有一个更好的解决方案，具有相同的结果。

任何想法引导我？

Answer 1

使用字典的setdefault()方法：

ordered_tasks = {} 
for task in tasks: 
    ordered_tasks.setdefault(task.parent.pk, {'parent': task.parent, 
               'tasks': []} 
          )['tasks'].append(task)