pythonnet/src/runtime/Types/Iterator.cs at v3.0.1 · pythonnet/pythonnet · GitHub

History

51 lines (46 loc) · 1.43 KB

Raw

using System;

using System.Collections;

namespace Python.Runtime

{

/// <summary>

/// Implements a generic Python iterator for IEnumerable objects and

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

/// </summary>

internal class Iterator : ExtensionType

{

private readonly IEnumerator iter;

private readonly Type elemType;

public Iterator(IEnumerator e, Type elemType)

{

iter = e;

this.elemType = elemType;

}

/// <summary>

/// Implements support for the Python iteration protocol.

/// </summary>

public static NewReference tp_iternext(BorrowedReference ob)

{

var self = (Iterator)GetManagedObject(ob)!;

try

{

if (!self.iter.MoveNext())

{

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

return default;

}

catch (Exception e)

{

if (e.InnerException != null)

{

e = e.InnerException;

}

Exceptions.SetError(e);

return default;

}

object item = self.iter.Current;

return Converter.ToPython(item, self.elemType);

}

public static NewReference tp_iter(BorrowedReference ob) => new (ob);

}

pythonnet/src/runtime/Types/Iterator.cs at v3.0.1 · pythonnet/pythonnet · GitHub

Uh oh!

Expand file tree

Breadcrumbs

Iterator.cs

Latest commit

History

Breadcrumbs

Iterator.cs

File metadata and controls