Changeset 1266

Timestamp:

Mar 20, 2014, 12:41:43 PM (10 years ago)

Author:

hodgestar

Message:

Add support for iterator arguments to _speedups Markup.join implementation so that it matches the Python implementation (fixes #574).

Location:

trunk/genshi

Files:

• _speedups.c (modified) (1 diff)
• tests/core.py (modified) (1 diff)

trunk/genshi/_speedups.c

@@ -243 +243 @@
 {
     static char *kwlist[] = {"seq", "escape_quotes", 0};
-    PyObject *seq = NULL, *seq2, *tmp, *tmp2;
+    PyObject *seq = NULL, *seq2, *it, *tmp, *tmp2;
     char quotes = 1;
-    Py_ssize_t n;
-    int i;
 
     if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|b", kwlist, &seq, &quotes)) {
         return NULL;
     }
-    if (!PySequence_Check(seq)) {
-        return NULL;
-    }
-    n = PySequence_Size(seq);
-    if (n < 0) {
-        return NULL;
-    }
-    seq2 = PyTuple_New(n);
+    it = PyObject_GetIter(seq);
+    if (it == NULL)
+        return NULL;
+    seq2 = PyList_New(0);
     if (seq2 == NULL) {
-        return NULL;
-    }
-    for (i = 0; i < n; i++) {
-        tmp = PySequence_GetItem(seq, i);
-        if (tmp == NULL) {
-            Py_DECREF(seq2);
-            return NULL;
-        }
+        Py_DECREF(it);
+        return NULL;
+    }
+    while ((tmp = PyIter_Next(it))) {
         tmp2 = escape(tmp, quotes);
         if (tmp2 == NULL) {
             Py_DECREF(seq2);
-            return NULL;
-        }
-        PyTuple_SET_ITEM(seq2, i, tmp2);
+            Py_DECREF(it);
+            return NULL;
+        }
+        PyList_Append(seq2, tmp2);
         Py_DECREF(tmp);
+    }
+    Py_DECREF(it);
+    if (PyErr_Occurred()) {
+        Py_DECREF(seq2);
+        return NULL;
     }
     tmp = PyUnicode_Join(self, seq2);

trunk/genshi/tests/core.py

@@ -140 +140 @@
         self.assertEquals('foo<br />&lt;bar /&gt;<br /><baz />', markup)
 
+    def test_join_over_iter(self):
+        items = ['foo', '<bar />', Markup('<baz />')]
+        markup = Markup('<br />').join(i for i in items)
+        self.assertEquals('foo<br />&lt;bar /&gt;<br /><baz />', markup)
+
     def test_stripentities_all(self):
         markup = Markup('&amp; &#106;').stripentities()