fix: seed_lsx_invite_data 找不到 lsx 时列出全部俱乐部并支持 --club-id
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
XingQue
@@ -1,5 +1,12 @@
|
||||
"""龙先生电竞(lsx)邀请链种子：自动插入组长+管事（含邀请码），无需先登录小程序。"""
|
||||
from django.core.management.base import BaseCommand
|
||||
"""龙先生电竞邀请链种子：插入组长+管事（含邀请码）。
|
||||
|
||||
默认 club_id=lsx；若找不到会列出库里全部俱乐部，便于核对真实 club_id。
|
||||
可用：
|
||||
python manage.py seed_lsx_invite_data
|
||||
python manage.py seed_lsx_invite_data --club-id 你的真实id
|
||||
python manage.py seed_lsx_invite_data --list-clubs
|
||||
"""
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
from django.db import transaction
|
||||
|
||||
from jituan.models import Club, ClubHuiyuanPrice
|
||||
@@ -7,7 +14,7 @@ from users.business_models import User
|
||||
from users.models import UserBoss, UserGuanshi, UserZuzhang
|
||||
from utils.invitationcode_utils import CreateInvitationCode
|
||||
|
||||
CLUB_ID = 'lsx'
|
||||
DEFAULT_CLUB_ID = 'lsx'
|
||||
WX_APPID = 'wx7ff90e9d024fcdb8'
|
||||
|
||||
# 固定种子账号（可重复执行，幂等；UID/手机避开 xzj 的 9900001/2）
|
||||
@@ -27,23 +34,62 @@ SEED_GUANSHI = {
|
||||
}
|
||||
|
||||
|
||||
def _ensure_club():
|
||||
"""仅检查俱乐部是否已由后台创建；不自动建空俱乐部。"""
|
||||
if Club.query.filter(club_id=CLUB_ID).exists():
|
||||
return
|
||||
raise SystemExit(
|
||||
f'俱乐部 {CLUB_ID} 不存在。请先在后台创建龙先生俱乐部并上传证书，'
|
||||
f'再执行 seed_lsx_invite_data（本命令不再自动 create_club）。'
|
||||
def _list_clubs(stdout, style):
|
||||
clubs = list(Club.query.order_by('sort_order', 'club_id'))
|
||||
stdout.write(style.SUCCESS('=' * 60))
|
||||
stdout.write(f'库里俱乐部共 {len(clubs)} 个：')
|
||||
for c in clubs:
|
||||
stdout.write('-' * 40)
|
||||
stdout.write(f' club_id={c.club_id!r} name={c.name!r} status={c.status}')
|
||||
stdout.write(f' wx_appid={c.wx_appid or "(空)"}')
|
||||
stdout.write(f' mch_id={c.mch_id or "(空)"} withdraw_mch_id={getattr(c, "withdraw_mch_id", "") or "(空)"}')
|
||||
stdout.write(
|
||||
f' pay_key_path={c.pay_key_path or "(空)"}'
|
||||
)
|
||||
stdout.write(style.SUCCESS('=' * 60))
|
||||
return clubs
|
||||
|
||||
|
||||
def _ensure_user(meta):
|
||||
def _resolve_club(club_id: str, stdout, style):
|
||||
"""优先精确 club_id；其次 wx_appid；再名称含「龙先生」。"""
|
||||
cid = (club_id or DEFAULT_CLUB_ID).strip()
|
||||
club = Club.query.filter(club_id=cid).first()
|
||||
if club:
|
||||
return club
|
||||
|
||||
stdout.write(style.WARNING(f'精确 club_id={cid!r} 不存在，尝试按 AppID/名称匹配…'))
|
||||
|
||||
by_app = Club.query.filter(wx_appid=WX_APPID).first()
|
||||
if by_app:
|
||||
stdout.write(style.WARNING(
|
||||
f'按 wx_appid={WX_APPID} 命中: club_id={by_app.club_id!r} name={by_app.name!r}'
|
||||
))
|
||||
return by_app
|
||||
|
||||
by_name = Club.query.filter(name__icontains='龙先生').order_by('club_id')
|
||||
names = list(by_name)
|
||||
if len(names) == 1:
|
||||
c = names[0]
|
||||
stdout.write(style.WARNING(
|
||||
f'按名称「龙先生」命中唯一: club_id={c.club_id!r} name={c.name!r}'
|
||||
))
|
||||
return c
|
||||
if len(names) > 1:
|
||||
stdout.write(style.ERROR('名称含「龙先生」的俱乐部有多条，请用 --club-id 指定：'))
|
||||
for c in names:
|
||||
stdout.write(f' - {c.club_id!r} / {c.name!r}')
|
||||
return None
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def _ensure_user(meta, club_id):
|
||||
user = User.query.filter(UserUID=meta['UserUID']).first()
|
||||
if not user:
|
||||
user = User.query.filter(OpenID=meta['OpenID']).first()
|
||||
if user:
|
||||
if user.ClubID != CLUB_ID:
|
||||
user.ClubID = CLUB_ID
|
||||
if (user.ClubID or '') != club_id:
|
||||
user.ClubID = club_id
|
||||
user.save(update_fields=['ClubID'])
|
||||
return user, False
|
||||
|
||||
@@ -52,15 +98,17 @@ def _ensure_user(meta):
|
||||
UserName=meta['UserName'],
|
||||
OpenID=meta['OpenID'],
|
||||
Phone=meta['Phone'],
|
||||
ClubID=CLUB_ID,
|
||||
ClubID=club_id,
|
||||
)
|
||||
UserBoss.query.create(user=user, nickname=meta['nickname'])
|
||||
return user, True
|
||||
|
||||
|
||||
def _ensure_zuzhang(user):
|
||||
if hasattr(user, 'ZuzhangProfile'):
|
||||
try:
|
||||
return user.ZuzhangProfile, False
|
||||
except UserZuzhang.DoesNotExist:
|
||||
pass
|
||||
yaoqingma = CreateInvitationCode(str(user.UserUID))
|
||||
z = UserZuzhang.query.create(
|
||||
user=user,
|
||||
@@ -71,8 +119,10 @@ def _ensure_zuzhang(user):
|
||||
|
||||
|
||||
def _ensure_guanshi(user, zuzhang_uid):
|
||||
if hasattr(user, 'GuanshiProfile'):
|
||||
try:
|
||||
return user.GuanshiProfile, False
|
||||
except UserGuanshi.DoesNotExist:
|
||||
pass
|
||||
yaoqingma = CreateInvitationCode(str(user.UserUID))
|
||||
g = UserGuanshi.query.create(
|
||||
user=user,
|
||||
@@ -84,9 +134,20 @@ def _ensure_guanshi(user, zuzhang_uid):
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = '插入龙先生电竞种子组长+管事（含邀请码），一条命令即可邀请注册'
|
||||
help = '插入龙先生电竞种子组长+管事（含邀请码）；找不到俱乐部时会列出全部 club'
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument(
|
||||
'--club-id',
|
||||
type=str,
|
||||
default=DEFAULT_CLUB_ID,
|
||||
help='目标俱乐部 ID（默认 lsx；若后台建的不是 lsx 请改这里）',
|
||||
)
|
||||
parser.add_argument(
|
||||
'--list-clubs',
|
||||
action='store_true',
|
||||
help='只列出库里所有俱乐部，不写入种子',
|
||||
)
|
||||
parser.add_argument(
|
||||
'--force-new-codes',
|
||||
action='store_true',
|
||||
@@ -94,25 +155,49 @@ class Command(BaseCommand):
|
||||
)
|
||||
|
||||
def handle(self, *args, **options):
|
||||
with transaction.atomic():
|
||||
_ensure_club()
|
||||
if options.get('list_clubs'):
|
||||
_list_clubs(self.stdout, self.style)
|
||||
return
|
||||
|
||||
zz_user, zz_new = _ensure_user(SEED_ZUZHANG)
|
||||
want_id = (options.get('club_id') or DEFAULT_CLUB_ID).strip()
|
||||
club = _resolve_club(want_id, self.stdout, self.style)
|
||||
if club is None:
|
||||
self.stdout.write(self.style.ERROR(
|
||||
f'找不到可用俱乐部（请求 club_id={want_id!r}）。'
|
||||
))
|
||||
self.stdout.write(self.style.WARNING(
|
||||
'可能原因：\n'
|
||||
' 1) 后台创建时 club_id 不是 lsx（例如写成了别的拼音）\n'
|
||||
' 2) 之前执行过 delete_club_by_mch_id --apply，把 lsx 主表删掉了\n'
|
||||
'请先看下面列表，用真实 club_id 再跑：\n'
|
||||
' python manage.py seed_lsx_invite_data --list-clubs\n'
|
||||
' python manage.py seed_lsx_invite_data --club-id 真实id'
|
||||
))
|
||||
_list_clubs(self.stdout, self.style)
|
||||
raise CommandError('俱乐部不存在，已中止（未写入任何种子）')
|
||||
|
||||
club_id = club.club_id
|
||||
self.stdout.write(self.style.SUCCESS(
|
||||
f'将写入种子到俱乐部: club_id={club_id!r} name={club.name!r} wx_appid={club.wx_appid or "(空)"}'
|
||||
))
|
||||
|
||||
with transaction.atomic():
|
||||
zz_user, zz_new = _ensure_user(SEED_ZUZHANG, club_id)
|
||||
zz_profile, zz_profile_new = _ensure_zuzhang(zz_user)
|
||||
if options.get('force_new_codes') and not zz_profile_new:
|
||||
zz_profile.yaoqingma = CreateInvitationCode(str(zz_user.UserUID))
|
||||
zz_profile.save(update_fields=['yaoqingma'])
|
||||
|
||||
gs_user, gs_new = _ensure_user(SEED_GUANSHI)
|
||||
gs_user, gs_new = _ensure_user(SEED_GUANSHI, club_id)
|
||||
gs_profile, gs_profile_new = _ensure_guanshi(gs_user, zz_user.UserUID)
|
||||
if options.get('force_new_codes') and not gs_profile_new:
|
||||
gs_profile.yaoqingma = CreateInvitationCode(str(gs_user.UserUID))
|
||||
gs_profile.save(update_fields=['yaoqingma'])
|
||||
|
||||
price_cnt = ClubHuiyuanPrice.query.filter(club_id=CLUB_ID).count()
|
||||
price_cnt = ClubHuiyuanPrice.query.filter(club_id=club_id).count()
|
||||
|
||||
self.stdout.write(self.style.SUCCESS('=' * 60))
|
||||
self.stdout.write(self.style.SUCCESS('龙先生电竞 lsx 邀请链种子数据已就绪'))
|
||||
self.stdout.write(self.style.SUCCESS(f'龙先生电竞邀请链种子已就绪（club_id={club_id}）'))
|
||||
self.stdout.write(self.style.SUCCESS('=' * 60))
|
||||
self.stdout.write(f'club_huiyuan_price 条数: {price_cnt}')
|
||||
self.stdout.write('')
|
||||
@@ -124,7 +209,7 @@ class Command(BaseCommand):
|
||||
self.stdout.write(f' UserUID={gs_user.UserUID} Phone={SEED_GUANSHI["Phone"]}')
|
||||
self.stdout.write(f' 管事邀请码 inviteCode = {gs_profile.yaoqingma}')
|
||||
self.stdout.write('')
|
||||
self.stdout.write('商家：请在后台「商家管理 → 添加商家」按用户UID开通（不再用前端邀请码）')
|
||||
self.stdout.write('商家：后台「商家管理 → 添加商家」按用户UID开通')
|
||||
self.stdout.write('')
|
||||
self.stdout.write('新建: 组长=%s 管事=%s' % (
|
||||
'是' if zz_new or zz_profile_new else '否（已存在）',
|
||||
|
||||
Reference in New Issue
Block a user