完成视频复刻和优化

core/backend/apps/assets/review.py

@@ -116,6 +116,73 @@ def reference_review_state(asset: Asset) -> str:
     return "unsubmitted"
 
 
+def _unsubmitted_review_qs(*, team=None):
+    qs = Asset.objects.filter(
+        category__in=Asset.REVIEW_CATEGORIES,
+        review_status="",
+        is_deleted=False,
+    )
+    if team is not None:
+        qs = qs.filter(team=team)
+    return qs
+
+
+def _processing_review_qs(*, team=None):
+    qs = Asset.objects.filter(
+        category__in=Asset.REVIEW_CATEGORIES,
+        review_status="processing",
+        is_deleted=False,
+    )
+    if team is not None:
+        qs = qs.filter(team=team)
+    return qs
+
+
+def submit_unsubmitted_reviews(*, team=None, limit: int = 25) -> int:
+    """把未送审的人脸资产自动丢给火山,不用人去后台点「批量送审」。"""
+    submitted = 0
+    cap = max(1, min(int(limit), 50))
+    for asset in _unsubmitted_review_qs(team=team).order_by("created_at")[:cap]:
+        if submit_asset_for_review(asset):
+            submitted += 1
+    return submitted
+
+
+def poll_processing_reviews(*, team=None, limit: int = 80) -> dict[str, str]:
+    cap = max(1, min(int(limit), 200))
+    out: dict[str, str] = {}
+    for asset in _processing_review_qs(team=team).order_by("updated_at")[:cap]:
+        out[str(asset.id)] = poll_asset_review(asset)
+    return out
+
+
+def drain_platform_reviews(*, submit_limit: int = 25, poll_limit: int = 80) -> dict:
+    """全平台:未送审自动送 + 审核中自动拉状态。worker 兜底循环,人不在后台页也会跑。"""
+    submitted = submit_unsubmitted_reviews(limit=submit_limit)
+    statuses = poll_processing_reviews(limit=poll_limit)
+    return {"submitted": submitted, "polled": len(statuses), "statuses": statuses}
+
+
+def ensure_review_drain_loop() -> None:
+    """第一次有人轮询审核时,顺手把 worker 兜底循环拉起来(没有 beat 也能自转)。"""
+    import sys
+
+    from django.conf import settings as dj_settings
+    from django.core.cache import cache
+
+    if getattr(dj_settings, "CELERY_TASK_ALWAYS_EAGER", False) or "test" in sys.argv:
+        return
+    if not cache.add("asset-review-drain-kick", "1", timeout=300):
+        return
+    try:
+        from apps.ai.tasks import drain_asset_reviews_task
+
+        drain_asset_reviews_task.delay()
+    except Exception:  # noqa: BLE001
+        cache.delete("asset-review-drain-kick")
+        logger.warning("kick asset review drain loop failed", exc_info=True)
+
+
 def poll_asset_review(asset: Asset) -> str:
     """查单个真人资产审核状态并更新 review_status。返回最新状态。
     只在状态变化时落库(保留 updated_at 作为「进入 processing 的时刻」);processing 超时兜底为 failed。"""
@@ -144,11 +211,8 @@ def poll_asset_review(asset: Asset) -> str:
 
 
 def poll_team_reviews(team) -> dict:
-    """轮询该团队所有「审核中」真人资产,更新状态。返回 {asset_id: status}。"""
-    out: dict[str, str] = {}
-    pending = Asset.objects.filter(
-        team=team, category__in=Asset.REVIEW_CATEGORIES, review_status="processing", is_deleted=False
-    )
-    for asset in pending:
-        out[str(asset.id)] = poll_asset_review(asset)
-    return out
+    """先把本团队未送审的人脸资产送出去,再轮询审核中状态。
+    前端基础资产趴定时打这里:用户不用去后台点送审,也不用等管理员。"""
+    ensure_review_drain_loop()
+    submit_unsubmitted_reviews(team=team, limit=20)
+    return poll_processing_reviews(team=team, limit=80)