1

我刚刚偶然发现了 Django Rest Framework 遇到的最棘手的问题。让我先给你我的模型,然后解释:

class Stampcardformat(models.Model):
    workunit    = models.ForeignKey(
        Workunit,
        on_delete=models.CASCADE
    )
    uuid        = models.UUIDField(
        default=uuid.uuid4,
        editable=False,
        unique=True
    )
    limit       = models.PositiveSmallIntegerField(
        default=10
    )
    category    = models.CharField(
        max_length=255
    )


class Stampcard(models.Model):
    stampcardformat = models.ForeignKey(
        Stampcardformat,
        on_delete=models.CASCADE
    )
    user        = models.ForeignKey(
        User,
        on_delete=models.CASCADE
    )
    uuid        = models.UUIDField(
        default=uuid.uuid4,
        editable=False,
        unique=True
    )


class Stamp(models.Model):
    stampcardformat = models.ForeignKey(
        Stampcardformat,
        on_delete=models.CASCADE
    )
    stampcard = models.ForeignKey(
        Stampcard,
        on_delete=models.CASCADE,
        blank=True,
        null=True
    )
    uuid        = models.UUIDField(
        default=uuid.uuid4,
        editable=False,
        unique=True
    )

这些模型描述了一个简单的邮票卡模型。一张邮票卡被认为是满的,当它通过外键关联的邮票与它的邮票卡格式的限制数量一样多时。我需要编写执行以下操作的视图:

  1. 该视图包含一个包含它们的 uuid 的标记列表(见下文)。
  2. 然后它需要为每个给定的邮票找到正确的邮票卡格式。
  3. 接下来,它需要检查请求用户是否有一张带有相应 stampcardformat 的邮票卡

    a)如果有,则需要检查印章卡是否已满

    i)如果已满,则需要新建给定格式的stampcard,并将stamps stampcard-foreignkey更新为创建的stampcard。

    ii)如果未满,则需要将 stamps stampcard-foreignkey 更新为找到的 stampcard

    b)如果用户没有给定stampcardformat 的stampcard,则需要创建一个新的stampcard,并将stamps stampcard-foreignkey 更新为创建的stampcard。

这是邮票的请求正文列表:

[
    {
        "stamp_uuid": "62c4070f-926a-41dd-a5b1-1ddc2afc01b2"
    },
    {
        "stamp_uuid": "4ad6513f-5171-4684-8377-1b00de4d6c87"
    },
    ...
]

基于类的视图似乎不支持这种行为。我尝试修改基于类的视图,但无济于事。除了很多点之外,我还失败了,因为视图抛出了错误:

AssertionError: Expected view StampUpdate to be called with a URL keyword argument named "pk". Fix your URL conf, or set the `.lookup_field` attribute on the view correctly.

编辑

对于其他上下文:我需要 url 没有 pk、slug 或任何东西。所以网址应该是这样的:

/api/stampcards/stamps/ 

并对其进行放置(或任何具有主体和工作的请求)。我写的路线是:

url(r'^stamps/$', StampUpdate.as_view(), name='stamp-api-update'),

编辑:巨大的更新。所以我设法拼凑出一个可行的观点。首先,我像这样更新了邮票卡模型(我确实添加了一个新字段“完成”来跟踪它是否已满):

class Stampcard(models.Model):
    stampcardformat = models.ForeignKey(
        Stampcardformat,
        on_delete=models.CASCADE
    )
    user        = models.ForeignKey(
        User,
        on_delete=models.CASCADE
    )
    uuid        = models.UUIDField(
        default=uuid.uuid4,
        editable=False,
        unique=True
    )
    done        = models.BooleanField(default=False)

然后我写了这样的视图:

class StampUpdate(APIView):
    permission_classes = (IsAuthenticated,)

    def get_object(self, uuid):
        try:
            return Stamp.objects.get(uuid=uuid)
        except Stamp.DoesNotExist():
            raise Http404

    def put(self, request, format=None):
    for stamp_data in request.data:
        stamp = self.get_object(stamp_data['stamp_uuid'])
        if stamp.stampcard==None:
            user_stampcard = self.request.user.stampcard_set.exclude(done=True).filter(stampcardformat=stamp.stampcardformat)
            if user_stampcard.exists():
                earliest_stampcard = user_stampcard.earliest('timestamp')
                stamp.stampcard = earliest_stampcard
                stamp.save()
                if earliest_stampcard.stamp_set.count() == earliest_stampcard.stampcardformat.limit:
                    earliest_stampcard.done=True
                    earliest_stampcard.save()
            else:
                new_stampcard = Stampcard(stampcardformat=stamp.stampcardformat, user=self.request.user)
                new_stampcard.save()
                stamp.stampcard = new_stampcard
                stamp.save()
    new_stampcards = Stampcard.objects.exclude(done=True).filter(user=self.request.user)
    last_full_stampcard = Stampcard.objects.filter(user=self.request.user).filter(done=True)
    if last_full_stampcard.exists():
        last_full_stampcard_uuid=last_full_stampcard.latest('updated').uuid
        last_full_stampcard = Stampcard.objects.filter(uuid=last_full_stampcard_uuid)
        stampcards = new_stampcards | last_full_stampcard
    else:
        stampcards = new_stampcards
    print(stampcards)
    stampcard_serializer = StampcardSerializer(stampcards, many=True)
    return Response(stampcard_serializer.data)

但是这段代码有两个问题:

  1. 我的直觉告诉我,在模型实例(例如)上调用 save() 的部分stamp.save()对于 api 来说是非常不安全的。我无法让它首先序列化数据。我的问题是:这种观点可以这样吗?或者我可以改进什么吗?例如,它不使用基于泛型类,但我不知道如何在这里使用它们......
  2. 如果用这种方法填满,我也很想退回邮票卡。但我也想排除所有不相关的邮票卡,这就是我打电话的原因.exclude(done=True)。不幸的是,一张被填满的邮票卡已经完成了=True!如何将在此过程中填写的印章卡添加到返回值?
4

1 回答 1

1

我不认为stamp.save()在 PUT 方法中使用它是不安全的,因为根据定义它应该改变对象的值。

要仅返回相关的邮票卡,您只需将邮票卡添加到这样的集合中

class StampUpdateView(APIView):
    def get_object(self, uuid):
        try:
            return Stamp.objects.get(uuid=uuid)
        except Stamp.DoesNotExist():
            raise Http404

    def put(self, request, *args, **kwargs):
        stampcard_set = set()
        for stamp_data in request.data:
            stamp = self.get_object(stamp_data['stamp_uuid'])

            user_stampcard = request.user.stampcard_set.exclude(done=True).filter(stampcardformat=stamp.stampcardformat)
            if user_stampcard.exists():
                stampcard = user_stampcard.earliest('timestamp')
            else:
                stampcard = Stampcard(stampcardformat=stamp.stampcardformat, user=request.user)
                stampcard.save()

            stamp.stampcard = stampcard
            stamp.save()

            if stampcard.stamp_set.count() == stampcard.stampcardformat.limit:
                stampcard.done = True
                stampcard.save()

            stampcard_set.add(stampcard)

        stampcard_serializer = StampcardSerializer(stampcard_set, many=True)
        return Response(stampcard_serializer.data)

这样,返回的邮票卡是否已经完成都没有关系。

另请注意,我将代码中的限制检查行向下移动到保存邮票之后,因为如果将限制设置为 1,则必须在添加邮票后立即将邮票卡设置为已完成。

于 2018-02-23T16:21:29.647 回答