feat(jituan): 集团多俱乐部改造 — club 隔离、财务/提现/分红/展示配置

jituan/services/user_stats.py

@@ -0,0 +1,80 @@
+"""用户管理汇总统计（集团汇总 + 按俱乐部拆分）。"""
+from django.utils import timezone
+
+from jituan.constants import DATA_SCOPE_ALL
+from jituan.models import Club
+from jituan.services.club_context import (
+    filter_queryset_by_club,
+    resolve_club_id_from_request,
+    resolve_club_scope,
+    filter_user_qs_by_club,
+    filter_user_related_by_club,
+)
+from products.models import Huiyuangoumai
+from users.business_models import User
+from users.models import UserDashou, UserGuanshi, UserShangjia, UserZuzhang
+
+
+def _dashou_valid_count(request, dashou_user_ids):
+    now = timezone.now()
+    hy_qs = filter_queryset_by_club(Huiyuangoumai.query.all(), request)
+    return hy_qs.filter(
+        yonghu_id__in=dashou_user_ids,
+        huiyuan_zhuangtai=1,
+        daoqi_time__gt=now,
+    ).values('yonghu_id').distinct().count()
+
+
+def _counts_for_club_id(club_id):
+    dashou_qs = UserDashou.query.filter(user__ClubID=club_id)
+    dashou_ids = dashou_qs.values_list('user__UserUID', flat=True)
+    now = timezone.now()
+    dashou_valid = Huiyuangoumai.query.filter(
+        club_id=club_id,
+        yonghu_id__in=dashou_ids,
+        huiyuan_zhuangtai=1,
+        daoqi_time__gt=now,
+    ).values('yonghu_id').distinct().count()
+    shangjia_qs = UserShangjia.query.filter(user__ClubID=club_id)
+    return {
+        'club_id': club_id,
+        'dashou_total': dashou_qs.count(),
+        'dashou_valid': dashou_valid,
+        'guanshi_total': User.query.filter(ClubID=club_id, GuanshiProfile__isnull=False).count(),
+        'shangjia_total': shangjia_qs.count(),
+        'shangjia_valid': shangjia_qs.filter(zhuangtai=1).count(),
+        'zuzhang_total': UserZuzhang.query.filter(user__ClubID=club_id).count(),
+        'zuzhang_active': UserZuzhang.query.filter(user__ClubID=club_id, zhuangtai=1).count(),
+    }
+
+
+def build_user_summary(request):
+    dashou_qs = filter_user_related_by_club(UserDashou.query.all(), request)
+    guanshi_qs = filter_user_qs_by_club(
+        User.query.filter(GuanshiProfile__isnull=False), request,
+    )
+    shangjia_qs = filter_user_related_by_club(UserShangjia.query.all(), request)
+    zuzhang_qs = filter_user_related_by_club(UserZuzhang.query.all(), request)
+
+    dashou_ids = dashou_qs.values_list('user__UserUID', flat=True)
+    result = {
+        'club_id': resolve_club_id_from_request(request),
+        'scope': resolve_club_scope(request),
+        'dashou_total': dashou_qs.count(),
+        'dashou_valid': _dashou_valid_count(request, dashou_ids),
+        'guanshi_total': guanshi_qs.count(),
+        'shangjia_total': shangjia_qs.count(),
+        'shangjia_valid': shangjia_qs.filter(zhuangtai=1).count(),
+        'zuzhang_total': zuzhang_qs.count(),
+        'zuzhang_active': zuzhang_qs.filter(zhuangtai=1).count(),
+    }
+
+    if resolve_club_scope(request) == DATA_SCOPE_ALL:
+        by_club = []
+        for club in Club.query.filter(status=1).order_by('sort_order', 'club_id'):
+            row = _counts_for_club_id(club.club_id)
+            row['name'] = club.name
+            by_club.append(row)
+        result['by_club'] = by_club
+
+    return result