From dc2fa091cb51ee18be555436cc034e0656057b17 Mon Sep 17 00:00:00 2001 From: youngS Date: Fri, 15 Oct 2021 13:39:41 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=87=AA=E5=AE=9A=E4=B9=89?= =?UTF-8?q?=E5=B9=BF=E5=91=8A=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fir_ser/api/views/advert.py | 141 ++++++++++++++++++++++++++++++++++++ 1 file changed, 141 insertions(+) create mode 100644 fir_ser/api/views/advert.py diff --git a/fir_ser/api/views/advert.py b/fir_ser/api/views/advert.py new file mode 100644 index 0000000..ce4cd52 --- /dev/null +++ b/fir_ser/api/views/advert.py @@ -0,0 +1,141 @@ +#!/usr/bin/env python +# -*- coding:utf-8 -*- +# project: 10月 +# author: liuyu +# date: 2021/10/14 +from rest_framework.views import APIView + +from api.utils.response import BaseResponse +from api.utils.auth import ExpiringTokenAuthentication, UserAdInfoPermission +from rest_framework.response import Response +from api.models import UserAdDisplayInfo +from api.utils.serializer import UserAdInfoSerializer +from rest_framework.pagination import PageNumberPagination +import logging + +from api.utils.storage.caches import reset_short_response_cache +from api.utils.storage.storage import Storage + +logger = logging.getLogger(__name__) + + +class PageNumber(PageNumberPagination): + page_size = 10 # 每页显示多少条 + page_size_query_param = 'size' # URL中每页显示条数的参数 + page_query_param = 'page' # URL中页码的参数 + max_page_size = None # 最大页码数限制 + + +class UserAdInfoView(APIView): + authentication_classes = [ExpiringTokenAuthentication, ] + permission_classes = [UserAdInfoPermission, ] + + def get(self, request): + + res = BaseResponse() + + ad_name = request.query_params.get("search_key", None) + ad_info_list = UserAdDisplayInfo.objects.filter(user_id=request.user) + if ad_name: + ad_info_list = ad_info_list.filter(ad_name=ad_name) + + page_obj = PageNumber() + page_paginate_obj = page_obj.paginate_queryset(queryset=ad_info_list.order_by("-created_time"), + request=request, + view=self) + page_serializer_obj = UserAdInfoSerializer(page_paginate_obj, many=True, ) + res.data = page_serializer_obj.data + res.count = ad_info_list.count() + return Response(res.dict) + + def put(self, request): + data = request.data + ad_pk = data.get("id", None) + if ad_pk: + ad_info_obj = UserAdDisplayInfo.objects.filter(user_id=request.user, pk=ad_pk).first() + res = BaseResponse() + update_fields = [] + logger.info(f"user {request.user} ad_info {ad_info_obj} update input data {data}") + f_fields = ["ad_name", "ad_uri", "description", "is_enable"] + for f_f in f_fields: + f_v = data.get(f_f) + if f_v is not None and f_v != '': + setattr(ad_info_obj, f_f, f_v) + update_fields.append(f_f) + else: + if f_f == 'description': + continue + else: + res.code = 1001 + res.msg = '参数有误' + return Response(res.dict) + try: + weight = int(data.get("weight", ad_info_obj.weight)) + if 0 <= weight <= 100: + ad_info_obj.weight = weight + update_fields.append("weight") + except Exception as e: + logger.error( + f"ad_info {ad_info_obj} weight {data.get('weight', ad_info_obj.weight)} get failed Exception:{e}") + + try: + ad_info_obj.save(update_fields=update_fields) + logger.info( + f"user {request.user} ad_info {ad_info_obj} update now data {ad_info_obj.__dict__}") + res.data = UserAdInfoSerializer(ad_info_obj).data + reset_short_response_cache(request.user) + return Response(res.dict) + except Exception as e: + logger.error( + f"user {request.user} ad_info {ad_info_obj} update error data {data} Exception {e}") + res.code = 1001 + return Response(res.dict) + + return self.get(request) + + def post(self, request): + data = request.data + data_info = {} + res = BaseResponse() + f_fields = ["ad_name", "ad_uri", "description", "is_enable"] + for f_f in f_fields: + f_v = data.get(f_f) + if f_v is not None and f_v != '': + data_info[f_f] = f_v + else: + if f_f == 'description': + continue + else: + res.code = 1001 + res.msg = "参数有误，添加失败" + return Response(res.dict) + + try: + weight = int(data.get("weight", 1)) + if 0 <= weight <= 100: + data_info["weight"] = weight + except: + pass + + try: + logger.error(f"user {request.user} add new ad {data.get('pk', '')} data {data_info}") + ad_info_obj = UserAdDisplayInfo.objects.create(user_id=request.user, **data_info) + res.data = UserAdInfoSerializer(ad_info_obj).data + return Response(res.dict) + except Exception as e: + logger.error(f"user {request.user} add new ad {data_info} failed Exception:{e}") + res.code = 1005 + res.msg = "广告名称已经存在，或者字段缺少" + return Response(res.dict) + + def delete(self, request): + ad_pk = request.query_params.get("pk", None) + if ad_pk: + ad_info_obj = UserAdDisplayInfo.objects.filter(user_id=request.user, pk=ad_pk).first() + if ad_info_obj: + storage = Storage(request.user) + storage.delete_file(ad_info_obj.ad_pic) + reset_short_response_cache(request.user) + ad_info_obj.delete() + + return self.get(request)