flux-jobs: add -i, --include=HOSTS|RANKS option

Problem: There is no easy way to restrict the output of flux-jobs(1)
based on a set of hosts or ranks, but this is basic and useful
functionality.

Add a new `-i, --include=` optiont that takes a hostlist or idset of
targets, and adds a constraint to the JobList query that restricts
matching jobs to those that were allocated the provided targets.

Fixes #6202

Author: grondo

src/cmd/flux-jobs.py

@@ -17,6 +17,8 @@
 import sys
 
 import flux.constants
+from flux.hostlist import Hostlist
+from flux.idset import IDset
 from flux.job import JobID, JobInfo, JobInfoFormat, JobList, job_fields_to_attrs
 from flux.job.stats import JobStats
 from flux.util import (
@@ -156,6 +158,16 @@ def fetch_jobs_flux(args, fields, flux_handle=None):
     if not args.filter:
         args.filter = {"pending", "running"}
 
+    constraint = None
+    if args.include:
+        try:
+            constraint = {"ranks": [IDset(args.include).encode()]}
+        except ValueError:
+            try:
+                constraint = {"hostlist": [Hostlist(args.include).encode()]}
+            except ValueError:
+                raise ValueError(f"-i/--include: invalid targets: {args.include}")
+
     jobs_rpc = JobList(
         flux_handle,
         ids=args.jobids,
@@ -166,6 +178,7 @@ def fetch_jobs_flux(args, fields, flux_handle=None):
         since=since,
         name=args.name,
         queue=args.queue,
+        constraint=constraint,
     )
 
     jobs = jobs_rpc.jobs()
@@ -270,6 +283,14 @@ def parse_args():
         metavar="QUEUE,...",
         help="Limit output to specific queue or queues",
     )
+    parser.add_argument(
+        "-i",
+        "--include",
+        type=str,
+        metavar="HOSTS|RANKS",
+        help="Limit output to jobs that were allocated to the specified "
+        + "HOSTS or RANKS provided as a hostlist or idset.",
+    )
     parser.add_argument(
         "-o",
         "--format",