lib/pycall/utils.rb in pycall-0.1.0.alpha.20170419b vs lib/pycall/utils.rb in pycall-0.1.0.alpha.20170419

@@ -79,22 +79,12 @@
       @type.(pyobj)
     end
 
     def with(ctx)
       __exit__ = PyCall.getattr(ctx, :__exit__)
-      begin
-        yield PyCall.getattr(ctx,:__enter__).()
-      rescue Exception => err
-        if err.kind_of? PyError
-          exit_value = __exit__.(err.type, err.value, err.traceback)
-        else
-          # TODO: support telling what exception has been catched
-          exit_value = __exit__.(PyCall.None, PyCall.None, PyCall.None)
-        end
-        raise err unless exit_value.equal? true
-      else
-       __exit__.(PyCall.None, PyCall.None, PyCall.None)
-      end
+      yield PyCall.getattr(ctx,:__enter__).()
+    ensure
+      __exit__.()
     end
 
     def format_traceback(pyobj)
       @format_tb ||= import_module('traceback').format_tb
       @format_tb.(pyobj)