Add json_load

zsaladin · zsaladin · commit 73549a5e6f6f · 2019-08-15T16:09:36.000+09:00
diff --git a/vm/src/stdlib/json.rs b/vm/src/stdlib/json.rs
@@ -39,6 +39,11 @@ pub fn json_loads(string: PyStringRef, vm: &VirtualMachine) -> PyResult {
     })
 }
 
+pub fn json_load(fp: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+    let result = vm.call_method(&fp, "read", vec![])?;
+    json_loads(result.downcast()?, vm)
+}
+
 pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
     let ctx = &vm.ctx;
 
@@ -53,6 +58,7 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
         "dumps" => ctx.new_rustfunc(json_dumps),
         "dump" => ctx.new_rustfunc(json_dump),
         "loads" => ctx.new_rustfunc(json_loads),
+        "load" => ctx.new_rustfunc(json_load),
         "JSONDecodeError" => json_decode_error
     })
 }
