# 序列化
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
继承Modelviews重写(继承或覆盖)get,post,put,delete请求
最新推荐文章于 2024-05-28 16:25:40 发布