diff --git a/mimo/connection/input.py b/mimo/connection/input.py index f68130f..9926c8e 100644 --- a/mimo/connection/input.py +++ b/mimo/connection/input.py @@ -27,7 +27,7 @@ async def push(self, item): async def peek(self): while self.empty(): - if self._closed and len(self._putters) == 0: + if self.is_closed(): raise ConnectionClosed getter = self._loop.create_future() @@ -71,3 +71,9 @@ def close(self): getter.set_exception(ConnectionClosed) else: getter.set_result(None) + + def is_closed(self): + return self._closed and len(self._queue) == 0 and len(self._putters) == 0 + + def is_open(self): + return not self.is_closed() diff --git a/setup.py b/setup.py index 1072c72..7518732 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ setup( name='mimo', - version='1.0.8', + version='1.0.9', author='Liam H. Childs', author_email='liam.h.childs@gmail.com', packages=find_packages(exclude=['tests']),