继承Modelviews重写(继承或覆盖)get,post,put,delete请求

# 序列化
class KeyPointBuinessSerializers(serializers.ModelSerializer):
    class Meta:
        model = KeyPointBuiness
        fields = "__all__"

class BusinessConfigViewSet(ModelViewSet):
    queryset = KeyPointBuiness.objects.all()
    serializer_class = serializers.KeyPointBuinessSerializers

    # 重写get请求对应的方法
    def list(self, request, *args, **kwargs):
        data = []
        key_business_info = KeyPointBuiness.objects.all()
        for i in key_business_info:
            data_i = {}
            business_id_str = i.buisness_include
            title = i.title
            id = i.id
            print(id)
            if business_id_str:
                business_ids = business_id_str.split(',')
                buisness_include = list(Business.objects.filter(id__in=business_ids).values_list('name', flat=True))
            else:
                buisness_include = ""
            data_i['id'] = id
            data_i['title'] = title
            data_i['buisness_include'] = buisness_include
            data.append(data_i)
        return JsonResponse({'code': 0, 'msg': '数据正常', 'data': data})

    # 重写post请求对应的方法
    def create(self, request):
        title = request.POST.get('title')
        business_id_list = request.POST.getlist('buisness_include')
        if not title:
            return JsonResponse({'code': -1, 'msg': '标题必填'})
        if not business_name_list:
            return JsonResponse({'code': -1, 'msg': '请选择需要关联的业务'})
        title_exists = KeyPointBuiness.objects.filter(title=title)
        if title_exists:
            return JsonResponse({'code': -1, 'msg': '标题已经存在'})
        for business_id in list(business_id_list):
            KeyPointBuiness.objects.create(title=title, buisness_include=business_id)

        return JsonResponse({'code': 0, 'msg': '新建配置成功', 'data': []})

    # 重写PUT请求对应的方法
    def update(self, request, pk):  # request后面再有参数,就是路由参数,在路由最后的 / 后拼接的,一般id都用pk表示,是约定俗成的,方便代码阅读
        buisness_include = request.POST.get('buisness_include', '')
        k_business = KeyPointBuiness.objects.filter(id=pk)
        if not k_business:
            return JsonResponse({'code': -1, 'msg': '数据不存在'})
        k_business.update(buisness_include=buisness_include)
        return JsonResponse({'code': 0, 'msg': '数据修改成功', 'data': []})

    # delete请求的方法不修改了,但修改delete方法的返回值,让删除后提示符合自己的要求
    def destroy(self, request, *args, **kwargs):
        # 继承原来的destroy
        response = super().destroy(self, request, *args, **kwargs)
        # 重写返回值
        return Response({'code': 0, 'msg': '删除成功'})

# 类视图的路由要重新注册,不能和函数视图写在一起        
from rest_framework_bulk.routes import BulkRouter
router = BulkRouter()
urlpatterns = [
	url(xxxxxxxxxxxx),  # 函数视图的路由
]
router.register(r'xxxxxxxxxx', views.BusinessConfigViewSet, "business_config_details")  # 类视图路径
urlpatterns += router.urls
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值