|
from django.shortcuts import render
from django_query import get_query_value
from django.db import transaction
from django_response import response
from django.shortcuts import HttpResponse
from django_logit import logit
from pywe_exception import WeChatPayException
from pywe_pay import WeChatPay
from pywe_pay_notify import check_pay_notify
from pywe_response import WXPAY_NOTIFY_FAIL, WXPAY_NOTIFY_SUCCESS
from TimeConvert import TimeConvert as tc
from django.conf import settings
from live.models import liveGoodsInfo, RoomGoodsInfo, RoomOrderInfo, RoomInfo, AnchorInfo
from account.models import UserInfo
from utils.error.errno_utils import OrderStatusCode, UserStatusCode
WECHAT = settings.WECHAT
def anchor_detail(request):
anchor_id = get_query_value(request, 'anchor_id', '')
try:
anchor = AnchorInfo.objects.get(anchor_id=anchor_id)
except:
return response()
rooms = RoomInfo.objects.filter(anchor_id=anchor_id)
rooms = [room.anchorData for room in rooms]
return response(200, 'Get Room Anchor Details Success', '获取主播详情成功', data={
'anchor': {
'anchor_id': anchor.anchor_id,
'anchor_name': anchor.anchor_name,
'anchor_avatar': anchor.anchor_avatar_url,
'anchor_intro': anchor.anchor_intro,
'banner': anchor.banner_url,
},
'rooms': rooms
})
def room_goods_detail(request):
goods_id = get_query_value(request, 'goods_id', '')
room_id = get_query_value(request, 'room_id', '')
# 校验直播间和商品
try:
room_goods_info = RoomGoodsInfo.objects.get(goods_id=goods_id, room_id=room_id)
except:
return response()
try:
goods_info = liveGoodsInfo.objects.get(goods_id=goods_id)
except:
return response()
return response(200, 'Get Room Goods Detail Success', '获取直播间商品成功', data={
'goods_info': {
'goods_img': goods_info.goods_img_url,
'name': goods_info.name,
'price_type': goods_info.price_type,
'price': goods_info.price,
'price2': goods_info.price2,
'inventory': room_goods_info.inventory,
},
'anchor_id': room_goods_info.anchor_id,
})
def room_anchor_details(request):
room_id = request.POST.get('room_id', '')
try:
room = RoomInfo.objects.get(room_id=room_id)
except:
return response()
try:
anchor = AnchorInfo.objects.get(anchor_id=room.anchor_id)
except:
return response()
rooms = RoomInfo.objects.filter(anchor_id=room.anchor_id)
rooms = [room.anchorData for room in rooms]
return response(200, 'Get Room Anchor Details Success', '获取主播详情成功', data={
'anchor': {
'anchor_id': anchor.anchor_id,
'anchor_name': anchor.anchor_name,
'anchor_avatar': anchor.anchor_avatar_url,
'anchor_intro': anchor.anchor_intro,
'banner': anchor.banner_url,
},
'rooms': rooms
})
@logit
@transaction.atomic
def live_order_create(request):
""" 订单创建 """
user_id = request.POST.get('user_id', '')
room_id = request.POST.get('room_id', '')
anchor_id = request.POST.get('anchor_id', '')
goods_id = request.POST.get('goods_id', '')
share_openid = request.POST.get('share_openid', '')
name = request.POST.get('name', '')
phone = request.POST.get('phone', '')
province = request.POST.get('province', '')
city = request.POST.get('city', '')
county = request.POST.get('county', '')
address = request.POST.get('address', '')
amount = int(request.POST.get('amount', 0))
total_fee = int(request.POST.get('total_fee', 0)) # 总金额，单位分
body = request.POST.get('body', '尖货直播') # 商品描述
# 用户校验
try:
user = UserInfo.objects.get(user_id=user_id, status=True)
user.consignee_name = name
user.consignee_phone = phone
user.consignee_province = province
user.consignee_city = city
user.consignee_county = county
user.consignee_address = address
user.save()
except UserInfo.DoesNotExist:
return response(UserStatusCode.USER_NOT_FOUND)
# 校验直播间和商品
try:
room_goods_info = RoomGoodsInfo.objects.get(goods_id=goods_id, room_id=room_id, anchor_id=anchor_id)
except:
return response(400001, 'Room Goods Not Found', description='直播间商品不存在')
# 金额校验
try:
goods_info = liveGoodsInfo.objects.get(goods_id=goods_id)
if amount * int(goods_info.price) != total_fee:
return response(404091, 'FEE Check Fail', description='金额校验失败')
except:
return response(404091, 'FEE Check Fail', description='金额校验失败')
# 消库存
room_goods_info.inventory -= 1
room_goods_info.save()
# JSAPI--公众号支付、NATIVE--原生扫码支付、APP--app支付，统一下单接口trade_type的传参可参考这里
trade_type = 'JSAPI'
# 根据 trade_type 获取 wechat 配置
wxcfg = WECHAT.get(trade_type, {})
# WeChatPay 初始化
wxpay = WeChatPay(wxcfg.get('appID'), wxcfg.get('apiKey'), wxcfg.get('mchID'))
# 生成订单
order = RoomOrderInfo.objects.create(
user_id=user_id,
room_id=room_id,
anchor_id=anchor_id,
goods_id=goods_id,
share_openid=share_openid,
amount=amount,
total_fee=total_fee,
trade_type=trade_type,
name=name,
phone=phone,
province=province,
city=city,
county=county,
address=address,
)
try:
prepay_data = wxpay.order.create(
body=body,
notify_url=settings.API_DOMAIN + 'live/pay/notify_url',
out_trade_no=order.order_id,
total_fee=total_fee,
trade_type=trade_type,
openid=user.openid, # 可选，用户在商户appid下的唯一标识。trade_type=JSAPI，此参数必传
)
except WeChatPayException as e:
order.unifiedorder_result = e.args
order.save()
return response(OrderStatusCode.UNIFIED_ORDER_FAIL)
prepay_id = prepay_data.get('prepay_id', '')
order.prepay_id = prepay_id
order.save()
wxpay_params = wxpay.jsapi.get_jsapi_params(prepay_id)
return response(200, 'Order Create Success', '订单创建成功', {
'order_id': order.order_id,
'prepay_id': prepay_id,
'wxpay_params': wxpay_params,
})
def live_order_cancel(request):
user_id = request.POST.get('user_id', '')
order_id = request.POST.get('order_id', '')
prepay_id = request.POST.get('prepay_id', '')
try:
order = RoomOrderInfo.objects.get(user_id=user_id, order_id=order_id, prepay_id=prepay_id)
except:
return response(400001, 'Order Not Found', description='直播间订单不存在')
if order.pay_status == RoomOrderInfo.FAIL:
return response(200, 'Order Cancel Success', '订单取消成功')
order.pay_status = RoomOrderInfo.FAIL
order.save()
try:
goods_info = RoomGoodsInfo.objects.get(room_id=order.room_id, goods_id=order.goods_id, anchor_id=order.anchor_id)
goods_info.inventory += order.amount
goods_info.save()
except:
return response(400001, 'Room Goods Not Found', description='直播间商品不存在')
return response(200, 'Order Cancel Success', '订单取消成功')
def order_paid_success(order):
if order.pay_status == RoomOrderInfo.PAID:
return
order.pay_status = RoomOrderInfo.PAID
order.paid_at = tc.utc_datetime()
order.save()
try:
goods_info = RoomGoodsInfo.objects.get(room_id=order.room_id, goods_id=order.goods_id, anchor_id=order.anchor_id)
goods_info.sale_infos += [order.order_id]
goods_info.save()
except:
return
def order_paid_fail(order):
if order.pay_status == RoomGoodsInfo.FAIL:
return
order.pay_status = RoomGoodsInfo.FAIL
order.save()
try:
goods_info = RoomGoodsInfo.objects.get(room_id=order.room_id, goods_id=order.goods_id, anchor_id=order.anchor_id)
goods_info.inventory += order.amount
goods_info.save()
except:
return
@logit
@transaction.atomic
def notify_url_api(request):
""" 支付异步通知回调地址 """
notify_data, success = check_pay_notify(request.body, wx_configs=settings.WECHAT)
if not success:
return HttpResponse(WXPAY_NOTIFY_FAIL)
order = RoomOrderInfo.objects.select_for_update().get(order_id=notify_data.get('out_trade_no', ''), status=True)
order.notify_msg = request.body
order.transaction_id = notify_data.get('transaction_id', '')
order.save()
result_code = notify_data.get('result_code', '')
if result_code == 'SUCCESS':
order_paid_success(order)
else:
order_paid_fail(order)
return HttpResponse(WXPAY_NOTIFY_SUCCESS)
|
