Restart Conversation total timeout upon __aenter__

This commit is contained in:
Lonami Exo 2018-08-04 15:35:04 +02:00
parent a688c8a4ce
commit 5f73482d29

View File

@ -33,10 +33,8 @@ class Conversation(ChatGetter):
self._broadcast = None
self._timeout = timeout
if total_timeout:
self._total_due = time.time() + total_timeout
else:
self._total_due = float('inf')
self._total_timeout = total_timeout
self._total_due = None
self._outgoing = set()
self._last_outgoing = 0
@ -305,6 +303,11 @@ class Conversation(ChatGetter):
self._last_incoming = 0
self._pending_responses.clear()
self._response_indices.clear()
if self._total_timeout:
self._total_due = time.time() + self._total_timeout
else:
self._total_due = float('inf')
return self
async def __aexit__(self, *args):