LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-19 21:11:06 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make sure not to include sentinel in containers

Browse Source
This commit is contained in:
Lonami Exo 2018-06-16 18:40:08 +02:00
parent bb2cacd525
commit 8086772506
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/network/mtprotosender.py
View File

@ -631,13 +631,15 @@ class _ContainerQueue(asyncio.Queue):
""" """
async def get(self): async def get(self):
result = await super().get() 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 return result
result = [result] result = [result]
while not self.empty(): while not self.empty():
item = self.get_nowait() item = self.get_nowait()
if isinstance(item.obj, MessageContainer): if item == _reconnect_sentinel or\
isinstance(item.obj, MessageContainer):
self.put_nowait(item) self.put_nowait(item)
break break
else: else: