Django REST中实现消息标记已读方法的路由问题求解

你的问题出在试图直接把视图集里的单个方法注册到路由，DRF的SimpleRouter是用来注册视图集类的，不是单独的视图函数，所以才会报base_name的错误。下面给你两种最常用的解决方案：

方案一：用DRF的@action装饰器（推荐） #

这是DRF扩展视图集自定义动作的标准方式，不需要额外修改路由，router会自动识别并生成对应URL。

步骤1：修改视图集代码 #

先修正mark_read方法的参数（视图集方法必须包含self，如果需要请求对象还要加request），然后用@action装饰器标记这个自定义动作：

from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response
from .models import Message
from .serializers import MessageSerializer

class MessageViewSet(viewsets.ModelViewSet):
    serializer_class = MessageSerializer
    queryset = Message.objects.filter(isread=False)

    # detail=False 表示这是一个批量操作（不是针对单个实例）
    # methods 指定允许的请求方式，这里用get/post都可以
    @action(detail=False, methods=['get', 'post'])
    def mark_read(self, request):
        # 只更新未读的消息，避免重复操作已读消息
        updated_count = Message.objects.filter(isread=False).update(isread=True)
        return Response({'read': updated_count})

步骤2：路由保持不变 #

你的原路由配置不用改：

from rest_framework.routers import SimpleRouter
from .views import MessageViewSet, MainView

router = SimpleRouter()
router.register(r'api/get_messages', MessageViewSet)
urlpatterns = [
    url(r'^$', MainView.as_view(), name='main'),
    url(r'^', include(router.urls))
]

现在你可以通过/api/get_messages/mark_read/这个URL访问标记已读的功能啦～

方案二：单独写一个APIView（更灵活） #

如果不想依赖视图集的结构，可以单独创建一个APIView来处理标记已读的逻辑：

步骤1：创建新的视图类 #

from rest_framework.views import APIView
from rest_framework.response import Response
from .models import Message

class MarkAllMessagesRead(APIView):
    def get(self, request):
        updated_count = Message.objects.filter(isread=False).update(isread=True)
        return Response({'read': updated_count})

步骤2：添加路由 #

在urls.py里直接注册这个视图的路由：

from django.conf.urls import url
from .views import MessageViewSet, MainView, MarkAllMessagesRead
from rest_framework.routers import SimpleRouter

router = SimpleRouter()
router.register(r'api/get_messages', MessageViewSet)
urlpatterns = [
    url(r'^$', MainView.as_view(), name='main'),
    # 新增标记已读的路由
    url(r'^api/mark_read/$', MarkAllMessagesRead.as_view(), name='mark_all_read'),
    url(r'^', include(router.urls))
]

现在访问/api/mark_read/就能批量标记所有未读消息为已读了。

额外提醒 #

你原来的mark_read方法里用Message.objects.update(isread=True)会把所有消息（包括已读的）都更新一遍，建议加上filter(isread=False)，这样只处理未读消息，返回的updated_count也更准确。

内容的提问来源于stack exchange，提问作者Frankenstein Dumbledore