dfdf

cmdlet/get_relationship.py

@@ -29,12 +29,12 @@ from Store import Store
 CMDLET = Cmdlet(
     name="get-relationship",
     summary="Print relationships for the selected file (Hydrus or Local).",
-    usage="get-relationship [-hash <sha256>]",
+    usage="get-relationship [-query \"hash:<sha256>\"]",
     alias=[
         "get-rel",
     ],
     arg=[
-        SharedArgs.HASH,
+        SharedArgs.QUERY,
         SharedArgs.STORE,
     ],
     detail=[
@@ -48,20 +48,28 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
         log(f"Cmdlet: {CMDLET.name}\nSummary: {CMDLET.summary}\nUsage: {CMDLET.usage}")
         return 0
 
-    # Parse -hash and -store override
-    override_hash: str | None = None
+    # Parse -query and -store override
+    override_query: str | None = None
     override_store: str | None = None
     args_list = list(_args)
     i = 0
     while i < len(args_list):
         a = args_list[i]
         low = str(a).lower()
-        if low in {"-hash", "--hash", "hash"} and i + 1 < len(args_list):
-            override_hash = str(args_list[i + 1]).strip()
-            break
+        if low in {"-query", "--query", "query"} and i + 1 < len(args_list):
+            override_query = str(args_list[i + 1]).strip()
+            i += 2
+            continue
         if low in {"-store", "--store", "store"} and i + 1 < len(args_list):
             override_store = str(args_list[i + 1]).strip()
+            i += 2
+            continue
         i += 1
+
+    override_hash: str | None = sh.parse_single_hash_query(override_query) if override_query else None
+    if override_query and not override_hash:
+        log("get-relationship requires -query \"hash:<sha256>\"", file=sys.stderr)
+        return 1
     
     # Handle @N selection which creates a list
     # This cmdlet is single-subject; require disambiguation when multiple items are provided.
@@ -69,7 +77,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
         if len(result) == 0:
             result = None
         elif len(result) > 1 and not override_hash:
-            log("get-relationship expects a single item; select one row (e.g. @1) or pass -hash", file=sys.stderr)
+            log("get-relationship expects a single item; select one row (e.g. @1) or pass -query \"hash:<sha256>\"", file=sys.stderr)
             return 1
         else:
             result = result[0]
@@ -439,8 +447,7 @@ def _run(result: Any, _args: Sequence[str], config: Dict[str, Any]) -> int:
         pipeline_results.append(res_obj)
         
         # Set selection args
-        # If it has a path, we can use it directly. If hash, maybe get-file -hash?
-        table.set_row_selection_args(i, ["-store", str(item['store']), "-hash", item['hash']])
+        table.set_row_selection_args(i, ["-store", str(item['store']), "-query", f"hash:{item['hash']}"])
 
     ctx.set_last_result_table(table, pipeline_results)
     print(table)