def buy_vit(uid):
add_ct = 10
u = UserAction(uid)
user_info = u.get_model_info()
re_action = RecordAction(uid)
record_info = re_action.get_model_info()
cur_buy_ct = record_info.get(re_action.buy_vit_str, 0)
price = 10
cost = 2**cur_buy_ct * price
cur_diamond = user_info.get(u.diamond_str, 0)
if cur_diamond < cost:
print (uid, u.diamond_str, 'not enough', cur_diamond, cost, cur_buy_ct)
return False
data = {
u.diamond_str: cur_diamond - cost,
u.vit_str: user_info.get(u.vit_str, 0) + add_ct
}
# 记录次数
re_action.update_model_info({re_action.buy_vit_str: cur_buy_ct+1})
res = {}
if u.set_values(data):
res['award'] = {'vit': add_ct}
return res
def set_userinfo(uid, data):
u = UserAction(uid)
u.set_values(data)
return u
