From 4885eb2040a0813f4b54fe8aa5fb21be9d10d3f7 Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Wed, 6 Jul 2016 10:19:22 -0400
Subject: [PATCH] removed unneeded stuff on the iterator
---
goblin/driver/connection.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/goblin/driver/connection.py b/goblin/driver/connection.py
index 1653088..1f24970 100644
--- a/goblin/driver/connection.py
+++ b/goblin/driver/connection.py
@@ -16,9 +16,10 @@ Message = collections.namedtuple(
class AsyncResponseIter:
- def __init__(self, response_queue, loop:
+ def __init__(self, response_queue, loop, conn):
self._response_queue = response_queue
self._loop = loop
+ self._conn = conn
async def __aiter__(self):
return self
@@ -134,9 +135,7 @@ class Connection(AbstractConnection):
response_queue = asyncio.Queue(loop=self._loop)
self.response_queues[request_id] = response_queue
self._ws.send_bytes(message)
- return AsyncResponseIter(response_queue, self._loop, self,
- self._username, self._password,
- processor, session)
+ return AsyncResponseIter(response_queue, self._loop, self)
async def close(self):
await self._ws.close()
--
GitLab