Add setattr/setitem

coolreader18 · coolreader18 · commit e83e3ccda332 · 2019-04-19T08:08:31.000-05:00
diff --git a/wasm/lib/src/objjsvalue.rs b/wasm/lib/src/objjsvalue.rs
@@ -21,6 +21,10 @@ fn get_prop(value: JsValue, name: &str, vm: &VirtualMachine) -> Option<PyObjectR
     }
 }
 
+fn set_prop(value: &JsValue, name: &str, val: PyObjectRef, vm: &VirtualMachine) {
+    Reflect::set(value, &name.into(), &convert::py_to_js(vm, val)).expect("Reflect failed");
+}
+
 #[pyclass(name = "JsValue")]
 #[derive(Debug)]
 pub struct PyJsValue {
@@ -54,6 +58,11 @@ impl PyJsValue {
         })
     }
 
+    #[pymethod(name = "__setattr__")]
+    fn setattr(&self, attr_name: PyStringRef, val: PyObjectRef, vm: &VirtualMachine) {
+        set_prop(self.value(), attr_name.as_str(), val, vm);
+    }
+
     #[pymethod(name = "__repr__")]
     fn repr(&self, _vm: &VirtualMachine) -> String {
         format!("{:?}", self.value())
@@ -131,6 +140,11 @@ impl PyJsProps {
         get_prop(self.value.clone(), item_name.as_str(), vm)
             .ok_or_else(|| vm.new_key_error(format!("{:?}", item_name.as_str())))
     }
+
+    #[pymethod(name = "__setitem__")]
+    fn setitem(&self, item_name: PyStringRef, val: PyObjectRef, vm: &VirtualMachine) {
+        set_prop(&self.value, item_name.as_str(), val, vm);
+    }
 }
 
 pub fn init(ctx: &PyContext) {
