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.
62 lines
2.2 KiB
62 lines
2.2 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 4月
|
|
# author: liuyu
|
|
# date: 2021/4/11
|
|
|
|
import logging
|
|
|
|
from django_filters import rest_framework as filters
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework.filters import OrderingFilter
|
|
|
|
from admin.utils.serializer import AdminOrdersSerializer
|
|
from admin.utils.utils import BaseModelSet, AppsPageNumber, ApiResponse
|
|
from api.models import UserInfo, Order
|
|
from api.utils.auth import AdminTokenAuthentication
|
|
from api.utils.storage.caches import update_order_info, admin_change_user_download_times
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class OrderFilter(filters.FilterSet):
|
|
user_id = filters.NumberFilter(field_name="user_id__id")
|
|
|
|
class Meta:
|
|
model = Order
|
|
fields = ["id", "payment_type", "payment_name", "payment_number", "order_number", "status", "order_type"]
|
|
|
|
|
|
class OrderInfoView(BaseModelSet):
|
|
authentication_classes = [AdminTokenAuthentication, ]
|
|
queryset = Order.objects.all()
|
|
serializer_class = AdminOrdersSerializer
|
|
pagination_class = AppsPageNumber
|
|
|
|
filter_backends = [DjangoFilterBackend, OrderingFilter]
|
|
ordering_fields = ['created_time', 'pay_time', 'actual_amount']
|
|
filterset_class = OrderFilter
|
|
|
|
def update(self, request, *args, **kwargs):
|
|
instance = self.get_object()
|
|
old_status = instance.status
|
|
data = super().update(request, *args, **kwargs).data
|
|
new_status = request.data.get('status', -99)
|
|
if old_status != new_status and new_status == 0:
|
|
update_order_info(instance.user_id.pk, instance.order_number, instance.order_number, instance.payment_type)
|
|
return ApiResponse(**data)
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
data = request.data
|
|
pk = data.get("id", None)
|
|
amount = data.get("amount", 0)
|
|
if not pk:
|
|
return ApiResponse(code=1003, msg="参数错误")
|
|
obj = UserInfo.objects.filter(pk=pk).first()
|
|
if obj:
|
|
if amount > 0:
|
|
if admin_change_user_download_times(obj, amount):
|
|
return ApiResponse()
|
|
else:
|
|
return ApiResponse(code=1005, msg="订单创建失败")
|
|
return ApiResponse(code=1004, msg="数据校验失败")
|
|
|