io.__del__

youknowone · youknowone · commit bc02c5f619e9 · 2020-09-13T06:12:16.000+09:00
diff --git a/vm/src/stdlib/io.rs b/vm/src/stdlib/io.rs
@@ -408,6 +408,11 @@ fn io_base_cm_enter(instance: PyObjectRef) -> PyObjectRef {
     instance
 }
 
+fn io_base_cm_del(instance: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
+    vm.call_method(&instance, "close", vec![])?;
+    Ok(())
+}
+
 fn io_base_cm_exit(instance: PyObjectRef, _args: PyFuncArgs, vm: &VirtualMachine) -> PyResult<()> {
     vm.call_method(&instance, "close", vec![])?;
     Ok(())
@@ -1269,6 +1274,7 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
     // IOBase the abstract base class of the IO Module
     let io_base = py_class!(ctx, "_IOBase", &ctx.types.object_type, {
         "__enter__" => ctx.new_method(io_base_cm_enter),
+        "__del__" => ctx.new_method(io_base_cm_del),
         "__exit__" => ctx.new_method(io_base_cm_exit),
         "seekable" => ctx.new_method(io_base_seekable),
         "readable" => ctx.new_method(io_base_readable),
