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)