pythonnet/src/runtime/iterator.cs at fix_stackoverflow · matthid/pythonnet · GitHub

44 lines (37 loc) · 1.31 KB

using System;

using System.Collections;

using System.Reflection;

namespace Python.Runtime

{

//========================================================================

// Implements a generic Python iterator for IEnumerable objects and

// managed array objects. This supports 'for i in object:' in Python.

//========================================================================

internal class Iterator : ExtensionType

{

IEnumerator iter;

public Iterator(IEnumerator e) : base()

{

this.iter = e;

}

//====================================================================

// Implements support for the Python iteration protocol.

//====================================================================

public static IntPtr tp_iternext(IntPtr ob)

{

Iterator self = GetManagedObject(ob) as Iterator;

if (!self.iter.MoveNext())

{

Exceptions.SetError(Exceptions.StopIteration, Runtime.PyNone);

return IntPtr.Zero;

}

object item = self.iter.Current;

return Converter.ToPythonImplicit(item);

}

public static IntPtr tp_iter(IntPtr ob)

{

Runtime.Incref(ob);

return ob;

}

Expand file tree