在django中优化模型查找

Question

我有一个ToDo模型，它代表一个任务。每项任务都属于一个组织，该组织内的一个团队和一名工作人员。

这是我的筛选任务，在访问量：

def task_list(request, param=None, param_id=None): 
    if param == "org": 
     tasks = ToDo.objects.filter(org__id = param_id).exclude(todo_status=4) 
    elif param == "orgstaff": 
     tasks = ToDo.objects.filter(assigned_to__id = param_id).exclude(todo_status=4) 
    elif param == "orgteam": 
     tasks = ToDo.objects.filter(team__id = param_id).exclude(todo_status=4) 
    return render(request, "task_list.html", {"tasks":tasks}) 

有没有一种方法，我可以创建类型的字典，并更换这些上述6行到1或2？

Answer 1

使用字典和dict unpacking：

def task_list(request, param=None, param_id=None): 
    field_mapping = {'org': 'org__id', 'orgstaff': 'assigned_to__id', 'orgteam': 'team__id'} 
    tasks = ToDo.objects.filter(**{field_mapping[param]: param_id}).exclude(todo_status=4) 
    return render(request, "task_list.html", {"tasks":tasks})