119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
"""admin_assignment 任职 CRUD（修复停用不生效、重复插入）。"""
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
|
|
from jituan.constants import DATA_SCOPE_ALL, DATA_SCOPE_SINGLE
|
|
from jituan.models import AdminAssignment
|
|
|
|
|
|
def _club_filter(qs, club_id):
|
|
if club_id is None:
|
|
return qs.filter(Q(club_id__isnull=True) | Q(club_id=''))
|
|
return qs.filter(club_id=club_id)
|
|
|
|
|
|
def find_assignment(yonghuid, club_id, role_code):
|
|
qs = AdminAssignment.query.filter(yonghuid=yonghuid, role_code=role_code)
|
|
return _club_filter(qs, club_id).first()
|
|
|
|
|
|
def find_active_assignment(yonghuid, club_id, role_code):
|
|
qs = AdminAssignment.query.filter(yonghuid=yonghuid, role_code=role_code, status=1)
|
|
return _club_filter(qs, club_id).first()
|
|
|
|
|
|
def clear_other_primary_assignments(yonghuid, except_id=None):
|
|
qs = AdminAssignment.query.filter(yonghuid=yonghuid, status=1, is_primary=True)
|
|
if except_id:
|
|
qs = qs.exclude(id=except_id)
|
|
ids = list(qs.values_list('id', flat=True))
|
|
if ids:
|
|
AdminAssignment.objects.filter(pk__in=ids).update(
|
|
is_primary=False,
|
|
UpdateTime=timezone.now(),
|
|
)
|
|
|
|
|
|
def _clear_other_primary(yonghuid, except_id=None):
|
|
clear_other_primary_assignments(yonghuid, except_id=except_id)
|
|
|
|
|
|
def create_or_reactivate_assignment(
|
|
yonghuid,
|
|
club_id,
|
|
role_code,
|
|
data_scope,
|
|
is_primary,
|
|
granted_by,
|
|
):
|
|
active = find_active_assignment(yonghuid, club_id, role_code)
|
|
if active:
|
|
return None, '该任职记录已存在且有效'
|
|
|
|
if is_primary:
|
|
_clear_other_primary(yonghuid)
|
|
|
|
inactive = find_assignment(yonghuid, club_id, role_code)
|
|
if inactive and inactive.status != 1:
|
|
if is_primary:
|
|
_clear_other_primary(yonghuid)
|
|
AdminAssignment.objects.filter(pk=inactive.id).update(
|
|
status=1,
|
|
data_scope=data_scope,
|
|
is_primary=is_primary,
|
|
granted_by=granted_by,
|
|
UpdateTime=timezone.now(),
|
|
)
|
|
return inactive.id, None
|
|
|
|
row = AdminAssignment.query.create(
|
|
yonghuid=yonghuid,
|
|
club_id=club_id,
|
|
role_code=role_code,
|
|
data_scope=data_scope,
|
|
is_primary=is_primary,
|
|
granted_by=granted_by,
|
|
status=1,
|
|
)
|
|
return row.id, None
|
|
|
|
|
|
def deactivate_assignment(assignment_id):
|
|
try:
|
|
pk = int(assignment_id)
|
|
except (TypeError, ValueError):
|
|
return False, '任职 id 无效'
|
|
|
|
updated = AdminAssignment.objects.filter(pk=pk, status=1).update(
|
|
status=0,
|
|
UpdateTime=timezone.now(),
|
|
)
|
|
if updated:
|
|
return True, None
|
|
|
|
exists = AdminAssignment.objects.filter(pk=pk).exists()
|
|
if exists:
|
|
return True, None
|
|
return False, '任职记录不存在'
|
|
|
|
|
|
def dedupe_active_assignments():
|
|
"""停用重复的有效任职，每组 (yonghuid, club_id, role_code) 只保留一条。"""
|
|
rows = list(
|
|
AdminAssignment.query.filter(status=1).order_by('yonghuid', 'club_id', 'role_code', 'id')
|
|
)
|
|
seen = {}
|
|
dup_ids = []
|
|
for row in rows:
|
|
key = (row.yonghuid, row.club_id or '', row.role_code)
|
|
if key in seen:
|
|
dup_ids.append(row.id)
|
|
else:
|
|
seen[key] = row.id
|
|
if dup_ids:
|
|
AdminAssignment.objects.filter(pk__in=dup_ids).update(
|
|
status=0,
|
|
UpdateTime=timezone.now(),
|
|
)
|
|
return len(dup_ids)
|