From 2ff670a9ae1066dac32f23975177b80d0ac05e7c Mon Sep 17 00:00:00 2001 From: Daniel Gradman-Svendsen Date: Wed, 29 Jul 2026 15:06:37 +0200 Subject: [PATCH] Read SAB cat as category Treat SAB history cat/category fields equivalently for #20. --- importarr/readiness.py | 3 ++- tests/test_sab_readiness.py | 5 +++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/importarr/readiness.py b/importarr/readiness.py index 2848a41..fe0031e 100644 --- a/importarr/readiness.py +++ b/importarr/readiness.py @@ -30,7 +30,8 @@ def classify_history_item(item: dict[str, Any], active_nzo_ids: set[str], catego nzo_id = str(item.get("nzo_id") or item.get("nzoid") or "") if not force_status and nzo_id and nzo_id in active_nzo_ids: return Readiness("processing", "SAB job is still present in queue") - if str(item.get("category") or "") != category: + item_category = str(item.get("category") or item.get("cat") or "") + if item_category != category: return Readiness("ignored", "SAB category is not owned by Importarr") status = str(item.get("status") or "") if not force_status and status == "Failed": diff --git a/tests/test_sab_readiness.py b/tests/test_sab_readiness.py index 2c7463f..e12856d 100644 --- a/tests/test_sab_readiness.py +++ b/tests/test_sab_readiness.py @@ -22,6 +22,11 @@ def test_wrong_category_ignored(): assert result.state == "ignored" +def test_sab_cat_field_is_treated_as_category(): + result = classify_history_item(item(category=None, cat="manual"), set(), "manual", ROOT) + assert result.ready + + def test_queue_item_not_ready(): result = classify_history_item(item(), {"1"}, "manual", ROOT) assert result.state == "processing"