flyapps/fir_ser/admin/views/order.py

#!/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
from api.models import UserInfo, Order
from common.core.auth import AdminTokenAuthentication
from common.core.response import ApiResponse
from common.utils.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="数据校验失败")