dfd

cmdlets/add_relationship.py

@@ -145,6 +145,49 @@ def _resolve_king_reference(king_arg: str) -> Optional[str]:
     return None
 
 
+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
+
+
 @register(["add-relationship", "add-rel"])  # primary name and alias
 def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
     """Associate file relationships in Hydrus.
@@ -253,6 +296,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
                         f"[add-relationship] Set {rel_type} relationship: {file_hash} <-> {king_hash}",
                         file=sys.stderr
                     )
+                    _refresh_relationship_view_if_current(file_hash, file_path_from_result, king_hash, config)
                 except Exception as exc:
                     log(f"Failed to set relationship: {exc}", file=sys.stderr)
                     return 1
@@ -280,6 +324,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
                             f"[add-relationship] Set {rel_type} relationship: {file_hash} <-> {existing_king}",
                             file=sys.stderr
                         )
+                        _refresh_relationship_view_if_current(file_hash, file_path_from_result, existing_king, config)
                     except Exception as exc:
                         log(f"Failed to set relationship: {exc}", file=sys.stderr)
                         return 1
@@ -300,6 +345,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
                         with LocalLibrarySearchOptimizer(local_storage_path) as db:
                             db.set_relationship(file_path_obj, king_file_path, rel_type)
                         log(f"Set {rel_type} relationship: {file_path_obj.name} -> {king_file_path.name}", file=sys.stderr)
+                        _refresh_relationship_view_if_current(None, str(file_path_obj), str(king_file_path), config)
                     else:
                         log(f"King file not found or invalid: {king_hash}", file=sys.stderr)
                         return 1
@@ -323,6 +369,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
                             with LocalLibrarySearchOptimizer(local_storage_path) as db:
                                 db.set_relationship(file_path_obj, Path(king_path), rel_type)
                             log(f"Set {rel_type} relationship: {file_path_obj.name} -> {Path(king_path).name}", file=sys.stderr)
+                            _refresh_relationship_view_if_current(None, str(file_path_obj), str(king_path), config)
                         except Exception as exc:
                             log(f"Failed to set relationship: {exc}", file=sys.stderr)
                             return 1