def user_identify():
# 添加银行卡和身份证
u_data = request.get_json()
token = u_data.get('token')
if check_token(token):
dao = UserDao()
user_id = get_token_user_id(token)
print(user_id)
data = dao.check_real(user_id)
print(data)
if data:
u_bank = data[0]['u_bank']
user_card = data[0]['user_card']
if (u_bank and user_card) is not None:
return jsonify({'code': 300, 'msg': '您已添加过该信息！'})
else:
dict1 = {
'u_bank': u_data.get('u_bank'),
'user_card': u_data.get('user_card'),
'is_val': 1,
'asset': 0
}
for k, v in dict1.items():
user_save = dao.update('jd_user', k, v, 'user_id', user_id)
if user_save:
return jsonify({'code': 200, 'msg': '验证通过！'})
else:
return jsonify({'code': 400, 'msg': '添加失败！'})
else:
return jsonify({'code': 400, 'msg': '数据库查询失败！'})
else:
return jsonify({'code': 400, 'msg': 'token验证失败！'})
def recharge():
# 余额充值
req_data = request.get_json()
token = req_data.get('token')
amount = req_data.get('amount') # 获取上传的充值金额
if check_token(token):
user_id = get_token_user_id(token)
ass_data = VouDao().check_asset(user_id)
balance = int(ass_data[0]['asset'])
print(balance)
if balance is not None: # 查询到金额
balance += amount
dao = UserDao()
dao.update('jd_user', 'asset', balance, 'user_id', user_id)
return jsonify({'code': 200, 'data': balance})
else:
return jsonify({'code': 400, 'msg': '查询金额失败！'})
else:
return jsonify({'code': 400, 'msg': 'token验证失败！'})
def user_update():
# 用户登录后修改自己的个人信息
u_data = request.get_json()
token = u_data.get('token') # 获取token
print(token)
old_pwd = u_data.get('old_pwd') # 获取用户上传的原密码
new_pwd = u_data.get('new_pwd') # 获取用户上传的新密码
if check_token(token): # 验证token
user_id = get_token_user_id(token) # 通过token获取用户的ID
check_data = UserDao().get_profile(user_id) # 通过用户ID查询用户的详细信息
if check_data: # 如果查询成功
dao = UserDao()
if all((old_pwd, new_pwd)): # 修改支付密码
pay_pwd = check_data[0]['pay_pwd'] # 获取数据库中的密码
if old_pwd == pay_pwd: # 若原密码等于库中的密码
user_save = dao.update('jd_user', "pay_pwd", new_pwd,
'user_id', user_id) # 将新密码存入数据库
if user_save: # 如果保存成功
data = {
'userPhoto': u_data.get('u_img'),
'userID': user_id,
'userName': u_data.get('user_name'),
'realname': check_data[0]['is_val'],
'TEL': check_data[0]['tel'],
'Email': check_data[0]['u_email'],
}
return jsonify({
'code': 200,
'msg': '修改成功！',
'data': data
})
else:
return jsonify({'code': 400, 'msg': '修改失败！'})
else:
return jsonify({'code': 400, 'msg': '原密码输入错误！'})
elif old_pwd: # 设置支付密码
user_save = False
dict1 = {
'u_img': u_data.get('u_img'),
'user_name': u_data.get('user_name'),
'pay_pwd': u_data.get('old_pwd')
}
for k, v in dict1.items():
user_save = dao.update('jd_user', k, v, 'user_id', user_id)
if user_save:
data = {
'userPhoto': u_data.get('u_img'),
'userID': user_id,
'userName': u_data.get('user_name'),
'realname': check_data[0]['is_val'],
'TEL': check_data[0]['tel'],
'Email': check_data[0]['u_email'],
}
return jsonify({'code': 200, 'msg': '修改成功！', 'data': data})
else:
return jsonify({'code': 400, 'msg': '修改失败！'})
else: # 修改除密码外的其他信息
user_save = False
dict1 = {
'u_img': u_data.get('u_img'),
'user_name': u_data.get('user_name')
}
for k, v in dict1.items():
user_save = dao.update('jd_user', k, v, 'user_id', user_id)
if user_save:
data = {
'userPhoto': u_data.get('u_img'),
'userID': user_id,
'userName': u_data.get('user_name'),
'realname': check_data[0]['is_val'],
'TEL': check_data[0]['tel'],
'Email': check_data[0]['u_email'],
}
return jsonify({'code': 200, 'msg': '修改成功！', 'data': data})
else:
return jsonify({'code': 200, 'msg': '修改失败！'})
else:
return jsonify({
'code': 400,
'msg': '查询数据库失败！',
})
else:
return jsonify({'code': 400, 'msg': 'token验证失败！'})
