#!/usr/bin/env python # -*- coding:utf-8 -*- # project: 3月 # author: NinEveN # date: 2021/3/29 import logging from rest_framework.pagination import PageNumberPagination from rest_framework.response import Response from rest_framework.views import APIView from api.models import Price, Order from api.utils.auth import ExpiringTokenAuthentication from api.utils.pay.util import get_pay_obj_form_name, get_enable_pay_choices, get_payment_type from api.utils.response import BaseResponse from api.utils.serializer import PriceSerializer, OrdersSerializer from api.utils.storage.caches import update_order_status from common.base.baseutils import get_order_num, get_choices_dict 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 OrderView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] def get(self, request): res = BaseResponse() order_id = request.query_params.get("order_id", None) order_obj_lists = Order.objects.filter(user_id=request.user) if order_id: order_obj_lists = order_obj_lists.filter(order_number=order_id) page_obj = PageNumber() order_info_serializer = page_obj.paginate_queryset(queryset=order_obj_lists.order_by("-created_time"), request=request, view=self) order_info = OrdersSerializer(order_info_serializer, many=True, ) res.data = order_info.data res.count = order_obj_lists.count() res.payment_type_choices = get_choices_dict(Order.payment_type_choices) res.status_choices = get_choices_dict(Order.status_choices) res.order_type_choices = get_choices_dict(Order.order_type_choices) return Response(res.dict) def post(self, request): res = BaseResponse() price_id = request.data.get("price_id", None) pay_id = request.data.get("pay_id", None) order_number = request.data.get("order_number", None) if (price_id and pay_id) or order_number: price_obj = Price.objects.filter(name=price_id).first() order_obj = Order.objects.filter(user_id=request.user, order_number=order_number).first() if order_obj and order_obj.status in [1, 2] and order_obj.payment_name: pay_obj = get_pay_obj_form_name(order_obj.payment_name) if pay_obj: pay_url = pay_obj.get_pay_pc_url(order_number, int(order_obj.actual_amount), {'user_id': request.user.id}) res.data = pay_url logger.info(f"{request.user} 下单成功 {res.dict}") return Response(res.dict) if price_obj: try: order_number = get_order_num() actual_amount = price_obj.price pay_obj = get_pay_obj_form_name(pay_id) if pay_obj: pay_url = pay_obj.get_pay_pc_url(order_number, int(actual_amount), {'user_id': request.user.id}) Order.objects.create(payment_type=get_payment_type(pay_obj.p_type), order_number=order_number, user_id=request.user, status=1, order_type=0, actual_amount=actual_amount, actual_download_times=price_obj.package_size, payment_name=pay_obj.name, actual_download_gift_times=price_obj.download_count_gift) res.data = pay_url logger.info(f"{request.user} 下单成功 {res.dict}") return Response(res.dict) except Exception as e: logger.error(f"{request.user} 订单 {price_id} 保存失败 Exception:{e}") res.code = 1003 res.msg = "订单保存失败" else: logger.error(f"{request.user} 价格 {price_id} 获取失败") res.code = 1002 res.msg = "价格获取失败,请稍后重试" else: res.code = 1001 res.msg = "错误的价格" return Response(res.dict) def put(self, request): res = BaseResponse() order_number = request.data.get("order_number", None) act = request.data.get("act", None) if order_number: order_obj = Order.objects.filter(user_id=request.user, order_number=order_number).first() if order_obj: try: if act == 'cancel' and order_obj.status != 0: update_order_status(order_number, 5) elif act == 'status' and order_obj.status in [1, 2]: pass # alipay = Alipay() # alipay.update_order_status(order_obj.order_number) # wxpay = Weixinpay() # wxpay.update_order_status(order_obj.order_number) except Exception as e: logger.error(f"{request.user} 订单 {order_number} 更新失败 Exception:{e}") res.code = 1003 return Response(res.dict) else: logger.error(f"{request.user} 订单 {order_number} 获取失败") res.code = 1003 res.msg = "订单获取失败,或订单已经支付" else: res.code = 1001 res.msg = "订单有误" return Response(res.dict) class PriceView(APIView): authentication_classes = [ExpiringTokenAuthentication, ] def get(self, request): res = BaseResponse() price_obj_lists = Price.objects.filter(is_enable=True).all().order_by("updated_time").order_by("price") res.data = PriceSerializer(price_obj_lists, many=True).data res.pay_choices = get_enable_pay_choices() return Response(res.dict) def delete(self, request): res = BaseResponse() return Response(res.dict) def put(self, request): res = BaseResponse() return Response(res.dict) class PaySuccess(APIView): def get(self, request): print(request.META) return Response(111) def post(self, request, name): pay_obj = get_pay_obj_form_name(name) msg = 'failure' if pay_obj: logger.info(f"支付回调参数:{request.body}") logger.info(f"支付回调头部:{request.META}") if pay_obj.p_type == 'ALI': if pay_obj.valid_order(request): msg = 'success' return Response(msg) elif pay_obj.p_type == 'WX': if pay_obj.valid_order(request): return Response(msg) else: return Response(status=201) return Response(status=201, data=msg)