From: Guido van Rossum <guido@python.org>
Date: Fri, 31 Jan 2003 16:04:15 +0000 (+0000)
Subject: Make StringIO its own iterator, similar to real files.
X-Git-Tag: v2.3c1~2159
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=c1265bd9a65b510c290667f7d9fbbf6e3156f386;p=thirdparty%2FPython%2Fcpython.git

Make StringIO its own iterator, similar to real files.

(This should also be done to cStringIO.)
---

diff --git a/Lib/StringIO.py b/Lib/StringIO.py
index 79ab7e16b329..89dda5e95d11 100644
--- a/Lib/StringIO.py
+++ b/Lib/StringIO.py
@@ -59,7 +59,15 @@ class StringIO:
         self.softspace = 0
 
     def __iter__(self):
-        return iter(self.readline, '')
+        return self
+
+    def next(self):
+        if self.closed:
+            raise StopIteration
+        r = self.readline()
+        if not r:
+            raise StopIteration
+        return r
 
     def close(self):
         """Free the memory buffer.