66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
"""为后台客服账号初始化 admin_assignment（数据范围，非功能权限）。"""
|
||
import logging
|
||
|
||
from django.core.management.base import BaseCommand
|
||
from django.db import transaction
|
||
|
||
from jituan.constants import CLUB_ID_DEFAULT, DATA_SCOPE_SINGLE, ADMIN_ROLE_LABELS
|
||
from jituan.models import AdminAssignment
|
||
from users.business_models import User
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = '为 kefu 账号创建默认子公司任职（club=xq, CLUB_ADMIN）；超管可不建 assignment'
|
||
|
||
def add_arguments(self, parser):
|
||
parser.add_argument(
|
||
'--club-id', default=CLUB_ID_DEFAULT, help='目标俱乐部 ID，默认 xq',
|
||
)
|
||
parser.add_argument(
|
||
'--role-code', default='CLUB_ADMIN', help='任职角色码',
|
||
)
|
||
parser.add_argument(
|
||
'--dry-run', action='store_true', help='只打印不写入',
|
||
)
|
||
|
||
def handle(self, *args, **options):
|
||
club_id = options['club_id']
|
||
role_code = options['role_code']
|
||
dry_run = options['dry_run']
|
||
|
||
kefu_users = User.query.filter(UserType='kefu')
|
||
created = 0
|
||
skipped = 0
|
||
|
||
with transaction.atomic():
|
||
for u in kefu_users:
|
||
if u.IsSuperuser:
|
||
skipped += 1
|
||
continue
|
||
exists = AdminAssignment.query.filter(
|
||
yonghuid=u.UserUID, club_id=club_id, role_code=role_code,
|
||
).exists()
|
||
if exists:
|
||
skipped += 1
|
||
continue
|
||
if dry_run:
|
||
self.stdout.write(f'would assign {u.UserUID} -> {club_id} {role_code}')
|
||
created += 1
|
||
continue
|
||
AdminAssignment.query.create(
|
||
yonghuid=u.UserUID,
|
||
club_id=club_id,
|
||
role_code=role_code,
|
||
data_scope=DATA_SCOPE_SINGLE,
|
||
is_primary=True,
|
||
status=1,
|
||
)
|
||
created += 1
|
||
|
||
label = ADMIN_ROLE_LABELS.get(role_code, role_code)
|
||
self.stdout.write(self.style.SUCCESS(
|
||
f'完成：新建 {created} 条，跳过 {skipped} 条（club={club_id}, role={label}）'
|
||
))
|