Comparable for SchedParam

youknowone · youknowone · commit 5d6d72f9d560 · 2026-02-03T11:18:50.000+09:00
diff --git a/crates/vm/src/stdlib/posix.rs b/crates/vm/src/stdlib/posix.rs
@@ -22,13 +22,13 @@ pub mod module {
         builtins::{PyDictRef, PyInt, PyListRef, PyStr, PyTupleRef, PyType},
         convert::{IntoPyException, ToPyObject, TryFromObject},
         exceptions::OSErrorBuilder,
-        function::{Either, KwArgs, OptionalArg},
+        function::{Either, KwArgs, OptionalArg, PyComparisonValue},
         ospath::{OsPath, OsPathOrFd},
         stdlib::os::{
             _os, DirFd, FollowSymlinks, SupportFunc, TargetIsDirectory, fs_metadata,
             warn_if_bool_fd,
         },
-        types::{Constructor, Representable},
+        types::{Comparable, Constructor, PyComparisonOp, Representable},
     };
     #[cfg(any(
         target_os = "android",
@@ -904,7 +904,7 @@ pub mod module {
         }
     }
 
-    #[pyclass(with(Constructor, Representable))]
+    #[pyclass(with(Comparable, Constructor, Representable))]
     impl SchedParam {
         #[pygetset]
         fn sched_priority(&self, vm: &VirtualMachine) -> PyObjectRef {
@@ -988,6 +988,24 @@ pub mod module {
         }
     }
 
+    impl Comparable for SchedParam {
+        fn cmp(
+            zelf: &Py<Self>,
+            other: &crate::PyObject,
+            op: PyComparisonOp,
+            vm: &VirtualMachine,
+        ) -> PyResult<PyComparisonValue> {
+            op.eq_only(|| {
+                let other = class_or_notimplemented!(Self, other);
+                let eq = vm.identical_or_equal(
+                    &zelf.sched_priority,
+                    &other.sched_priority,
+                )?;
+                Ok(eq.into())
+            })
+        }
+    }
+
     #[cfg(any(
         target_os = "linux",
         target_os = "netbsd",
