f.write('!')
print `f.getvalue()`
f.close()
+
+ f = module.StringIO()
+ f.writelines(["a", "b", "c"])
+ f.seek(0)
+ print `f.getvalue()`
+ f.close()
+
f = module.StringIO()
f.write(s)
f.seek(10)
else:
print "Failed to catch ValueError writing to closed StringIO."
-# Don't bother testing cStringIO without
import StringIO, cStringIO
do_test(StringIO)
do_test(cStringIO)
static PyObject *
O_writelines(Oobject *self, PyObject *args) {
PyObject *tmp = 0;
- static PyObject *string_joinfields = 0;
+ static PyObject *joiner = NULL;
UNLESS (PyArg_ParseTuple(args, "O:writelines", &args)) return NULL;
- if (!string_joinfields) {
- UNLESS (tmp = PyImport_ImportModule("string")) return NULL;
- string_joinfields=PyObject_GetAttrString(tmp, "joinfields");
- Py_DECREF(tmp);
- UNLESS (string_joinfields) return NULL;
- }
+ if (!joiner) {
+ PyObject *empty_string = PyString_FromString("");
+ if (empty_string == NULL)
+ return NULL;
+ joiner = PyObject_GetAttrString(empty_string, "join");
+ Py_DECREF(empty_string);
+ if (joiner == NULL)
+ return NULL;
+ }
if (PyObject_Size(args) < 0) return NULL;
- tmp = PyObject_CallFunction(string_joinfields, "Os", args, "");
+ tmp = PyObject_CallFunction(joiner, "O", args);
UNLESS (tmp) return NULL;
args = Py_BuildValue("(O)", tmp);