def get_pay():
dbg('get_pay')
reply, status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
pay_id = data['pay_id']
except:
print_exception_info()
raise ApiError('get pay error', error.ERROR_PARAM)
pay = dbapi.get_pay(pay_id=pay_id)
if not pay:
raise ApiError('get pay error', error.ERROR_PAY_NOT_FOUND)
data = {
'trade_no': pay.trade_no,
'title': pay.title,
'body': pay.body,
'total_fee': pay.total_fee,
'time': int(pay.utime.timestamp()),
'status': pay.status
}
reply['data'] = data
return make_response(jsonify(reply), status_code)
def device_multi_add_coupon():
dbg('device_multi_add_coupon')
reply ,status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
device_ids = data['device_ids']
device_ids = list(set(device_ids))
coupon_id = data['coupon_id']
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
coupon = dbapi.get_coupon(id=coupon_id)
if not coupon:
raise ApiError('ERROR_COUPON_NOT_FOUND', error.ERROR_COUPON_NOT_FOUND)
for device_id in device_ids:
device = dbapi.get_device(id=device_id)
if not device:
db.session.rollback()
dbg('device_id: %s not exists' % device_id)
reply['code'] = 999
reply['msg'] = '自编号: %s 不存在' % device_id
status_code = 422
return make_response(jsonify(reply), status_code)
device_coupon = dbapi.get_device_coupon(device_id=device_id, coupon_id=coupon_id)
if device_coupon:
db.session.rollback()
dbg('device_id: %d added before' % device_id)
reply['code'] = 999
reply['msg'] = '自编号: %s 已添加该优惠券' % device_id
status_code = 422
return make_response(jsonify(reply), status_code)
else:
# 建立设备和优惠券之间的联系
device_coupon = dbapi.make_new_device_coupon(coupon_id, device_id)
# 更新pay表中的imei，用户查询订单收益
pay = dbapi.get_pay(pay_id=coupon.pay_id)
pay = dbapi.update_pay(pay, imei=device.imei)
try:
db.session.commit()
except:
db.session.rollback()
raise
return make_response(jsonify(reply), status_code)
def get_coupon():
dbg('get_coupon')
reply ,status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
advertiser_id = data.get('advertiser_id')
device_id = data.get('device_id')
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
page = request.args.get('page', 1, type=int)
psize = request.args.get('psize', 10, type=int)
data = []
if advertiser_id:
count, coupons = dbapi.get_coupon(advertiser_id=int(advertiser_id), page=page, psize=psize)
elif device_id:
count, coupons = dbapi.get_coupon(device_id=int(device_id), page=page, psize=psize)
else:
agent = dbapi.get_agent(user_id=current_user.id)
if not agent:
raise ApiError('ERROR_AGENT_NOT_FOUND', error.ERROR_AGENT_NOT_FOUND)
agent_id = agent.id
count, coupons = dbapi.get_coupon(agent_id=agent_id, page=page, psize=psize)
if count:
for coupon in coupons:
product = dbapi.get_product(product_id=coupon.product_id)
pay = dbapi.get_pay(pay_id=coupon.pay_id)
info = {
'id': coupon.id,
'title': coupon.title,
'desc': coupon.desc,
'img': url_for('static', filename=coupon.img),
'total': coupon.total,
'left': product.inventory,
'pay_status': pay.status,
'time': int(coupon.ctime.timestamp())
}
data.append(info)
reply['data'] = {
'count': count,
'coupons': data
}
return make_response(jsonify(reply), status_code)
def manual_refund():
dbg('manual_refund')
reply ,status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
pay_id = data['pay_id']
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
pay = dbapi.get_pay(pay_id=pay_id)
if not pay:
raise ApiError('ERROR_PAY_NOT_FOUND', error.ERROR_PAY_NOT_FOUND)
# 充值套餐、广告优惠券，不支持退款
not_support_list = (99, 199)
if pay.cat in not_support_list:
raise ApiError('not support this', error.ERROR_AGENT_NO_PERMISSION)
imei = pay.imei
device = dbapi.get_device(imei=imei)
if not device:
raise ApiError('ERROR_DEVICE_NOT_FOUND', error.ERROR_DEVICE_NOT_FOUND)
if device.owner_agent_id != current_user.agent_id:
raise ApiError('ERROR_AGENT_NO_PERMISSION', error.ERROR_AGENT_NO_PERMISSION)
refund_available = check_refund_available(device, pay.total_fee)
if not refund_available:
raise ApiError('ERROR_REFUND_NOT_AVAILABLE', error.ERROR_REFUND_NOT_AVAILABLE)
if pay.status != 1:
raise ApiError('ERROR_NOT_PAY', error.ERROR_NOT_PAY)
product = dbapi.get_product(product_id=pay.product_id)
if not product:
raise ApiError('can not find product: %s' % pay.product_id)
# 退款
pay_refund(pay, product)
db.session.commit()
return make_response(jsonify(reply), status_code)
def pay_coupon():
dbg('pay_coupon')
reply ,status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
coupon_id = data['coupon_id']
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
coupon = dbapi.get_coupon(id=coupon_id)
if not coupon:
raise ApiError('ERROR_COUPON_NOT_FOUND', error.ERROR_COUPON_NOT_FOUND)
pay = dbapi.get_pay(pay_id=coupon.pay_id)
if not pay:
raise ApiError('ERROR_PAY_NOT_FOUND', error.ERROR_PAY_NOT_FOUND)
if pay.status != 0 or coupon.paid:
raise ApiError('ERROR_ALREADY_PAID', error.ERROR_ALREADY_PAID)
if int(datetime.now().timestamp()) - int(pay.utime.timestamp()) >= 2*60*60:
product = dbapi.get_product(product_id=coupon.product_id)
trade_no = pay.trade_no
pay = wechat_qrcode_update(pay, product, trade_no, current_user.id)
reply['data'] = {
'pay_request': {
'prepay_id': pay.prepay_id,
'qrcode': pay.qrcode
},
'trade_no' : pay.trade_no,
'id' : pay.id
}
return make_response(jsonify(reply), status_code)
def machine_start():
dbg('machine_start')
reply, status_code = {'code': 0, 'msg': ''}, 200
try:
data = request.get_json()
cat = data['cat']
pay_id = data['pay_id']
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
pay = dbapi.get_pay(pay_id=pay_id)
if not pay:
raise ApiError('machine start error', error.ERROR_PAY_NOT_FOUND)
imei = pay.imei
if pay.status != 1:
raise ApiError('machine start error', error.ERROR_NOT_PAY)
product = dbapi.get_product(product_id=pay.product_id)
if not product:
raise ApiError('can not find product: %s' % pay.product_id)
if product.cat == 99:
raise ApiError('ERROR_PAY_NOT_FOUND', error.ERROR_PAY_NOT_FOUND)
if product.cat == 0:
# 倒计时限制同时支付
last_pay = dbapi.get_last_pay(pay.imei, not_include_pay_id=pay_id)
if last_pay and datetime.now().timestamp() - last_pay.utime.timestamp(
) < 5:
# 退款
pay_refund(pay, product)
raise ApiError('ERROR_PAY_TOO_FREQUENCE',
error.ERROR_PAY_TOO_FREQUENCE)
dbg('cat: %s' % cat)
device = dbapi.get_device(imei=imei)
if not device:
raise ApiError('ERROR_DEVICE_NOT_FOUND', error.ERROR_DEVICE_NOT_FOUND)
value = product.value
record = dbapi.get_record(pay_id=pay.id)
if record:
raise ApiError('machine start error', error.ERROR_ALREADY_START)
high = device.high
low = device.low
cat = product.cat
dbg('imei: %s, value: %d, cat: %d, high: %d, low: %d' %
(imei, value, cat, low, high))
cat = product.cat
if cat in (0, 2, 3):
device_type = device.cat
succ, result = launch_relay_signal_deivce_v2(imei, value, device_type)
elif cat == 1:
if imei == '868575023189139' and value == 3:
succ, result = launch_pulse_signal_deivce(imei, 2, high, low)
time.sleep(2)
succ, result = launch_pulse_signal_deivce(imei, 1, high, low)
else:
succ, result = launch_pulse_signal_deivce(imei, value, high, low)
else:
raise ApiError('not support')
if not succ:
# 退款
pay_refund(pay, product)
try:
data = json.loads(result)
dbg((data['code'], data['msg']))
reply['code'] = data['code']
reply['msg'] = data['msg']
status_code = 422
except:
print_exception_info()
reply['code'] = 99
reply['msg'] = 'server error'
status_code = 500
return make_response(jsonify(reply), status_code)
if cat == 0:
# 按摩椅需要记录开始记录
pay_way = pay.pay_mode
stime = datetime.now()
etime = stime + timedelta(seconds=product.value)
record = dbapi.make_new_record(pay.id, pay_way, pay.user_id,
product.agent_id, pay.product_id, stime,
etime)
else:
# 投币器也需要记录开始记录
pay_way = pay.pay_mode
stime = datetime.now()
etime = stime
record = dbapi.make_new_record(pay.id, pay_way, pay.user_id,
product.agent_id, pay.product_id, stime,
etime)
db.session.commit()
reply['data'] = {
'id': record.id,
'stime': int(record.stime.timestamp()),
'etime': int(record.etime.timestamp()),
'now': int(time.time()),
'user_id': record.user_id,
}
return make_response(jsonify(reply), status_code)
def ali_notify():
dbg('ali_notify')
reply = {"return_code": 'SUCCESS', "return_msg": ""}
try:
trade_no = request.form['out_trade_no']
nofity_res = {}
for k in request.form:
print(k, request.form[k])
nofity_res[k] = request.form[k]
trade_status = request.form['trade_status']
ali_trade_no = request.form['trade_no']
buyer_id = request.form['buyer_id']
buyer_logon_id = request.form['buyer_logon_id']
cash_fee = request.form.get(
'buyer_pay_amount') or request.form['total_amount']
except:
print_exception_info()
raise ApiError('ERROR_PARAM', error.ERROR_PARAM)
# 检查订单
pay = dbapi.get_pay(trade_no=trade_no)
if not pay:
raise ApiError('can not find pay: %s' % trade_no)
# 签名
# sign = request.form['sign']
# ali_config = dbapi.get_ali_config(agent_id=int(agent_id))
# key_path = ali_config.pub_path
# 如果已经支付，则直接返回Ok
if pay.status > 0:
return make_response('success')
if trade_status == 'TRADE_CLOSED':
dbapi.update_pay(pay,
user_id=pay.user_id,
status=2,
nofity_res=json.dumps(nofity_res))
try:
db.session.commit()
except:
print_exception_info()
db.session.rollback()
raise ApiError('ali_notify error')
elif trade_status == 'TRADE_SUCCESS':
dbapi.update_pay(pay=pay,
user_id=pay.user_id,
status=1,
cash_fee=cash_fee,
nofity_res=json.dumps(nofity_res),
ali_trade_no=ali_trade_no)
product = dbapi.get_product(product_id=pay.product_id)
if product.cat == 99:
# 钱包充值
user_id = pay.user_id
wallet = dbapi.get_wallet(role=0,
user_id=user_id,
agent_id=pay.agent_id)
wallet_id = wallet.id
trade_type = dbapi.WALLET_TRADE_TYPE_DEPOSIT
receipt = product.value
withdrawable_receipt = 0
remark = product.title
wallet_receipt = dbapi.make_new_wallet_receipt(
user_id, pay.agent_id, wallet_id, trade_type, receipt,
withdrawable_receipt, remark)
balance = wallet.balance + receipt
wallet = dbapi.update_wallet(wallet, balance=balance)
wechat_pay_receipt(pay)
try:
db.session.commit()
except:
print_exception_info()
db.session.rollback()
raise ApiError('ali_notify error', error.ERROR_DATABASE_COMMIT)
return make_response('success')
def pay_notify():
dbg('pay_notify')
reply = {"return_code": 'SUCCESS', "return_msg": ""}
try:
data_info = request.data.decode('utf8')
data = wechatpay.xml_to_dict(data_info)
nofity_res = json.dumps(data)
dbg('data_info: %s' % data_info)
return_code = None
result_code = None
if "result_code" in data and "return_code" in data:
return_code = data["return_code"]
result_code = data["result_code"]
if return_code != "SUCCESS":
raise ApiError('wechat interface return error')
# return_code为`SUCCESS`, 则包含out_trade_no
# 检查订单
trade_no = data['out_trade_no']
pay = dbapi.get_pay(trade_no=trade_no)
if not pay:
raise ApiError('can not find pay: %s' % trade_no)
# 如果已经支付，则直接返回Ok
if pay.status > 0:
return make_response(wechatpay.dict_to_xml(reply).encode('utf-8'))
# 检查签名
sign = data['sign']
data.pop('sign')
wechat_config = dbapi.get_wechat_config(agent_id=pay.agent_id)
key = wechat_config.mchkey
local_sign = wechatpay.params_sign(data, key)
data['sign'] = sign
dbg((sign, local_sign))
if sign != local_sign:
raise ApiError('sign error')
if result_code != "SUCCESS":
'''
``` 支付失败, 更新支付异常
'''
dbapi.update_pay(pay, status=2, nofity_res=nofity_res)
try:
db.session.commit()
except:
print_exception_info()
db.session.rollback()
raise ApiError('pay_notify error')
else:
'''
``` 支付成功, 并更新支付成功
'''
cash_fee = data['cash_fee']
dbapi.update_pay(pay=pay,
status=1,
cash_fee=cash_fee,
nofity_res=nofity_res)
product = dbapi.get_product(product_id=pay.product_id)
if product.cat == 99:
# 钱包充值
user_id = pay.user_id
wallet = dbapi.get_wallet(role=0,
user_id=user_id,
agent_id=pay.agent_id)
wallet_id = wallet.id
trade_type = dbapi.WALLET_TRADE_TYPE_DEPOSIT
receipt = product.value
withdrawable_receipt = 0
remark = product.title
wallet_receipt = dbapi.make_new_wallet_receipt(
user_id, pay.agent_id, wallet_id, trade_type, receipt,
withdrawable_receipt, remark)
balance = wallet.balance + receipt
wallet = dbapi.update_wallet(wallet, balance=balance)
if product.cat == dbapi.PRODUCT_CAT_COUPON:
# 优惠券
coupon = dbapi.get_coupon(pay_id=pay.id)
if coupon:
coupon = dbapi.update_coupon(coupon, paid=1)
wechat_pay_receipt(pay)
try:
db.session.commit()
except:
print_exception_info()
db.session.rollback()
raise ApiError('wechat_pay_notify error',
error.ERROR_DATABASE_COMMIT)
except ApiError as e:
# if e.msg != 'wechat pay failed':
print_exception_info()
reply['return_code'] = "FAIL"
reply['return_msg'] = e.msg
except:
print_exception_info()
reply['return_code'] = "FAIL"
reply['return_msg'] = 'server error'
res_data = wechatpay.dict_to_xml(reply).encode('utf-8')
print(res_data)
return make_response(res_data)