You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
5.2 KiB
134 lines
5.2 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 10月
|
|
# author: liuyu
|
|
# date: 2021/10/14
|
|
import logging
|
|
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from api.models import UserAdDisplayInfo
|
|
from api.utils.modelutils import PageNumber
|
|
from api.utils.response import BaseResponse
|
|
from api.utils.serializer import UserAdInfoSerializer
|
|
from common.core.auth import ExpiringTokenAuthentication, UserAdInfoPermission
|
|
from common.utils.caches import reset_short_response_cache
|
|
from common.utils.storage import Storage
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
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)
|
|
|