python学习
This commit is contained in:
@@ -50,7 +50,6 @@ def iPagination( params ):
|
||||
ret['range'] = range( ret['from'],ret['end'] + 1 )
|
||||
return ret
|
||||
|
||||
|
||||
'''
|
||||
统一渲染方法
|
||||
'''
|
||||
@@ -108,7 +107,7 @@ def selectFilterObj( obj,field ):
|
||||
return ret
|
||||
|
||||
|
||||
def getDictFilterField( db_model,select_filed,key_field,id_list ):
|
||||
def getDictListFilterField( db_model,select_filed,key_field,id_list ):
|
||||
ret = {}
|
||||
query = db_model.query
|
||||
if id_list and len( id_list ) > 0:
|
||||
|
||||
35
common/libs/member/CartService.py
Normal file
35
common/libs/member/CartService.py
Normal file
@@ -0,0 +1,35 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import hashlib,requests,random,string,json
|
||||
from application import app,db
|
||||
from common.models.member.MemberCart import MemberCart
|
||||
from common.libs.Helper import getCurrentDate
|
||||
class CartService():
|
||||
|
||||
@staticmethod
|
||||
def deleteItem( member_id = 0,items = None ):
|
||||
if member_id < 1 or not items:
|
||||
return False
|
||||
for item in items:
|
||||
MemberCart.query.filter_by( food_id = item['id'],member_id = member_id ).delete()
|
||||
db.session.commit()
|
||||
return True
|
||||
|
||||
@staticmethod
|
||||
def setItems( member_id = 0,food_id = 0,number = 0 ):
|
||||
if member_id < 1 or food_id < 1 or number < 1:
|
||||
return False
|
||||
cart_info = MemberCart.query.filter_by( food_id = food_id, member_id= member_id ).first()
|
||||
if cart_info:
|
||||
model_cart = cart_info
|
||||
else:
|
||||
model_cart = MemberCart()
|
||||
model_cart.member_id = member_id
|
||||
model_cart.created_time = getCurrentDate()
|
||||
|
||||
model_cart.food_id = food_id
|
||||
model_cart.quantity = number
|
||||
model_cart.updated_time = getCurrentDate()
|
||||
db.session.add(model_cart)
|
||||
db.session.commit()
|
||||
return True
|
||||
|
||||
191
common/libs/pay/PayService.py
Normal file
191
common/libs/pay/PayService.py
Normal file
@@ -0,0 +1,191 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import hashlib,time,random,decimal,json
|
||||
from application import app,db
|
||||
from common.models.food.Food import Food
|
||||
from common.models.food.FoodSaleChangeLog import FoodSaleChangeLog
|
||||
from common.models.pay.PayOrder import PayOrder
|
||||
from common.models.pay.PayOrderItem import PayOrderItem
|
||||
from common.models.pay.PayOrderCallbackData import PayOrderCallbackData
|
||||
from common.libs.Helper import getCurrentDate
|
||||
from common.libs.queue.QueueService import QueueService
|
||||
from common.libs.food.FoodService import FoodService
|
||||
class PayService():
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def createOrder(self,member_id,items = None,params = None):
|
||||
resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
|
||||
pay_price = decimal.Decimal( 0.00 )
|
||||
continue_cnt = 0
|
||||
food_ids = []
|
||||
for item in items:
|
||||
if decimal.Decimal( item['price'] ) < 0 :
|
||||
continue_cnt += 1
|
||||
continue
|
||||
|
||||
pay_price = pay_price + decimal.Decimal( item['price'] ) * int( item['number'] )
|
||||
food_ids.append( item['id'] )
|
||||
|
||||
if continue_cnt >= len(items ) :
|
||||
resp['code'] = -1
|
||||
resp['msg'] = '商品items为空~~'
|
||||
return resp
|
||||
|
||||
yun_price = params['yun_price'] if params and 'yun_price' in params else 0
|
||||
note = params['note'] if params and 'note' in params else ''
|
||||
express_address_id = params['express_address_id'] if params and 'express_address_id' in params else 0
|
||||
express_info = params['express_info'] if params and 'express_info' in params else {}
|
||||
yun_price = decimal.Decimal( yun_price )
|
||||
total_price = pay_price + yun_price
|
||||
try:
|
||||
#为了防止并发库存出问题了，我们坐下selectfor update, 这里可以给大家演示下
|
||||
tmp_food_list = db.session.query( Food ).filter( Food.id.in_( food_ids ) )\
|
||||
.with_for_update().all()
|
||||
|
||||
tmp_food_stock_mapping = {}
|
||||
for tmp_item in tmp_food_list:
|
||||
tmp_food_stock_mapping[ tmp_item.id ] = tmp_item.stock
|
||||
|
||||
model_pay_order = PayOrder()
|
||||
model_pay_order.order_sn = self.geneOrderSn()
|
||||
model_pay_order.member_id = member_id
|
||||
model_pay_order.total_price = total_price
|
||||
model_pay_order.yun_price = yun_price
|
||||
model_pay_order.pay_price = pay_price
|
||||
model_pay_order.note = note
|
||||
model_pay_order.status = -8
|
||||
model_pay_order.express_status = -8
|
||||
model_pay_order.express_address_id = express_address_id
|
||||
model_pay_order.express_info = json.dumps( express_info )
|
||||
model_pay_order.updated_time = model_pay_order.created_time = getCurrentDate()
|
||||
db.session.add( model_pay_order )
|
||||
#db.session.flush()
|
||||
for item in items:
|
||||
tmp_left_stock = tmp_food_stock_mapping[ item['id'] ]
|
||||
|
||||
if decimal.Decimal(item['price']) < 0:
|
||||
continue
|
||||
|
||||
if int( item['number'] ) > int( tmp_left_stock ):
|
||||
raise Exception( "您购买的这美食太火爆了，剩余：%s,你购买%s~~"%( tmp_left_stock,item['number'] ) )
|
||||
|
||||
tmp_ret = Food.query.filter_by( id = item['id'] ).update({
|
||||
"stock":int(tmp_left_stock) - int(item['number'])
|
||||
})
|
||||
if not tmp_ret:
|
||||
raise Exception("下单失败请重新下单")
|
||||
|
||||
tmp_pay_item = PayOrderItem()
|
||||
tmp_pay_item.pay_order_id = model_pay_order.id
|
||||
tmp_pay_item.member_id = member_id
|
||||
tmp_pay_item.quantity = item['number']
|
||||
tmp_pay_item.price = item['price']
|
||||
tmp_pay_item.food_id = item['id']
|
||||
tmp_pay_item.note = note
|
||||
tmp_pay_item.updated_time = tmp_pay_item.created_time = getCurrentDate()
|
||||
db.session.add( tmp_pay_item )
|
||||
#db.session.flush()
|
||||
|
||||
FoodService.setStockChangeLog( item['id'],-item['number'],"在线购买" )
|
||||
db.session.commit()
|
||||
resp['data'] = {
|
||||
'id' : model_pay_order.id,
|
||||
'order_sn' : model_pay_order.order_sn,
|
||||
'total_price':str( total_price )
|
||||
}
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
print( e )
|
||||
resp['code'] = -1
|
||||
resp['msg'] = "下单失败请重新下单"
|
||||
resp['msg'] = str(e)
|
||||
return resp
|
||||
return resp
|
||||
|
||||
def closeOrder(self,pay_order_id = 0):
|
||||
if pay_order_id < 1:
|
||||
return False
|
||||
pay_order_info = PayOrder.query.filter_by( id = pay_order_id ,status = -8 ).first()
|
||||
if not pay_order_info:
|
||||
return False
|
||||
|
||||
pay_order_items = PayOrderItem.query.filter_by( pay_order_id = pay_order_id ).all()
|
||||
if pay_order_items:
|
||||
#需要归还库存
|
||||
for item in pay_order_items:
|
||||
tmp_food_info = Food.query.filter_by( id = item.food_id ).first()
|
||||
if tmp_food_info:
|
||||
tmp_food_info.stock = tmp_food_info.stock + item.quantity
|
||||
tmp_food_info.updated_time = getCurrentDate()
|
||||
db.session.add( tmp_food_info )
|
||||
db.session.commit()
|
||||
FoodService.setStockChangeLog( item.food_id, item.quantity, "订单取消")
|
||||
|
||||
pay_order_info.status = 0
|
||||
pay_order_info.updated_time = getCurrentDate()
|
||||
db.session.add( pay_order_info )
|
||||
db.session.commit()
|
||||
return True
|
||||
|
||||
def orderSuccess(self,pay_order_id = 0,params = None):
|
||||
try:
|
||||
pay_order_info = PayOrder.query.filter_by( id = pay_order_id ).first()
|
||||
if not pay_order_info or pay_order_info.status not in [ -8,-7 ]:
|
||||
return True
|
||||
|
||||
pay_order_info.pay_sn = params['pay_sn'] if params and 'pay_sn' in params else ''
|
||||
pay_order_info.status = 1
|
||||
pay_order_info.express_status = -7
|
||||
pay_order_info.updated_time = getCurrentDate()
|
||||
db.session.add( pay_order_info )
|
||||
|
||||
|
||||
pay_order_items = PayOrderItem.query.filter_by( pay_order_id = pay_order_id ).all()
|
||||
for order_item in pay_order_items:
|
||||
tmp_model_sale_log = FoodSaleChangeLog()
|
||||
tmp_model_sale_log.food_id = order_item.food_id
|
||||
tmp_model_sale_log.quantity = order_item.quantity
|
||||
tmp_model_sale_log.price = order_item.price
|
||||
tmp_model_sale_log.member_id = order_item.member_id
|
||||
tmp_model_sale_log.created_time = getCurrentDate()
|
||||
db.session.add( tmp_model_sale_log )
|
||||
|
||||
db.session.commit()
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
print(e)
|
||||
return False
|
||||
|
||||
#加入通知队列，做消息提醒和
|
||||
QueueService.addQueue( "pay",{
|
||||
"member_id": pay_order_info.member_id,
|
||||
"pay_order_id":pay_order_info.id
|
||||
})
|
||||
return True
|
||||
|
||||
def addPayCallbackData(self,pay_order_id = 0,type = 'pay',data = ''):
|
||||
model_callback = PayOrderCallbackData()
|
||||
model_callback.pay_order_id = pay_order_id
|
||||
if type == "pay":
|
||||
model_callback.pay_data = data
|
||||
model_callback.refund_data = ''
|
||||
else:
|
||||
model_callback.refund_data = data
|
||||
model_callback.pay_data = ''
|
||||
|
||||
model_callback.created_time = model_callback.updated_time = getCurrentDate()
|
||||
db.session.add( model_callback )
|
||||
db.session.commit()
|
||||
return True
|
||||
|
||||
def geneOrderSn(self):
|
||||
m = hashlib.md5()
|
||||
sn = None
|
||||
while True:
|
||||
str = "%s-%s"%( int( round( time.time() * 1000) ),random.randint( 0,9999999 ) )
|
||||
m.update(str.encode("utf-8"))
|
||||
sn = m.hexdigest()
|
||||
if not PayOrder.query.filter_by( order_sn = sn ).first():
|
||||
break
|
||||
return sn
|
||||
113
common/libs/pay/WeChatService.py
Normal file
113
common/libs/pay/WeChatService.py
Normal file
@@ -0,0 +1,113 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import hashlib,requests,uuid,json,datetime
|
||||
import xml.etree.ElementTree as ET
|
||||
from application import app,db
|
||||
from common.models.pay.OauthAccessToken import OauthAccessToken
|
||||
from common.libs.Helper import getCurrentDate
|
||||
class WeChatService():
|
||||
|
||||
def __init__(self,merchant_key = None):
|
||||
self.merchant_key = merchant_key
|
||||
|
||||
def create_sign(self, pay_data):
|
||||
'''
|
||||
生成签名
|
||||
:return:
|
||||
'''
|
||||
stringA = '&'.join(["{0}={1}".format(k, pay_data.get(k)) for k in sorted(pay_data)])
|
||||
stringSignTemp = '{0}&key={1}'.format(stringA, self.merchant_key)
|
||||
sign = hashlib.md5( stringSignTemp.encode("utf-8") ).hexdigest()
|
||||
return sign.upper()
|
||||
|
||||
def get_pay_info(self,pay_data = None):
|
||||
'''
|
||||
获取支付信息
|
||||
:param xml_data:
|
||||
:return:
|
||||
'''
|
||||
|
||||
sign = self.create_sign( pay_data )
|
||||
pay_data ['sign'] = sign
|
||||
xml_data = self.dict_to_xml( pay_data )
|
||||
headers = {'Content-Type': 'application/xml'}
|
||||
url = "https://api.mch.weixin.qq.com/pay/unifiedorder"
|
||||
r = requests.post( url= url, data=xml_data.encode('utf-8'),headers = headers)
|
||||
r.encoding = "utf-8"
|
||||
app.logger.info( r.text )
|
||||
if r.status_code == 200:
|
||||
prepay_id = self.xml_to_dict( r.text ).get('prepay_id')
|
||||
pay_sign_data = {
|
||||
'appId': pay_data.get('appid'),
|
||||
'timeStamp': pay_data.get('out_trade_no'),
|
||||
'nonceStr': pay_data.get('nonce_str'),
|
||||
'package': 'prepay_id={0}'.format(prepay_id),
|
||||
'signType': 'MD5'
|
||||
}
|
||||
pay_sign = self.create_sign( pay_sign_data )
|
||||
pay_sign_data.pop('appId')
|
||||
pay_sign_data['paySign'] = pay_sign
|
||||
pay_sign_data['prepay_id'] = prepay_id
|
||||
return pay_sign_data
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def dict_to_xml(self,dict_data):
|
||||
'''
|
||||
dict to xml
|
||||
:param dict_data:
|
||||
:return:
|
||||
'''
|
||||
xml = ["<xml>"]
|
||||
for k, v in dict_data.items():
|
||||
xml.append("<{0}>{1}</{0}>".format(k, v))
|
||||
xml.append("</xml>")
|
||||
return "".join(xml)
|
||||
|
||||
def xml_to_dict(self,xml_data):
|
||||
'''
|
||||
xml to dict
|
||||
:param xml_data:
|
||||
:return:
|
||||
'''
|
||||
xml_dict = {}
|
||||
root = ET.fromstring(xml_data)
|
||||
for child in root:
|
||||
xml_dict[child.tag] = child.text
|
||||
return xml_dict
|
||||
|
||||
def get_nonce_str(self):
|
||||
'''
|
||||
获取随机字符串
|
||||
:return:
|
||||
'''
|
||||
return str(uuid.uuid4()).replace('-', '')
|
||||
|
||||
def getAccessToken(self):
|
||||
token = None
|
||||
token_info = OauthAccessToken.query.filter( OauthAccessToken.expired_time >= getCurrentDate() ).first()
|
||||
if token_info:
|
||||
token = token_info.access_token
|
||||
return token
|
||||
|
||||
config_mina = app.config['MINA_APP']
|
||||
url = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid={0}&secret={1}"\
|
||||
.format( config_mina['appid'],config_mina['appkey'] )
|
||||
|
||||
r = requests.get( url = url )
|
||||
if r.status_code != 200 or not r.text:
|
||||
return token
|
||||
|
||||
data = json.loads( r.text )
|
||||
now = datetime.datetime.now()
|
||||
date = now + datetime.timedelta(seconds=data['expires_in'] - 200)
|
||||
model_token = OauthAccessToken()
|
||||
model_token.access_token = data['access_token']
|
||||
model_token.expired_time = date.strftime( "%Y-%m-%d %H:%M:%S" )
|
||||
model_token.created_time = getCurrentDate()
|
||||
db.session.add( model_token )
|
||||
db.session.commit()
|
||||
|
||||
return data['access_token']
|
||||
|
||||
|
||||
21
common/libs/queue/QueueService.py
Normal file
21
common/libs/queue/QueueService.py
Normal file
@@ -0,0 +1,21 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import json
|
||||
from common.models.queue.QueueList import QueueList
|
||||
from common.libs.Helper import getCurrentDate
|
||||
from application import app,db
|
||||
class QueueService():
|
||||
|
||||
@staticmethod
|
||||
def addQueue( queue_name,data = None ):
|
||||
model_queue = QueueList()
|
||||
model_queue.queue_name = queue_name
|
||||
if data:
|
||||
model_queue.data = json.dumps( data )
|
||||
|
||||
model_queue.created_time = model_queue.updated_time = getCurrentDate()
|
||||
db.session.add( model_queue )
|
||||
db.session.commit()
|
||||
return True
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user