From 6ed4e3160bb61f3d916d3b3cf5e005f1210c9d45 Mon Sep 17 00:00:00 2001 From: vegeta1k95 <32207817+vegeta1k95@users.noreply.github.com> Date: Sun, 8 Apr 2018 19:46:46 +0300 Subject: [PATCH] Added support for saving Event state --- telethon/tl/tlobject.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/telethon/tl/tlobject.py b/telethon/tl/tlobject.py index 22b219a5..57975ec3 100644 --- a/telethon/tl/tlobject.py +++ b/telethon/tl/tlobject.py @@ -10,12 +10,23 @@ class TLObject: # These should be overrode self.content_related = False # Only requests/functions/queries are + + # Internal parameter to tell pickler in which state Event object was + self._event_is_set = False self._set_event() def _set_event(self): self.confirm_received = Event() + + # Set Event state to 'set' if needed + if self._event_is_set: + self.confirm_received.set() def __getstate__(self): + # Save state of the Event object + self._event_is_set = self.confirm_received.is_set() + + # Exclude Event object from dict and return new state new_dct = dict(self.__dict__) del new_dct["confirm_received"] return new_dct