@miravia_test/miravia-publish-skill 0.1.2 → 0.1.4

package/dist/scripts/product_skill.py

@@ -31,7 +31,52 @@ import i18n
 import skill_logger
 import telemetry
 from miravia_client import MiraviaClient, MiraviaError
-from modules.publish_module import PublishModule
+from publish_module import PublishModule
+
+
+def _print_category_tree(
+    tree: Any,
+    indent: int = 0,
+    permissions: Optional[Dict[int, bool]] = None,
+) -> None:
+    """Recursively print a category tree with indentation.
+
+    Handles both dict and list structures, showing category name, ID,
+    and permission status at each level. Limits depth to 2 levels.
+    """
+    max_depth = 2
+    if indent >= max_depth:
+        return
+
+    if permissions is None:
+        permissions = {}
+
+    if isinstance(tree, dict):
+        # Try common tree structures
+        categories = tree.get("categories") or tree.get("data") or tree.get("children") or []
+        if isinstance(categories, list):
+            for cat in categories:
+                _print_category_tree(cat, indent, permissions)
+        else:
+            # Single category node
+            name = tree.get("name") or tree.get("category_name") or ""
+            cid = tree.get("category_id") or tree.get("id") or ""
+            prefix = "  " * indent
+            if name:
+                # Show permission status
+                if cid and int(cid) in permissions:
+                    has_perm = permissions[int(cid)]
+                    perm_marker = "[OK]" if has_perm else "[NO PERMISSION]"
+                    print(f"{prefix}• {name} (ID: {cid}) {perm_marker}")
+                else:
+                    print(f"{prefix}• {name} (ID: {cid})" if cid else f"{prefix}• {name}")
+            children = tree.get("children") or tree.get("sub_categories") or []
+            if isinstance(children, list):
+                for child in children:
+                    _print_category_tree(child, indent + 1, permissions)
+    elif isinstance(tree, list):
+        for item in tree:
+            _print_category_tree(item, indent, permissions)
 
 
 def _parse_args() -> argparse.Namespace:
@@ -118,6 +163,11 @@ def _parse_args() -> argparse.Namespace:
         action="store_true",
         help="Seller has confirmed the authorization request; skip the authorization step",
     )
+    parser.add_argument(
+        "--decline-auth",
+        action="store_true",
+        help="Seller explicitly declines authorization; stop the flow and prompt for re-upload",
+    )
     parser.add_argument(
         "--output-schema",
         action="store_true",
@@ -201,6 +251,24 @@ def _print_conversational_result(result: Dict[str, Any]) -> None:
         print(result.get("hint", ""))
         return
 
+    # Multiple products detected in Excel
+    if status == "multiple_products_detected":
+        print(result.get("message", ""))
+        print()
+        for product in result.get("product_summaries", []):
+            print(
+                i18n._("pub_multiple_products_product_line").format(
+                    index=product.get("index", ""),
+                    title=product.get("title", "") or "(no title)",
+                    price=product.get("price", "") or "(no price)",
+                    sku=product.get("sku", "") or "(no SKU)",
+                )
+            )
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
     # Interactive preview (brand / attributes)
     if status == "need_brand":
         print(result.get("message", ""))
@@ -233,6 +301,58 @@ def _print_conversational_result(result: Dict[str, Any]) -> None:
             print(f"• {action}")
         return
 
+    # Authorization cancelled
+    if status == "authorization_cancelled":
+        print(result.get("title", ""))
+        print()
+        print(result.get("message", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Unsupported file format
+    if status == "unsupported_format":
+        print(result.get("message", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # File empty or corrupted
+    if status == "file_empty_or_corrupted":
+        print(result.get("message", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Invalid URL
+    if status == "invalid_url":
+        print(result.get("message", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Generic file read error
+    if status == "file_read_error":
+        print(result.get("message", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Title invalid with format suggestion
+    if status == "title_invalid":
+        print(result.get("message", ""))
+        print()
+        print(result.get("format_suggestion", ""))
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
     # Category mandatory attributes missing (AI extraction flow)
     if status == "need_category_attributes":
         print(result.get("title", ""))
@@ -257,6 +377,16 @@ def _print_conversational_result(result: Dict[str, Any]) -> None:
 
     if status == "need_category":
         print(result.get("message", ""))
+        # Display category tree if available
+        category_tree = result.get("category_tree")
+        category_permissions = result.get("category_permissions", {})
+        if category_tree:
+            print()
+            print("--- Available Categories ---")
+            _print_category_tree(category_tree, permissions=category_permissions)
+        tree_link = result.get("category_tree_link")
+        if tree_link:
+            print(f"\n{tree_link}")
         print()
         for choice in result.get("choices", []):
             name = choice.get("name") or choice.get("category_name")
@@ -267,6 +397,19 @@ def _print_conversational_result(result: Dict[str, Any]) -> None:
             print(f"• {action}")
         return
 
+    # Category permission required
+    if status == "need_category_permission":
+        print(result.get("title", ""))
+        print()
+        print(result.get("message", ""))
+        apply_link = result.get("apply_link")
+        if apply_link:
+            print(f"\n{apply_link}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
     # Preview awaiting confirmation
     if status == "preview":
         print(result.get("message", "请确认以下发布操作："))
@@ -354,19 +497,113 @@ def _print_conversational_result(result: Dict[str, Any]) -> None:
             print(f"• {action}")
         return
 
+    # Compliance fields missing — suggest Draft
+    if status == "compliance_missing_suggest_draft":
+        print(result.get("message", ""))
+        print()
+        for issue in result.get("compliance_issues", []):
+            print(f"  • {issue}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # GPSR fields missing — suggest Draft
+    if status == "gpsr_missing_suggest_draft":
+        print(result.get("message", ""))
+        print()
+        for issue in result.get("gpsr_issues", []):
+            print(f"  • {issue}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Post-to-inactive fields missing — suggest Draft
+    if status == "post_to_inactive_missing_suggest_draft":
+        print(result.get("message", ""))
+        print()
+        for field in result.get("missing_fields", []):
+            print(f"  • {field}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Special category rules — hard block (invalid PPU, ISBN missing for books)
+    if status == "special_category_block":
+        print(result.get("message", ""))
+        print()
+        for issue in result.get("block_issues", []):
+            print(f"  • {issue}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # DBM limit exceeded with no DBS permission — hard block
+    if status == "dbm_limit_exceeded_no_dbs":
+        print(result.get("message", ""))
+        print()
+        for reason in result.get("reasons", []):
+            print(f"  • {reason}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # SKU already exists — concurrency conflict
+    if status == "sku_already_exists":
+        print(result.get("message", ""))
+        code = result.get("code", "")
+        if code:
+            print(f"\n错误码：{code}")
+        hint = result.get("hint")
+        if hint:
+            print(f"\n{hint}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
+    # Special category rules — suggest Draft (PPU missing, food attrs missing)
+    if status == "special_category_suggest_draft":
+        print(result.get("message", ""))
+        print()
+        for field in result.get("missing_fields", []):
+            print(f"  • {field}")
+        print()
+        for action in result.get("next_actions", []):
+            print(f"• {action}")
+        return
+
     # Field extraction summary / draft question
     if status == "need_field_summary":
         print(result.get("title", ""))
         print(result.get("message", ""))
         print()
-        for line in result.get("summary", []):
-            print(f"  {line}")
-        key_missing = result.get("key_missing", [])
-        if key_missing:
+
+        # Display summary in table-like format with Mandatory/Conditional sections
+        summary = result.get("summary", [])
+        for line in summary:
+            print(line)
+
+        # Show missing fields by category
+        mandatory_missing = result.get("mandatory_missing", [])
+        conditional_missing = result.get("conditional_missing", [])
+
+        if mandatory_missing:
             print()
-            print("关键缺失字段：")
-            for field in key_missing:
-                print(f"  • {field}")
+            print("⚠ 缺失必填字段（阻断发布）：")
+            for field in mandatory_missing:
+                print(f"   {field}")
+
+        if conditional_missing:
+            print()
+            print("⚡ 缺失条件必填字段（可转 Draft）：")
+            for field in conditional_missing:
+                print(f"  ○ {field}")
+
         country_note = result.get("country_price_note")
         if country_note:
             print()
@@ -499,6 +736,19 @@ def main() -> None:
     # Authorization check (flow step ii): before any API calls, the seller
     # must confirm that they authorize the skill to read profile, access
     # product data, and update product data.
+    if getattr(args, "decline_auth", False):
+        auth_payload = {
+            "status": "authorization_cancelled",
+            "title": i18n._("pub_authorization_title"),
+            "message": i18n._("pub_authorization_cancelled"),
+            "next_actions": [i18n._("pub_authorization_confirm_next")],
+        }
+        if args.format == "conversational":
+            _print_conversational_result(auth_payload)
+        else:
+            print(json.dumps(auth_payload, indent=2, ensure_ascii=False))
+        return
+
     if not skip_credential_check and args.scenario and not getattr(args, "authorized", False):
         auth_payload = {
             "status": "need_authorization",