Missing wrappers for iterators and generators about pycall.rb HOT 4 OPEN

module PyCall
  module_function

  def enum(obj)
    gen = obj.__iter__
    Enumerator.new do |y|
      begin
        yield gen.__next__ while true
        obj
      rescue PyError, err
        # ignore StopIteration exception
        return obj if err.type.match?(LibPython::API::PyExc_StopIteration)
        raise
      end
    end
  end
end

module PyCall
  module_function

  def enum(generator)
    Enumerator.new do |enum|
      begin
        loop do
          enum << PyCall.builtins.next(generator)
        end
      rescue PyError => error
        raise err unless error.type.to_s.match?("StopIteration")
      end
    end
  end
end