Skip to content
Snippets Groups Projects
Commit 4885eb20 authored by davebshow's avatar davebshow
Browse files

removed unneeded stuff on the iterator

parent 6761c173
No related branches found
No related tags found
No related merge requests found
...@@ -16,9 +16,10 @@ Message = collections.namedtuple( ...@@ -16,9 +16,10 @@ Message = collections.namedtuple(
class AsyncResponseIter: class AsyncResponseIter:
def __init__(self, response_queue, loop: def __init__(self, response_queue, loop, conn):
self._response_queue = response_queue self._response_queue = response_queue
self._loop = loop self._loop = loop
self._conn = conn
async def __aiter__(self): async def __aiter__(self):
return self return self
...@@ -134,9 +135,7 @@ class Connection(AbstractConnection): ...@@ -134,9 +135,7 @@ class Connection(AbstractConnection):
response_queue = asyncio.Queue(loop=self._loop) response_queue = asyncio.Queue(loop=self._loop)
self.response_queues[request_id] = response_queue self.response_queues[request_id] = response_queue
self._ws.send_bytes(message) self._ws.send_bytes(message)
return AsyncResponseIter(response_queue, self._loop, self, return AsyncResponseIter(response_queue, self._loop, self)
self._username, self._password,
processor, session)
async def close(self): async def close(self):
await self._ws.close() await self._ws.close()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment