Merge pull request #1482 from losttech/Mini/PyIterFromPyObj

filmor · web-flow · commit 191bc89364ed · 2021-07-08T01:34:15.000+02:00
Create PyIter from existing PyObject
diff --git a/src/runtime/pyiter.cs b/src/runtime/pyiter.cs
@@ -25,6 +25,22 @@ public PyIter(IntPtr ptr) : base(ptr)
         {
         }
 
+        /// <summary>
+        /// Creates new <see cref="PyIter"/> from an untyped reference to Python object.
+        /// The object must support iterator protocol.
+        /// </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) { }
+
         /// <summary>
         /// PyIter factory function.
         /// </summary>
