Implement C API function '_PySet_NextEntry'

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/setobject.c

@@ -69,6 +69,20 @@ int PySet_Contains(PyObject *anyset, PyObject *key) {
     return UPCALL_CEXT_I(_jls_PySet_Contains, native_to_java(anyset), native_to_java(key));
 }
 
+UPCALL_ID(PySet_NextEntry);
+int _PySet_NextEntry(PyObject *set, Py_ssize_t *pos, PyObject **key, Py_hash_t *hash) {
+    PyObject *tresult = UPCALL_CEXT_O(_jls_PySet_NextEntry, native_to_java(set), *pos);
+    if (tresult == NULL) {
+        *key = NULL;
+        *hash = 0;
+    	return 0;
+    }
+    (*pos)++;
+    *key = PyTuple_GetItem(tresult, 0);
+    *hash = PyLong_AsSsize_t(PyTuple_GetItem(tresult, 1));
+    return 1;
+}
+
 UPCALL_ID(PySet_Pop);
 PyObject * PySet_Pop(PyObject *set) {
     return UPCALL_CEXT_O(_jls_PySet_Pop, native_to_java(set));

graalpython/lib-graalpython/python_cext.py

@@ -245,6 +245,19 @@ def PySet_Contains(anyset, item):
     return item in anyset
 
 
+@may_raise
+def PySet_NextEntry(anyset, pos):
+    curPos = 0
+    max = len(anyset)
+    if pos >= max:
+        return native_null
+    for key in anyset:
+        if curPos == pos:
+            return key, hash(key)
+        curPos = curPos + 1
+    return native_null
+
+
 @may_raise
 def PySet_Pop(anyset):
     if not isinstance(anyset, set):