Merge pull request #5259 from chu11/python_joblist_jobid

python: allow JobID to take a JobID

Author: mergify[bot]

src/bindings/python/flux/job/JobID.py

@@ -53,6 +53,8 @@ class JobID(int):
     """
 
     def __new__(cls, value):
+        if isinstance(value, JobID):
+            return cls(value.orig_str)
         if isinstance(value, int):
             jobid = value
         else:

src/bindings/python/flux/job/list.py

@@ -13,6 +13,7 @@
 
 import flux.constants
 from flux.future import WaitAllFuture
+from flux.job import JobID
 from flux.job.info import JobInfo
 from flux.rpc import RPC
 
@@ -211,7 +212,7 @@ def __init__(
         self.since = since
         self.name = name
         self.queue = queue
-        self.ids = ids
+        self.ids = list(map(JobID, ids)) if ids else None
         self.errors = []
         for fname in filters:
             for x in fname.split(","):

t/python/t0010-job.py

@@ -468,6 +468,12 @@ def test_24_jobid(self):
                     # Ensure encode back to same type works
                     self.assertEqual(getattr(jobid, key), test[key])
 
+        # JobID can also take a JobID
+        jobid1 = job.JobID(1234)
+        jobid2 = job.JobID(jobid1)
+        self.assertEqual(jobid1, jobid2)
+        self.assertEqual(jobid2.orig, "1234")
+
     def test_25_job_list_attrs(self):
         expected_attrs = [
             "userid",