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

Added support for saving Event state

Browse Source
This commit is contained in:
vegeta1k95 2018-04-08 19:46:46 +03:00 committed by GitHub
parent 6e184c8dcb
commit 6ed4e3160b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/tl/tlobject.py
View File

@ -10,12 +10,23 @@ class TLObject:
# These should be overrode # These should be overrode
self.content_related = False # Only requests/functions/queries are 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() self._set_event()
def _set_event(self): def _set_event(self):
self.confirm_received = Event() self.confirm_received = Event()
# Set Event state to 'set' if needed
if self._event_is_set:
self.confirm_received.set()
def __getstate__(self): 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__) new_dct = dict(self.__dict__)
del new_dct["confirm_received"] del new_dct["confirm_received"]
return new_dct return new_dct