rbac/wwwroot/phone_login.dspy
Hermes Agent 70a19238c1 feat: 注册时检查促销码并绑定客户归属关系
- register.dspy: 用户名密码注册后检查promo_code_id
- sms_register.dspy: 短信注册后检查promo_code_id
- code_login.dspy: 验证码登录/自动注册后检查promo_code_id
- phone_login.dspy: 手机登录/自动注册后检查promo_code_id
- 绑定失败只记日志不影响注册成功(try/except)
2026-06-24 00:56:33 +08:00

127 lines
3.0 KiB
Plaintext

# 用短信模块检查验证码是否正确
debug(f'phone_login.dspy:{params_kw=}')
if params_kw.cellphone is None:
return {
"status": "error",
"data":{
"message": "需输入手机号"
}
}
if params_kw.sms_code is None:
return {
"status": "error",
"data": {
"message": "需输入验证码"
}
}
if params_kw.key is None:
return {
"status": "error",
"data": {
"message": "需要短信验证key"
}
}
f = await sms_engine.check_sms_code(params_kw.key, params_kw.sms_code)
if not f:
return {
"status": "error",
"data": {
"message": "手机短信验证码出错"
}
}
ns = {
"username": params_kw.cellphone,
"password": "^&%UHI",
"cfm_password": "^&%UHI",
"mobile": params_kw.cellphone,
"user_status": "0"
}
udata = DictObject(**ns)
try:
async with get_sor_context(request._run_ns, 'rbac') as sor:
recs = await sor.R('users', {'mobile': params_kw.cellphone})
if recs:
if len(recs) == 1:
r = recs[0]
# Update last_login atomically (standard SQL, no DB-specific functions)
now_str = timestampstr()
await sor.sqlExe("""
UPDATE users
SET last_login = ${now}$, login_fail_count = 0,
last_login_fail = NULL
WHERE id = ${id}$
""", {'id': r.id, 'now': now_str})
await remember_user(r.id, username=r.username, userorgid=r.orgid)
debug(f'here')
return {
"status": "ok",
"data":{
"user": r
}
}
if params_kw.selected_id:
for r in recs:
if r.id == params_kw.selected_id:
now_str = timestampstr()
await sor.sqlExe("""
UPDATE users
SET last_login = ${now}$, login_fail_count = 0,
last_login_fail = NULL
WHERE id = ${id}$
""", {'id': r.id, 'now': now_str})
await remember_user(r.id, username=r.username, userorgid=r.orgid)
debug(f'here')
return {
"status": "ok",
"data":{
"user": r
}
}
else:
debug(f'here')
return {
"status": "choose",
"data": {
"users": recs
}
}
d = await register_user(sor, udata)
if d['status'] == 'error':
debug(f'here, {d}')
return d
try:
ownerid = await get_owner_orgid(sor, orgid)
await openCustomerAccounts(sor, ownerid, orgid)
except Exception as e:
exception(f'{e}')
# 检查是否有促销码,如果有则绑定客户归属关系
promo_code_id = params_kw.get('promo_code_id')
if promo_code_id:
try:
_new_orgid = d['data']['user'].orgid
await bind_customer(request, DictObject(promo_code_id=promo_code_id, customerid=_new_orgid))
debug(f'phone_login: customer {_new_orgid} bound via promo_code {promo_code_id}')
except Exception as e:
exception(f'phone_login: bind_customer failed: {e}')
r = d['data']['user']
await remember_user(r.id, username=r.username, userorgid=r.orgid)
debug(f'here')
return {
"status": "ok",
"data":{
"user": r
}
}
except Exception as e:
exception(f'Error happend{e}')
return {
"status": "error",
"data":{
"message": f"{e}"
}
}