commit 33c6b569b7b123976230c92c6d8e74b2c10fef2e
author Yury Selivanov <yselivanov@sprymix.com> Wed May 13 14:23:29 2015 -0400
committer Yury Selivanov <yselivanov@sprymix.com> Wed May 13 14:23:29 2015 -0400
tree fe2ad5b421ed45a15afdabd81ffc4395121c3bfa
parent d4be6914943fbc829bcbb62c3aa59f84c917c522

Issue 24179: Support 'async for' for asyncio.StreamReader.

Lib/asyncio/streams.py

diff --git a/Lib/asyncio/streams.py b/Lib/asyncio/streams.py
index 64ff3d2..176c65e 100644
--- a/Lib/asyncio/streams.py
+++ b/Lib/asyncio/streams.py
@@ -6,6 +6,7 @@
            ]
 
 import socket
+import sys
 
 if hasattr(socket, 'AF_UNIX'):
     __all__.extend(['open_unix_connection', 'start_unix_server'])
@@ -19,6 +20,7 @@
 
 
 _DEFAULT_LIMIT = 2**16
+_PY35 = sys.version_info >= (3, 5)
 
 
 class IncompleteReadError(EOFError):
@@ -485,3 +487,15 @@
             n -= len(block)
 
         return b''.join(blocks)
+
+    if _PY35:
+        @coroutine
+        def __aiter__(self):
+            return self
+
+        @coroutine
+        def __anext__(self):
+            val = yield from self.readline()
+            if val == b'':
+                raise StopAsyncIteration
+            return val