fix iter_chunks type hint (#3230) (#3232)

aio-libs · Sep 1, 2018 · 29e3662 · 29e3662
1 parent 06438b0
commit 29e3662
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/aiohttp/streams.py b/aiohttp/streams.py
@@ -37,9 +37,16 @@ async def __anext__(self) -> bytes:
         return rv
 
 
-class ChunkTupleAsyncStreamIterator(AsyncStreamIterator):
-    async def __anext__(self) -> bytes:
-        rv = await self.read_func()
+class ChunkTupleAsyncStreamIterator:
+
+    def __init__(self, stream: 'StreamReader') -> None:
+        self._stream = stream
+
+    def __aiter__(self) -> 'ChunkTupleAsyncStreamIterator':
+        return self
+
+    async def __anext__(self) -> Tuple[bytes, bool]:
+        rv = await self._stream.readchunk()
         if rv == (b'', False):
             raise StopAsyncIteration  # NOQA
         return rv
@@ -72,7 +79,7 @@ def iter_chunks(self) -> ChunkTupleAsyncStreamIterator:
 
         Python-3.5 available for Python 3.5+ only
         """
-        return ChunkTupleAsyncStreamIterator(self.readchunk)  # type: ignore
+        return ChunkTupleAsyncStreamIterator(self)  # type: ignore
 
 
 class StreamReader(AsyncStreamReaderMixin):