Add __repr__ to the SysinspectReturn

tinythings · Nov 26, 2024 · 16d86cf · 16d86cf
1 parent 89ade4c
commit 16d86cf
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/libsysinspect/src/pylang/pylib/pysystem.rs b/libsysinspect/src/pylang/pylib/pysystem.rs
@@ -133,6 +133,15 @@ pub mod syscore {
         fn set_data(&self, data: PyRef<PyDict>) {
             self.inner.lock().data = data;
         }
+
+        #[pymethod(name = "__repr__")]
+        fn str(this: PyObjectRef, _vm: &VirtualMachine) -> String {
+            if let Some(x) = this.downcast_ref::<SysinspectReturn>() {
+                return format!("<SysinspectReturn: {:?}>", x);
+            }
+
+            "".to_string()
+        }
     }
 
     #[pyfunction]