commit c1265bd9a65b510c290667f7d9fbbf6e3156f386
author Guido van Rossum <guido@python.org> Fri Jan 31 16:04:15 2003 +0000
committer Guido van Rossum <guido@python.org> Fri Jan 31 16:04:15 2003 +0000
tree 28d9bb11271986a8ff056213b0774e34d9a29875
parent 5b8132ffa34244cc24b603d8462e6e733e014e71

Make StringIO its own iterator, similar to real files.

(This should also be done to cStringIO.)

Lib/StringIO.py

diff --git a/Lib/StringIO.py b/Lib/StringIO.py
index 79ab7e1..89dda5e 100644
--- a/Lib/StringIO.py
+++ b/Lib/StringIO.py
@@ -59,7 +59,15 @@
         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.