Add direct member access to example

GaelicGabe · GaelicGabe · commit 340816d4351a · 2022-11-21T21:50:31.000+01:00
In the example demonstrating how to call between rust and python add
direct member access.
diff --git a/examples/call_between_rust_and_python.py b/examples/call_between_rust_and_python.py
@@ -7,6 +7,7 @@ def __init__(self, name):
 def python_callback():
     python_person = PythonPerson("Peter Python")
     rust_object = rust_function(42, "This is a python string", python_person)
+    print("Printing member 'numbers' from rust struct: ", rust_object.numbers)
     rust_object.print_in_rust_from_python()
 
 def take_string(string):
diff --git a/examples/call_between_rust_and_python.rs b/examples/call_between_rust_and_python.rs
@@ -30,6 +30,7 @@ pub fn main() {
 #[pymodule]
 mod rust_py_module {
     use super::*;
+    use rustpython_vm::{builtins::PyList, convert::ToPyObject, PyObjectRef};
 
     #[pyfunction]
     fn rust_function(
@@ -45,16 +46,35 @@ string: {},
 python_person.name: {}",
             num, s, python_person.name
         );
-        Ok(RustStruct)
+        Ok(RustStruct {
+            numbers: NumVec(vec![1, 2, 3, 4]),
+        })
+    }
+
+    #[derive(Debug, Clone)]
+    struct NumVec(Vec<i32>);
+
+    impl ToPyObject for NumVec {
+        fn to_pyobject(self, vm: &VirtualMachine) -> PyObjectRef {
+            let list = self.0.into_iter().map(|e| vm.new_pyobj(e)).collect();
+            PyList::new_ref(list, vm.as_ref()).to_pyobject(vm)
+        }
     }
 
     #[pyattr]
     #[pyclass(module = "rust_py_module", name = "RustStruct")]
     #[derive(Debug, PyPayload)]
-    struct RustStruct;
+    struct RustStruct {
+        numbers: NumVec,
+    }
 
     #[pyclass]
     impl RustStruct {
+        #[pygetset]
+        fn numbers(&self) -> NumVec {
+            self.numbers.clone()
+        }
+
         #[pymethod]
         fn print_in_rust_from_python(&self) {
             println!("Calling a rust method from python");
