dfd

cmdlets/delete_relationship.py

@@ -15,6 +15,49 @@ from helper.local_library import LocalLibrarySearchOptimizer
 from config import get_local_storage_path
 
 
+def _refresh_relationship_view_if_current(target_hash: Optional[str], target_path: Optional[str], other: Optional[str], config: Dict[str, Any]) -> None:
+    """If the current subject matches the target, refresh relationships via get-relationship."""
+    try:
+        from cmdlets import get_relationship as get_rel_cmd  # type: ignore
+    except Exception:
+        return
+
+    try:
+        subject = ctx.get_last_result_subject()
+        if subject is None:
+            return
+
+        def norm(val: Any) -> str:
+            return str(val).lower()
+
+        target_hashes = [norm(v) for v in [target_hash, other] if v]
+        target_paths = [norm(v) for v in [target_path, other] if v]
+
+        subj_hashes: list[str] = []
+        subj_paths: list[str] = []
+        if isinstance(subject, dict):
+            subj_hashes = [norm(v) for v in [subject.get("hydrus_hash"), subject.get("hash"), subject.get("hash_hex"), subject.get("file_hash")] if v]
+            subj_paths = [norm(v) for v in [subject.get("file_path"), subject.get("path"), subject.get("target")] if v]
+        else:
+            subj_hashes = [norm(getattr(subject, f, None)) for f in ("hydrus_hash", "hash", "hash_hex", "file_hash") if getattr(subject, f, None)]
+            subj_paths = [norm(getattr(subject, f, None)) for f in ("file_path", "path", "target") if getattr(subject, f, None)]
+
+        is_match = False
+        if target_hashes and any(h in subj_hashes for h in target_hashes):
+            is_match = True
+        if target_paths and any(p in subj_paths for p in target_paths):
+            is_match = True
+        if not is_match:
+            return
+
+        refresh_args: list[str] = []
+        if target_hash:
+            refresh_args.extend(["-hash", target_hash])
+        get_rel_cmd._run(subject, refresh_args, config)
+    except Exception:
+        pass
+
+
 def _run(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
     """Delete relationships from files.
     
@@ -137,6 +180,7 @@ def _run(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
                     """, (file_id, json.dumps(relationships) if relationships else None))
                     
                     db.db.connection.commit()
+                    _refresh_relationship_view_if_current(None, str(file_path_obj), None, config)
                     deleted_count += 1
                     
             except Exception as exc: