53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
"""邀请码 / 邀请链俱乐部隔离。"""
|
||
import logging
|
||
|
||
from jituan.models import UserWxOpenid
|
||
from jituan.services.club_resolver import resolve_club_id_for_wechat_request
|
||
from jituan.services.club_user import resolve_invite_party_club
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _invitee_has_club_binding(user, club_id):
|
||
uid = getattr(user, 'UserUID', None)
|
||
if not uid or not club_id:
|
||
return False
|
||
return UserWxOpenid.query.filter(yonghuid=uid, club_id=club_id).exists()
|
||
|
||
|
||
def assert_invite_same_club(request, inviter_user, invitee_user=None):
|
||
"""
|
||
校验邀请人与被邀请人属于同一俱乐部，且与请求 club 一致。
|
||
未归属俱乐部的被邀请人允许在本小程序完成首次注册。
|
||
返回 (ok, message)
|
||
"""
|
||
club_id = resolve_club_id_for_wechat_request(request)
|
||
inviter_club = resolve_invite_party_club(inviter_user, club_id)
|
||
if inviter_club is None:
|
||
# 管事未落库 ClubID/绑定时，以当前小程序为准（星之界扫码邀请常见）
|
||
inviter_club = club_id
|
||
if inviter_club != club_id:
|
||
logger.warning(
|
||
'invite_guard 管事俱乐部不符 inviter=%s inviter_club=%s req_club=%s',
|
||
getattr(inviter_user, 'UserUID', '?'), inviter_club, club_id,
|
||
)
|
||
return False, '邀请码不属于当前小程序俱乐部'
|
||
|
||
if invitee_user is not None:
|
||
invitee_club = resolve_invite_party_club(invitee_user, club_id)
|
||
if invitee_club is not None and invitee_club != club_id:
|
||
# 仅在「管事码也不属于当前小程序」且「用户已绑定别俱乐部 wx」时拒绝
|
||
if (
|
||
inviter_club != club_id
|
||
and _invitee_has_club_binding(invitee_user, invitee_club)
|
||
and not _invitee_has_club_binding(invitee_user, club_id)
|
||
):
|
||
logger.warning(
|
||
'invite_guard 被邀请人俱乐部不符 invitee=%s invitee_club=%s req_club=%s',
|
||
getattr(invitee_user, 'UserUID', '?'), invitee_club, club_id,
|
||
)
|
||
return False, '您的账号归属与当前小程序不一致，请使用对应俱乐部小程序'
|
||
# 管事码已证明属于当前小程序时，允许老 xq 账号在 xzj 完成注册（注册后写 ClubID）
|
||
|
||
return True, ''
|