django - 计算 Django 数据库中的行数：AttributeError

Question

我有由 Django 模型创建的数据库，其中 idRecruteur 是表 Offre 的字段。

我需要计算 idRecruteur = 1 的行数。

这是我的“Offre”表代码：

@python_2_unicode_compatible
class Offre(models.Model):          
    titre = models.CharField(max_length=100)
    dateAjout = models.DateField(auto_now=False, auto_now_add=False)
    nature = models.CharField(max_length=50)
    duree = models.CharField(max_length=50)
    niveau = models.CharField(max_length=50)
    description = models.CharField(max_length=50)
    salaire = models.FloatField(null=True, blank=True)
    idRecruteur = models.ForeignKey(Recruteur, related_name="Offre", on_delete=models.CASCADE)
    def __str__(self):
        return "Offre: {}".format(self.title)

这是我的查询集：

class OffresparEntrepriseViewSet(ModelViewSet):
    queryset = Offre.objects.filter(idRecruteur=1).count()
    serializer_class = OffreSerializer

我收到错误“AttributeError: 'int' object has no attribute 'model'”

任何想法我做错了什么？

Answer 1

ViewSet 的queryset属性应该是QuerySet对象，但count()返回整数。您需要queryset = Offre.objects.filter(idRecruteur=1)用作视图集查询集，并将计数功能移至另一个级别。例如使用list_route：

class OffresparEntrepriseViewSet(ModelViewSet):
    queryset = Offre.objects.filter(idRecruteur=1)
    serializer_class = OffreSerializer

    @list_route()   
    def count(self, request):
        count = self.queryset.count()
        return Response({'count': count})

viewset_url/count现在您可以通过访问url来获得计数。