enable creating PyIter from existing PyObject

lostmsu · lostmsu · commit 61f694271e16 · 2020-10-17T13:06:33.000-07:00
diff --git a/src/runtime/pyiter.cs b/src/runtime/pyiter.cs
@@ -24,6 +24,19 @@ public class PyIter : PyObject, IEnumerator<PyObject>
         public PyIter(IntPtr ptr) : base(ptr)
         {
         }
+        /// <summary>
+        /// Creates new <see cref="PyIter"/> from an untyped reference to Python object.
+        /// </summary>
+        public PyIter(PyObject pyObject) : base(FromPyObject(pyObject)) { }
+        static BorrowedReference FromPyObject(PyObject pyObject) {
+            if (pyObject is null) throw new ArgumentNullException(nameof(pyObject));
+
+            if (!Runtime.PyIter_Check(pyObject.Reference))
+                throw new ArgumentException("Object does not support iterator protocol");
+
+            return pyObject.Reference;
+        }
+
         internal PyIter(BorrowedReference reference) : base(reference) { }
 
         protected override void Dispose(bool disposing)
