# -*- coding: utf-8 -*- # !/usr/bin/env python import datetime import logging import traceback from django.conf import settings from django.http import HttpResponse from typing import TYPE_CHECKING, Dict import simplejson as json from apilib.monetary import RMB from apps.web.common.errors import DuplicatedRefundRecordFound from apps.web.common.transaction.pay import PayNotifyAction, PayRecordPoller, OrderCacheMgr, PayNotifier, PayPullUp from apps.web.constant import PollRecordDefine from apps.web.core import ROLE from apps.web.core.payment import PaymentGateway, WithdrawGateway from apps.web.core.utils import async_operation, JsonOkResponse from apps.web.exceptions import UserServerException from apps.web.utils import get_alipay_gateway_result from library.alipay import AliValidationError, AliException, AliErrorCode from taskmanager.mediator import task_caller if TYPE_CHECKING: from apps.web.core.payment.type_checking import PaymentGatewayT from apps.web.core.payment.ali import AliPayGateway, AliPayWithdrawGateway from apps.web.common.transaction.pay import RechargeRecordT from typing import Dict from apps.web.common.models import WithdrawRecord, CapitalUser logger = logging.getLogger(__name__) def update_record(record, **payload): # type: (RechargeRecordT, Dict)->bool import arrow if 'send_pay_date' in payload: finished_time = arrow.get(payload['send_pay_date'], 'YYYY-MM-DD HH:mm:ss', tzinfo = settings.TIME_ZONE).naive elif 'gmt_payment' in payload: finished_time = arrow.get(payload['gmt_payment'], 'YYYY-MM-DD HH:mm:ss', tzinfo = settings.TIME_ZONE).naive else: finished_time = datetime.datetime.now() if 'notify_id' in payload: return record.succeed(wxOrderNo = payload['trade_no'], transactionId = payload['trade_no'], finishedTime = finished_time, **{ 'extraInfo.notify_id': payload.get('notify_id'), 'extraInfo.notify_time': payload.get('notify_time'), 'extraInfo.notify_type': payload.get('notify_type') }) else: return record.succeed(wxOrderNo = payload['trade_no'], transactionId = payload['trade_no'], finishedTime = finished_time, **{ 'extraInfo.notify_time': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') }) class AliPayRecordPoller(PayRecordPoller): def update_record(self, record, **payload): # type: (RechargeRecordT, Dict)->bool return update_record(record, **payload) def action_of_pay(self, payment_gateway, record): # type: (AliPayGateway, RechargeRecordT)->(PayNotifyAction, str) try: result = payment_gateway.api_trade_query(record.orderNo) trade_status = result['trade_status'] if trade_status == 'WAIT_BUYER_PAY': return PayNotifyAction.Unknown, result if trade_status == 'TRADE_CLOSED': return PayNotifyAction.NoHandle, result if trade_status in ['TRADE_SUCCESS', 'TRADE_FINISHED']: if 'trade_no' not in result or not result['trade_no']: return PayNotifyAction.Unknown, result else: return PayNotifyAction.NeedHandle, result return PayNotifyAction.Unknown, result except Exception as e: logger.exception(e) return PayNotifyAction.Unknown, {} class AliPayNotifier(PayNotifier): def __init__(self, request, record_cls_factory): super(AliPayNotifier, self).__init__(request = request, record_cls_factory = record_cls_factory) def parse_request(self, request): return request.POST.dict() # type: Dict @property def out_trade_no(self): return self.payload['out_trade_no'] def verify(self, gateway, record, payload): # type: (PaymentGatewayT, RechargeRecordT, Dict)->None notifier_fen = int(RMB(payload['total_amount']) * 100) if notifier_fen != record.fen_total_fee: raise AliValidationError( errmsg = u'total_amount校验失败', lvalue = record.fen_total_fee, rvalue = notifier_fen, client = gateway.client) def update_record(self, record, **payload): # type: (RechargeRecordT, Dict)->bool return update_record(record, **payload) @classmethod def reply(cls, msg): return HttpResponse(msg) def do(self, post_pay): # type: (callable)->HttpResponse try: payload = self.payload logger.info('received ali pay notify: %s' % str(payload)) order_no = payload['out_trade_no'] record = self.record_cls.get_record(order_no = order_no) # type: RechargeRecordT if record is None: logger.error( 'no such record. orderNo = {}'.format(order_no)) return HttpResponse('success') if record.is_success: logger.error('record has been finished. orderNo = {}'.format(order_no)) return HttpResponse('success') gateway = PaymentGateway.from_gateway_key( record.my_gateway, record.pay_gateway_key, record.pay_app_type) # type: AliPayGateway if payload['app_id'] != gateway.appid: raise AliValidationError( errmsg = u'appid校验失败', lvalue = gateway.appid, rvalue = payload['app_id'], client = gateway.client) if not gateway.check(payload): raise AliException( errCode = AliErrorCode.MY_ERROR_SIGNATURE, errMsg = u'签名错误', client = gateway.client) if 'payOpenId' in record.attachParas and record.attachParas['payOpenId'] and payload['buyer_id'] != \ record.attachParas['payOpenId']: raise AliValidationError( errmsg = u'buyer_id校验失败', lvalue = record.attachParas['payOpenId'], rvalue = payload['buyer_id'], client = gateway.client) if 'payOpenId' in record.extraInfo and record.extraInfo['payOpenId'] and payload['buyer_id'] != \ record.extraInfo['payOpenId']: raise AliValidationError( errmsg = u'buyer_id校验失败', lvalue = record.extraInfo['payOpenId'], rvalue = payload['buyer_id'], client = gateway.client) self.verify(gateway, record, payload) # 只有TRADE_SUCCESS和TRADE_FINISHED才是成功状态 if ('trade_status' not in payload) or (payload['trade_status'] not in ['TRADE_FINISHED', 'TRADE_SUCCESS']): return HttpResponse('UNFINISHED') order_cache_mgr = OrderCacheMgr(record) try: if order_cache_mgr.check_and_set_done(): logger.debug('%s has been done because cache in notify.' % repr(record)) return HttpResponse('success') except Exception as e: logger.error( 'cache key is not exist or exception. key = %s' % order_cache_mgr.key) modified = self.update_record(record, **payload) if not modified: logger.error('recharge record has done. orderNo = %s' % order_no) return HttpResponse('success') logger.info('%s successfully confirmed' % repr(record)) async_operation(post_pay, record = record) return HttpResponse('success') except Exception as e: logger.exception(e) return HttpResponse('ERROR') class AliPayWithdrawNotifier(object): def __init__(self, record_cls): self.record_cls = record_cls # type: WithdrawRecord def verify(self, gateway, record, payload): # type: (WithdrawGateway, WithdrawRecord, Dict)->None if str(record.actualPay) != str(payload['trans_amount']): raise AliValidationError( errmsg = u'total_amount校验失败', lvalue = str(record.actualPay), rvalue = str(payload['trans_amount']), client = gateway.client) @classmethod def reply(cls, msg): return HttpResponse(msg) def do(self, payload): # type: (dict)->HttpResponse try: order_no = payload['biz_content']['out_biz_no'] status = payload['biz_content']['status'] if status == 'DEALING': logger.debug('withdraw record is dealing. ignore. orderNo = {}'.format(order_no)) return self.reply('success') record = self.record_cls.get_record(order_no = order_no) # type: WithdrawRecord if record is None: logger.warn( 'no such record. orderNo = {}'.format(order_no)) return self.reply('success') if record.finished: logger.error('record has been finished. orderNo = {}'.format(order_no)) return self.reply('success') gateway = WithdrawGateway.from_withdraw_gateway_key( record.withdrawGatewayKey, record.extras.get('gateway_version', 'v1')) # type: AliPayWithdrawGateway if payload['app_id'] != gateway.appid: raise AliValidationError( errmsg = u'appid校验失败', lvalue = gateway.appid, rvalue = payload['app_id'], client = gateway.client) if not gateway.check(payload): raise AliException( errCode = AliErrorCode.MY_ERROR_SIGNATURE, errMsg = u'签名错误', client = gateway.client) self.verify(gateway, record, payload) order_cache_mgr = OrderCacheMgr(record) try: if order_cache_mgr.check_and_set_done(): logger.debug('%s has been done because cache in notify.' % repr(record)) return self.reply('success') except Exception as e: logger.error( 'cache key is not exist or exception. key = %s' % order_cache_mgr.key) logger.info('%s successfully confirmed' % repr(record)) payee = ROLE.from_role_id(record.role, record.ownerId) # type: CapitalUser if not payee: logger.warn('owner id is None. orderNo = {}'.format(order_no)) return self.reply('success') handler = payee.new_withdraw_handler(record) if status == 'SUCCESS': logger.debug('withdraw record is success. orderNo = {}'.format(order_no)) if 'pay_fund_order_id' not in payload['biz_content'] or not payload['biz_content']['pay_fund_order_id']: logger.error('no pay_fund_order_id. orderNo = {}'.format(order_no)) return self.reply('success') if 'order_id' not in payload['biz_content'] or not payload['biz_content']['order_id']: logger.error('no order_id. orderNo = {}'.format(order_no)) return self.reply('success') if 'pay_date' not in payload['biz_content'] or not payload['biz_content']['pay_date']: logger.error('has no pay_date. orderNo = {}'.format(order_no)) return self.reply('success') # SUCCESS有可能会变成REFUND. 把状态设置为银行处理, 如果两天后没有退票, 在认为转账成功 handler.bank_processing(**{ 'order_id': payload['biz_content']['order_id'], 'pay_fund_order_id': payload['biz_content']['pay_fund_order_id'], 'pay_date': payload['biz_content']['pay_date'] }) return self.reply('success') elif status == 'REFUND': try: handler.revoke(remarks = u'银行退单', description = u'银行退单') except DuplicatedRefundRecordFound as e: pass return self.reply('success') elif status == 'FAIL': error_code = payload.get('error_code', 'FAIL') fail_reason = payload.get('fail_reason', u'提现失败') error_msg = '{}({})'.format(fail_reason, error_code) handler.revoke(remarks = error_msg, description = error_msg) return self.reply('success') else: logger.warn('unsupport status = {}'.format(status)) return self.reply('success') except Exception as e: logger.exception(e) return self.reply('fail') class AlipayPullUp(PayPullUp): """ 支付宝支付获取支付参数。目前只作为资金池支付 """ def do(self): # type: ()->HttpResponse OrderCacheMgr(self.record).initial() try: result = get_alipay_gateway_result( gateway = self.payment_gateway, out_trade_no = self.record.orderNo, money = self.record.money, subject = self.record.subject, buyer_id = self.openId, body = json.dumps({'dealerId': self.record.ownerId}), notify_url = self.payload.get('notifyUrl')) except Exception as e: logger.exception(e) self.record.fail(description = u'拉起支付发生异常', extraInfo__traceback = traceback.format_exc()) raise UserServerException(u'拉起支付发生异常') else: if result['code'] == u'10000': task_caller('poll_user_recharge_record', delay = PollRecordDefine.DELAY_BEFORE, expires = PollRecordDefine.TASK_EXPIRES, pay_app_type = self.payment_gateway.pay_app_type, record_id = str(self.record.id), interval = PollRecordDefine.WAIT_EACH_ROUND, total_count = PollRecordDefine.TOTAL_ROUNDS) return JsonOkResponse( payload = { 'tradeNO': result['trade_no'], 'outTradeNo': self.record.orderNo, 'adShow': self.payload.get('showAd', False) }) else: logger.error('{openid} is recharging via({pay_app_type}) with fault. result = {result}'.format( openid = self.openId, pay_app_type = self.payment_gateway.pay_app_type, result = str(result))) self.record.fail(description = result['code']) raise UserServerException(u'系统错误，请重试')