计算Django数据库中的行数:AttributeError

时间:2018-03-27 11:02:59

标签: django django-rest-viewsets

我有由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'对象没有属性' model' "

任何想法我做错了什么?

1 个答案:

答案 0 :(得分:0)

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网址来获取数据。