From 8086772506d3b7d8864e129b4c070db9ddb35944 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Sat, 16 Jun 2018 18:40:08 +0200 Subject: [PATCH] Make sure not to include sentinel in containers --- telethon/network/mtprotosender.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/telethon/network/mtprotosender.py b/telethon/network/mtprotosender.py index ca9b87ae..04a6dace 100644 --- a/telethon/network/mtprotosender.py +++ b/telethon/network/mtprotosender.py @@ -631,13 +631,15 @@ class _ContainerQueue(asyncio.Queue): """ async def get(self): result = await super().get() - if self.empty() or isinstance(result.obj, MessageContainer): + if self.empty() or result == _reconnect_sentinel or\ + isinstance(result.obj, MessageContainer): return result result = [result] while not self.empty(): item = self.get_nowait() - if isinstance(item.obj, MessageContainer): + if item == _reconnect_sentinel or\ + isinstance(item.obj, MessageContainer): self.put_nowait(item) break else: