hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-18 02:56:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix handling message payloads that are too large

Browse Source
This commit is contained in:
Lonami Exo 2019-03-06 18:08:51 +01:00
parent 3d72c10ea5
commit baa8970bb6
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/extensions/messagepacker.py
View File

@ -69,14 +69,24 @@ class MessagePacker:
id(state.request)) id(state.request))
continue continue
if batch:
# Put the item back since it can't be sent in this batch # Put the item back since it can't be sent in this batch
self._deque.appendleft(state) self._deque.appendleft(state)
if batch:
break break
# If a single message exceeds the maximum size, then the # If a single message exceeds the maximum size, then the
# message payload cannot be sent. Telegram would forcibly # message payload cannot be sent. Telegram would forcibly
# close the connection; message would never be confirmed. # close the connection; message would never be confirmed.
#
# We don't put the item back because it can never be sent.
# If we did, we would loop again and reach this same path.
# Setting the exception twice results in `InvalidStateError`
# and this method should never return with error, which we
# really want to avoid.
self._log.warning(
'Message payload for %s is too long (%d) and cannot be sent',
state.request.__class__.__name__, len(state.data)
)
state.future.set_exception( state.future.set_exception(
ValueError('Request payload is too big')) ValueError('Request payload is too big'))