110 lines
2.9 KiB
Python
110 lines
2.9 KiB
Python
"""Shared product-status rules for the collection, generation, and apply flows."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import re
|
||
|
||
|
||
STATUS_NORMAL = "normal"
|
||
STATUS_UNLISTED = "unlisted"
|
||
STATUS_REVIEWING = "reviewing"
|
||
STATUS_UNKNOWN = "unknown"
|
||
|
||
VALID_PRODUCT_STATUSES = frozenset(
|
||
{
|
||
STATUS_NORMAL,
|
||
STATUS_UNLISTED,
|
||
STATUS_REVIEWING,
|
||
STATUS_UNKNOWN,
|
||
}
|
||
)
|
||
|
||
PRODUCT_STATUS_LABELS = {
|
||
STATUS_NORMAL: "正常",
|
||
STATUS_UNLISTED: "未上架",
|
||
STATUS_REVIEWING: "审核中",
|
||
STATUS_UNKNOWN: "状态未知",
|
||
}
|
||
|
||
_WHITESPACE_RE = re.compile(r"\s+")
|
||
_NOTE_LIMIT = 2000
|
||
|
||
|
||
def normalize_status(value) -> str:
|
||
"""Return a stable status code; legacy and invalid values are unknown."""
|
||
|
||
normalized = str(value or "").strip().lower()
|
||
if normalized in VALID_PRODUCT_STATUSES:
|
||
return normalized
|
||
return STATUS_UNKNOWN
|
||
|
||
|
||
def status_label(value) -> str:
|
||
return PRODUCT_STATUS_LABELS[normalize_status(value)]
|
||
|
||
|
||
def is_normal(value) -> bool:
|
||
return normalize_status(value) == STATUS_NORMAL
|
||
|
||
|
||
def is_known_abnormal(value) -> bool:
|
||
return normalize_status(value) in {STATUS_UNLISTED, STATUS_REVIEWING}
|
||
|
||
|
||
def normalize_text(value) -> str:
|
||
return _WHITESPACE_RE.sub(" ", str(value or "").strip())
|
||
|
||
|
||
def classify_alerts(alerts) -> dict:
|
||
"""Classify normalized EDS warning-alert data without retaining page HTML."""
|
||
|
||
if not isinstance(alerts, list):
|
||
return _snapshot(STATUS_UNKNOWN, "")
|
||
if not alerts:
|
||
return _snapshot(STATUS_NORMAL, "")
|
||
|
||
first_note = ""
|
||
valid_alert_seen = False
|
||
for alert in alerts:
|
||
if not isinstance(alert, dict):
|
||
continue
|
||
valid_alert_seen = True
|
||
title = normalize_text(alert.get("title"))
|
||
description = normalize_text(alert.get("description"))
|
||
note = _alert_note(title, description)
|
||
if not first_note:
|
||
first_note = note
|
||
if title in {"審核中", "审核中"}:
|
||
return _snapshot(STATUS_REVIEWING, note)
|
||
if "您的商品未上架" in title:
|
||
return _snapshot(STATUS_UNLISTED, note)
|
||
|
||
if not valid_alert_seen:
|
||
return _snapshot(STATUS_UNKNOWN, "")
|
||
return _snapshot(STATUS_UNKNOWN, first_note)
|
||
|
||
|
||
def partition_tasks(tasks) -> dict:
|
||
"""Return tasks grouped by normalized status for callers' run planners."""
|
||
|
||
grouped = {status: [] for status in VALID_PRODUCT_STATUSES}
|
||
for task in tasks:
|
||
grouped[normalize_status(getattr(task, "product_status", None))].append(task)
|
||
return grouped
|
||
|
||
|
||
def _alert_note(title: str, description: str) -> str:
|
||
parts = []
|
||
if title:
|
||
parts.append(f"标题：{title}")
|
||
if description:
|
||
parts.append(f"说明：{description}")
|
||
return "；".join(parts)[:_NOTE_LIMIT]
|
||
|
||
|
||
def _snapshot(status: str, note: str) -> dict:
|
||
return {
|
||
"product_status": normalize_status(status),
|
||
"product_status_note": str(note or "")[:_NOTE_LIMIT] or None,
|
||
}
|