carwin_activity/apps/customer/order/views.py

# coding=utf-8

import traceback

from django.db import transaction

from rest_framework import generics
from rest_framework.exceptions import NotFound
from rest_framework.views import APIView
from rest_framework.viewsets import ModelViewSet
from rest_framework.decorators import action

from util import response_ok, response_error
from util.wx.wechat import WeChat
from util.permission import IsCustomerUser
from util.exceptions import CustomError

from apps.foundation.models import BizLog
from apps.customer.order.serializers import OrderSerializer, MemberOrderSerializer, ActivityOrderSerializer
from apps.activity.models import Activity, Order
from apps.pay.models import Pay
from apps.activity.filters import OrderFilter


class OrderViewSet(ModelViewSet):
    permission_classes = [IsCustomerUser, ]
    serializer_class = OrderSerializer

    def create(self, request, *args, **kwargs):
        try:
            with transaction.atomic():
                serializer = self.get_serializer(data=request.data)
                serializer.is_valid(raise_exception=True)
                serializer.save()
                instance = serializer.instance
                validated_data = serializer.validated_data

                if instance.amount == 0:
                    instance.status = Order.FINISH
                    BizLog.objects.addnew(request.customer.user, BizLog.INSERT, u"添加商品订单,id=%d" % instance.id, validated_data)
                else:
                    openid = request.POST.get('openid', None)
                    if not openid:
                        raise CustomError(u'未获取openid！')
                    appid = request.POST.get('appid', None)
                    WeChat.checkAppid(appid)
                    pay, query_string = Pay.wechatPay(instance.branch, instance.member, instance.amount, openid)
                    instance.pay = pay
                    BizLog.objects.addnew(request.customer.user, BizLog.INSERT, u'添加商品订单，id=%d' % instance.id, validated_data)
                    if query_string:
                        return response_ok(query_string)
        except CustomError as e:
            return response_error(e.get_error_msg())
        except Exception as e:
            traceback.print_exc()
            return response_error(str(e))
        return response_ok()

    @action(methods=['post'], detail=True)
    def pay_order(self, request, pk):
        openid = request.POST.get('openid')
        appid = request.POST.get('appid')
        try:
            if not openid:
                raise CustomError(u'未获取openid！')
            WeChat.checkAppid(appid)
            order = Order.objects.filter(id=pk).first()
            if not order:
                raise CustomError(u'未找到相应的订单')
            order.activity.checkStatus()
            # 要不要设置一个时间  时间过了订单不能再支付  提示订单已过期  请重新下单 然后把订单状态改成失败
            with transaction.atomic():
                data = order.payOrder(openid)
                BizLog.objects.addnew(request.customer.user, BizLog.INSERT, u'支付订单，id=%d' % order.id, request.data)
            return response_ok(data)
        except CustomError as e:
            return response_error(e.get_error_msg())
        except Exception as e:
            traceback.print_exc()
            return response_error(str(e))


class ActivityOrderListView(generics.ListAPIView):
    '''
    小程序显示已有人购买订单列表--  查询这个活动的订单记录(删除的也显示)
    '''
    queryset = Order.objects.filter(status=Order.FINISH)
    serializer_class = ActivityOrderSerializer

    def filter_queryset(self, queryset):
        activity_id = self.request.GET.get('activity_id')
        appid = self.request.GET.get('appid')
        activity = Activity.getById(activity_id)
        WeChat.checkAppid(appid)

        queryset = queryset.filter(activity=activity)
        f = OrderFilter(self.request.GET, queryset=queryset)
        return f.qs

    def list(self, request, *args, **kwargs):
        try:
            data = super(ActivityOrderListView, self).list(request)
        except NotFound:
            return response_ok([])
        return data


class OrderListView(generics.ListAPIView):
    '''
    小程序显示拼团订单--显示用户所有订单信息(删除的不显示)
    '''
    permission_classes = [IsCustomerUser, ]
    queryset = Order.objects.filter(delete=False)
    serializer_class = MemberOrderSerializer

    def filter_queryset(self, queryset):
        queryset = queryset.filter(member=self.request.customer)
        f = OrderFilter(self.request.GET, queryset=queryset)
        return f.qs