你能在模型中按用户筛选查询集吗？

Question

我认为按用户筛选对象的正确方法是在视图中完成的，因为您可以很容易地要求登录，并且可以访问request.user。但是我有一个调用对象的classmethod的视图，它不能与筛选器结合，因为它将queryset转换为list对象-所以我想如果有一种方法可以通过用户直接在模型中筛选queryset，那么当我在视图中调用classmethod时，对象就已经被筛选了。

以下是我的观点：

def leads_by_city(request):
    # Invoke Lead classmethod to get the data
    data = Lead.objects.get_leads_per_city()

    return JsonResponse(data, safe=False)

带有自定义管理器的模型：

class Lead(models.Model):
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
    source = models.ForeignKey(LeadSource, on_delete=models.CASCADE)
    city = models.CharField(max_length=100)
    objects = LeadManager()


class LeadManager(models.Manager):

    def get_leads_per_city(self):
        queryset = self.values('city').annotate(Count('id')).order_by('city')
        data = list(queryset.values('city', 'id__count'))

        return data

在ClassMethod中有没有按用户筛选的方法？也许像：

class LeadManager(models.Manager):

    def get_leads_per_city(self,user):
        queryset = self.filter(user=Lead.user).values('city').annotate(Count('id')).order_by('city')
        data = list(queryset.values('city', 'id__count'))

        return data

Answer 1

是的，尽管您的筛选有点不正常，但您可以简单地设置.filter(user=user):

class LeadManager(models.Manager):

    def get_leads_per_city(self, user):
        queryset = self.filter(
            user=user
        ).values('city').annotate(Count('id')).order_by('city')
        return list(queryset.values('city', 'id__count'))

user=user参数名称不引用函数的user参数。Django ORM将查找具有该名称的字段，我们将参数调用的user传递给它。

在您看来，您可以使用以下命令调用此命令：

def leads_by_city(request):
    # Invoke Lead classmethod to get the data
    data = Lead.objects.get_leads_per_city(request.user)
    return JsonResponse(data, safe=False)

话虽如此，这里您使用Django的queryset机制来生成字典。这通常是不可取的。您可能希望使用序列化器，这是Django REST框架[drf-doc]提供的。

此外，将列表作为JSON响应传递也不是一个好主意。JSON利用了这个漏洞，您最好将其封装在字典中。

由于您需要登录的用户，因此最好在此处添加@login_required修饰符[Django-doc]，如果用户尚未登录，它将返回HTTP重定向响应：

from django.contrib.auth.decorators import login_required

@login_required
def leads_by_city(request):
    # Invoke Lead classmethod to get the data
    data = Lead.objects.get_leads_per_city(request.user)
    return JsonResponse({'data': data})