用于自定义操作的 Django Rest Framework URL 不起作用

问题描述

我认为有以下自定义操作：

class OrderAPIViewSet(viewsets.ViewSet):

    def create(self,request):
        print("Here: working")

    @action(detail=True,methods=['post'])
    def add(self,request,*arg,**kwargs):
        print("HERE in custom action")
        order = self.get_object()
        print(order)

我的应用程序的 urls.py 是：

from rest_framework import routers
from .views import OrderAPIViewSet


router = routers.DefaultRouter()
router.register(r'orders',OrderAPIViewSet,basename='order')

urlpatterns = router.urls

所以在我的测试中，当我尝试访问 orders/post 时它可以工作，但是当我尝试访问 orders/{pk}/add 时它失败了。我的意思是，反向本身是失败的：

ORDERS_LIST_URL = reverse('order-list')
ORDERS_ADD_URL = reverse('order-add')

class PublicOrderApiTests(TestCase):

   def test_sample_test(self):
       data = {}
       res = self.client.post(ORDERS_ADD_URL,data,format='json')

正如我之前所说，我有一个单独的测试，我像这样使用 ORDERS_LIST_URL：

res = self.client.post(ORDERS_LIST_URL,format='json')

但是在运行测试时出现以下错误：

导入错误：无法导入测试模块：orders.tests Traceback （最近一次通话）：文件 “/usr/local/lib/python3.7/unittest/loader.py”，第 436 行，在 _find_test_path module = self._get_module_from_name(name) 文件“/usr/local/lib/python3.7/unittest/loader.py”，第377行，在 _get_module_from_name import(name) 文件“/app/orders/tests.py”，第 22 行，在 ORDERS_ADD_URL = reverse('order-add') 文件“/usr/local/lib/python3.7/site-packages/django/urls/base.py”，第 87 行， 相反 返回 iri_to_uri(resolver._reverse_with_prefix(view,prefix,*args,**kwargs)) 文件“/usr/local/lib/python3.7/site-packages/django/urls/resolvers.py”， 第 685 行，在 _reverse_with_prefix 中 raise noreverseMatch(msg) django.urls.exceptions.noreverseMatch：在没有找到任何参数的情况下反转“order-add”。 2 个图案 试过：['orders/(?P[^/.]+)/add\.(?P[a-z0-9]+)/?$','订单/(?P[^/.]+)/add/$']

----------------------------------------------- ----------------------- 在 0.000 秒内运行 1 个测试

失败（错误=1）

根据文档，我不需要注册此端点，路由器应该自行完成。我错过了什么？

解决方法

您错过的第一件事是背面的 pk。由于 add API 需要 Order 对象的 pk，因此您需要将其传递给 reverse 函数。例如：

order_add_url = reverse('order-add',kwargs={'pk': 1})
print(order_add_url)  # which will print '/orders/1/add/'

所以我认为您应该将这部分移到 PublicOrderApiTests 的方法体中，因为您需要每个测试对象的动态网址。

另一个问题是ViewSet类不支持self.get_object()，如果你想使用这个方法你应该有自己的方法或者使用rest框架GenericViewSet（即{{ 1}} 并从此类继承而不是 from rest_framework.viewsets import GenericViewSet) 然后您可以访问 ViewSet 方法。您还可以在 rest framework docs 中阅读有关通用视图的更多信息。