[add] pyfor iterate clause

py4cl2-cffi.asd

@@ -55,6 +55,7 @@
                (:file "arg-list")
                (:file "import-export")
                (:file "lisp-classes")
+               (:file "iterate")
                (:file "do-after-load"))
   :perform (test-op (o c)
              (declare (ignore o c))

src/iterate.lisp

@@ -0,0 +1,19 @@
+(in-package :py4cl2-cffi)
+
+(declaim (inline pyerror-stop-iteration-p))
+(defun pyerror-stop-iteration-p (simple-pyerror)
+  (declare (optimize speed)
+           (type simple-pyerror simple-pyerror))
+  (string= "StopIteration" (slot-value simple-pyerror 'type)))
+
+(deftype pyerror-stop-iteration ()
+  `(and simple-pyerror
+        (satisfies pyerror-stop-iteration-p)))
+
+(defmacro-clause (PYFOR var IN pyobject-wrapper)
+  ;; TODO: Tests and docstrings
+  (let ((pygen (gensym "PYGENERATOR")))
+    `(progn
+       (with ,pygen = (pycall "iter" ,pyobject-wrapper))
+       (for ,var = (handler-case (pycall "next" ,pygen)
+                     (pyerror-stop-iteration () (finish)))))))

src/package.lisp

@@ -66,6 +66,8 @@
            #:+py-none-pointer+
 
            #:python-getattr
-           #:python-setattr))
+           #:python-setattr
+
+           #:pyfor))
 
 (in-package :py4cl2-cffi)