拍爱

views.py 11KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335
  1. # -*- coding: utf-8 -*-
  2. import xmltodict
  3. from django.conf import settings
  4. from django.db import transaction
  5. from django.shortcuts import HttpResponse
  6. from logit import logit
  7. from paginator import pagination
  8. from pywe_response import WXPAY_NOTIFY_FAIL, WXPAY_NOTIFY_SUCCESS
  9. from TimeConvert import TimeConvert as tc
  10. from wechatpy import WeChatPay, WeChatPayException
  11. from account.models import UserIncomeExpensesInfo, UserInfo
  12. from group.models import GroupPhotoInfo, GroupPhotoOrderInfo
  13. from pay.models import OrderInfo
  14. from photo.models import PhotosInfo
  15. from utils.error.errno_utils import (GroupPhotoStatusCode, OrderStatusCode, UserStatusCode, WechatStatusCode,
  16. WithdrawStatusCode)
  17. from utils.error.response_utils import response
  18. from utils.redis.rbrief import set_brief_info
  19. from utils.redis.rorder import set_lensman_order_record
  20. WECHAT = settings.WECHAT
  21. @logit
  22. @transaction.atomic
  23. def wx_order_create_api(request):
  24. """ 订单创建 """
  25. group_id = request.POST.get('group_id', '')
  26. user_id = request.POST.get('user_id', '')
  27. photo_id = request.POST.get('photo_id', '')
  28. photo_type = request.POST.get('photo_type', 'nomark') # nomark for 去除水印, origin for 获取高清图
  29. photo_type_ = OrderInfo.ORIGIN if photo_type == 'origin' else OrderInfo.NOMARK
  30. # 群组照片校验
  31. try:
  32. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  33. except GroupPhotoInfo.DoesNotExist:
  34. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  35. # 判断是否重复购买
  36. if OrderInfo.objects.filter(photo_id=photo_id, photo_type=photo_type_, from_uid=user_id, pay_status=OrderInfo.PAID).exists():
  37. return response(OrderStatusCode.WX_ORDER_PAID_ALREADY_EXISTS)
  38. body = request.POST.get('body', '') # 商品描述
  39. total_fee = int(request.POST.get('total_fee', 0)) # 总金额,单位分
  40. # 金额校验
  41. # if int(r.get(LENSMAN_PHOTO_PRICE % (user_id, photo_id, photo_type)) or 0) != total_fee:
  42. # return response(OrderStatusCode.FEE_CHECK_FAIL)
  43. # 获取 from_uid, to_uid
  44. from_uid = user_id
  45. to_uid = group_photo.lensman_id or group_photo.user_id
  46. # JSAPI--公众号支付、NATIVE--原生扫码支付、APP--app支付,统一下单接口trade_type的传参可参考这里
  47. trade_type = request.POST.get('trade_type', '')
  48. # 根据 trade_type 获取 wechat 配置
  49. wechat = WECHAT.get(trade_type, {})
  50. # WeChatPay 初始化
  51. wxpay = WeChatPay(wechat.get('appID'), wechat.get('apiKey'), wechat.get('mchID'))
  52. # 生成订单
  53. order = OrderInfo.objects.create(
  54. group_id=group_id,
  55. photo_id=photo_id,
  56. lensman_photo_id=group_photo.lensman_photo_id,
  57. photo_type=1 if photo_type == 'origin' else 0,
  58. from_uid=from_uid,
  59. to_uid=to_uid,
  60. session_id=group_photo.session_id,
  61. total_fee=total_fee,
  62. trade_type=trade_type,
  63. )
  64. try:
  65. prepay_data = wxpay.order.create(
  66. body=body,
  67. notify_url=settings.API_DOMAIN + '/wx/notify_url',
  68. out_trade_no=order.order_id,
  69. total_fee=total_fee,
  70. trade_type=trade_type,
  71. # user_id=None, # 可选,用户在商户appid下的唯一标识。trade_type=JSAPI,此参数必传
  72. )
  73. except WeChatPayException:
  74. return response(OrderStatusCode.WX_UNIFIED_ORDER_FAIL)
  75. prepay_id = prepay_data.get('prepay_id', '')
  76. if trade_type == 'JSAPI':
  77. wxpay_params = wxpay.jsapi.get_jsapi_params(prepay_id)
  78. elif trade_type == 'APP':
  79. wxpay_params = wxpay.order.get_appapi_params(prepay_id)
  80. return response(200, 'Order Create Success', u'订单创建成功', {
  81. 'order_id': order.order_id,
  82. 'prepay_id': prepay_id,
  83. 'wxpay_params': wxpay_params,
  84. })
  85. def order_paid_success(order):
  86. if order.pay_status == OrderInfo.PAID:
  87. return
  88. if order.photo_type == OrderInfo.NOMARK:
  89. order.photo_status = OrderInfo.FETCHED
  90. order.pay_status = OrderInfo.PAID
  91. order.paid_at = tc.utc_datetime()
  92. order.save()
  93. porder, created = GroupPhotoOrderInfo.objects.select_for_update().get_or_create(
  94. group_id=order.group_id,
  95. session_id=order.session_id,
  96. user_id=order.from_uid,
  97. photo_id=order.photo_id,
  98. lensman_photo_id=order.lensman_photo_id,
  99. )
  100. photo = PhotosInfo.objects.get(
  101. photo_id=order.lensman_photo_id,
  102. )
  103. if order.photo_type == OrderInfo.NOMARK:
  104. porder.m_photo_path = photo.m_photo_path
  105. elif order.photo_type == OrderInfo.ORIGIN:
  106. porder.r_photo_path = photo.r_photo_path
  107. porder.save()
  108. set_lensman_order_record(porder)
  109. to_uid = order.to_uid
  110. total_fee = order.total_fee
  111. try:
  112. user = UserInfo.objects.select_for_update().get(user_id=to_uid)
  113. except UserInfo.DoesNotExist:
  114. return
  115. if order.photo_type == OrderInfo.NOMARK:
  116. # 余额增加
  117. amount, freeze_income_amount = total_fee, 0
  118. user.balance += amount
  119. # Redis 数值更新
  120. set_brief_info(to_uid, order.photo_type, total_fee)
  121. # 余额记录
  122. UserIncomeExpensesInfo.objects.create(
  123. user_id=to_uid,
  124. photo_id=order.photo_id,
  125. type=UserIncomeExpensesInfo.INCOME,
  126. amount=amount,
  127. balance=user.balance,
  128. freeze_income_amount=freeze_income_amount,
  129. freeze_income_balance=user.freeze_income_balance,
  130. remark=u'图片购买',
  131. )
  132. elif order.photo_type == OrderInfo.ORIGIN:
  133. amount, freeze_income_amount = 0, total_fee
  134. user.freeze_income_balance += freeze_income_amount
  135. user.save()
  136. def order_paid_fail(order):
  137. if order.pay_status == OrderInfo.FAIL:
  138. return
  139. order.pay_status = OrderInfo.FAIL
  140. order.save()
  141. @logit
  142. @transaction.atomic
  143. def wx_order_query_api(request):
  144. """ 订单查询 """
  145. order_id = request.POST.get('order_id', '')
  146. transaction_id = request.POST.get('transaction_id', '')
  147. try:
  148. order = OrderInfo.objects.select_for_update().get(order_id=order_id)
  149. except OrderInfo.DoesNotExist:
  150. return response(OrderStatusCode.WX_ORDER_NOT_FOUND)
  151. if order.pay_status == OrderInfo.PAID:
  152. return response(200, 'Order Pay Success', u'订单支付成功')
  153. elif order.pay_status == OrderInfo.FAIL:
  154. return response(OrderStatusCode.WX_ORDER_PAY_FAIL)
  155. # 根据 trade_type 获取 wechat 配置
  156. wechat = WECHAT.get(order.trade_type, {})
  157. # WeChatPay 初始化
  158. wxpay = WeChatPay(wechat.get('appID'), wechat.get('apiKey'), wechat.get('mchID'))
  159. # 订单查询
  160. query_data = wxpay.order.query(transaction_id, order_id)
  161. # 签名校验
  162. if not wxpay.check_signature(query_data):
  163. return response(OrderStatusCode.SIGN_CHECK_FAIL)
  164. # 交易状态
  165. trade_state = query_data.get('trade_state')
  166. # 订单状态判断更新
  167. if trade_state == 'SUCCESS': # 订单支付成功
  168. order_paid_success(order)
  169. return response(200, 'Order Pay Success', u'订单支付成功')
  170. elif trade_state == 'NOTPAY': # 订单未支付
  171. return response(OrderStatusCode.WX_ORDER_NOT_PAY)
  172. elif trade_state == 'USERPAYING': # 订单支付中
  173. return response(OrderStatusCode.WX_ORDER_PAYING)
  174. else: # 订单支付失败
  175. order_paid_fail(order)
  176. return response(OrderStatusCode.WX_ORDER_PAY_FAIL)
  177. @logit
  178. @transaction.atomic
  179. def wx_order_list_api(request):
  180. """ 订单列表 """
  181. user_id = request.POST.get('user_id', '')
  182. page = int(request.POST.get('page', 1))
  183. num = int(request.POST.get('num', settings.ORDER_NUM_PER_PAGE))
  184. orders = OrderInfo.objects.filter(from_uid=user_id, pay_status=OrderInfo.PAID, status=True).order_by('-pk')
  185. orders, left = pagination(orders, page, num)
  186. orders = [order.data(user_id) for order in orders]
  187. return response(200, 'Get Order List Success', u'获取订单列表成功', {
  188. 'orders': orders,
  189. 'left': left,
  190. })
  191. @logit
  192. @transaction.atomic
  193. def wx_order_detail_api(request):
  194. """ 订单详情 """
  195. user_id = request.POST.get('user_id', '')
  196. order_id = request.POST.get('order_id', '')
  197. try:
  198. order = OrderInfo.objects.get(order_id=order_id)
  199. except OrderInfo.DoesNotExist:
  200. return response(OrderStatusCode.WX_ORDER_NOT_FOUND)
  201. if user_id not in [order.from_uid, order.to_uid]:
  202. return response(OrderStatusCode.NO_DETAIL_PERMISSION)
  203. return response(200, 'Get Order Detail Success', u'获取订单详情成功', order.data(user_id))
  204. @logit
  205. @transaction.atomic
  206. def wx_notify_url_api(request):
  207. """ 支付异步通知回调地址 """
  208. try:
  209. data = xmltodict.parse(request.body)['xml']
  210. except xmltodict.ParsingInterrupted:
  211. # 解析 XML 失败
  212. return HttpResponse(WXPAY_NOTIFY_FAIL)
  213. trade_type = data.get('trade_type', '')
  214. # 根据 trade_type 获取 wechat 配置
  215. wechat = WECHAT.get(trade_type, {})
  216. # WeChatPay 初始化
  217. wxpay = WeChatPay(wechat.get('appID'), wechat.get('apiKey'), wechat.get('mchID'))
  218. # 签名校验
  219. if not wxpay.check_signature(data):
  220. return response(OrderStatusCode.SIGN_CHECK_FAIL)
  221. out_trade_no = data.get('out_trade_no', '')
  222. return_code = data.get('return_code', '')
  223. result_code = data.get('result_code', '')
  224. if return_code != 'SUCCESS' or result_code != 'SUCCESS':
  225. return HttpResponse(WXPAY_NOTIFY_FAIL)
  226. try:
  227. order = OrderInfo.objects.get(order_id=out_trade_no)
  228. except OrderInfo.DoesNotExist:
  229. return response(OrderStatusCode.WX_ORDER_NOT_FOUND)
  230. order_paid_success(order)
  231. return HttpResponse(WXPAY_NOTIFY_SUCCESS)
  232. @logit
  233. @transaction.atomic
  234. def wx_balance_withdraw_api(request):
  235. user_id = request.POST.get('user_id', '')
  236. # 用户校验
  237. try:
  238. user = UserInfo.objects.select_for_update().get(user_id=user_id)
  239. except UserInfo.DoesNotExist:
  240. return response(UserStatusCode.USER_NOT_FOUND)
  241. # JSAPI--公众号支付、NATIVE--原生扫码支付、APP--app支付,统一下单接口trade_type的传参可参考这里
  242. trade_type = request.POST.get('trade_type', '')
  243. # TRANSFER--企业付款、PACKET--现金红包, 余额提现接口withdraw_type的传参可参考这里
  244. withdraw_type = request.POST.get('withdraw_type', 'TRANSFER')
  245. amount = int(request.POST.get('amount', 0))
  246. if not user.openid:
  247. return response(WechatStatusCode.OPENID_NOT_FOUND)
  248. if user.balance < amount:
  249. return response(WithdrawStatusCode.BALANCE_NOT_ENOUGH)
  250. # 根据 trade_type 获取 wechat 配置
  251. wechat = WECHAT.get(trade_type, {})
  252. # WeChatPay 初始化
  253. wxpay = WeChatPay(wechat.get('appID'), wechat.get('apiKey'), wechat.get('mchID'), mch_cert=wechat.get('mch_cert'), mch_key=wechat.get('mch_key'))
  254. if withdraw_type == 'TRANSFER':
  255. ret_data = wxpay.transfer.transfer(user.openid, amount, u'摄影师余额提现,企业付款', check_name='NO_CHECK')
  256. elif withdraw_type == 'PACKET':
  257. ret_data = wxpay.redpack.send(
  258. user.openid,
  259. amount,
  260. send_name=wechat.get('redpacket', {}).get('SEND_NAME'),
  261. nick_name=wechat.get('redpacket', {}).get('NICK_NAME'),
  262. act_name=wechat.get('redpacket', {}).get('ACT_NAME'),
  263. wishing=wechat.get('redpacket', {}).get('WISHING'),
  264. remark=wechat.get('redpacket', {}).get('REMARK'),
  265. )
  266. # 根据 ret_data 判断是否提现成功, 成功则减余额, 失败则提示
  267. user.balance -= amount
  268. user.save()
  269. return response(200, 'Withdraw Success', u'提现成功', {})