carwin_activity/apps/api/views.py

# coding=utf-8

import traceback

from django.db import transaction
from django.http import HttpResponse
from django.conf import settings

from rest_framework.views import APIView

from util.exceptions import CustomError
from util.wechatpay import WechatPayNotify
from util.wx.wechat import WeChat

from apps.pay.models import Pay
from apps.foundation.models import BizLog


class WechatNotifyView(APIView):

    def dispatch(self, request, *args, **kwargs):
        param = request.body
        appid = kwargs['appid']
        # param = request.body.decode('utf-8')
        notify = WechatPayNotify(param, settings.AGENT_KEY)
        try:
            WeChat.checkAppid(appid)
            data = notify.handle()
            if not data:
                raise CustomError(u'错误的请求！')

            result_code = data['result_code']
            if result_code != 'SUCCESS':
                raise CustomError(u'错误的请求！')

            no = data['out_trade_no']
            amount = float(data['total_fee']) / 100.0

            with transaction.atomic():
                pay = Pay.getByNo(no)
                pay.payConfirm(amount)
                BizLog.objects.addnew(pay.customer.user, BizLog.INSERT, u'微信支付成功，no=%s' % no, param)
        except Exception as e:
            traceback.print_exc()
            return HttpResponse(WechatPayNotify.response_fail())
        return HttpResponse(WechatPayNotify.response_ok())