hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-09 13:29:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix setattr for events.NewMessage/custom.Message

Browse Source
This commit is contained in:
Lonami Exo
2018-06-03 13:00:07 +02:00
parent 0a3151175d
commit b2ed6caff4
2 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/tl/custom/message.py
View File

@@ -19,7 +19,10 @@ class Message:
in the original :tl:`Message`.
"""
def __init__(self, client, original, entities, input_chat):
self.__dict__['_init'] = False
# Share the original dictionary. Modifications to this
# object should also be reflected in the original one.
# This way there's no need to worry about get/setattr.
self.__dict__ = original.__dict__
self.original_message = original
self.stringify = self.original_message.stringify
self.to_dict = self.original_message.to_dict
@@ -43,7 +46,6 @@ class Message:
elif fwd.channel_id:
self._fwd_from_entity = entities.get(get_peer_id(
types.PeerChannel(fwd.channel_id)))
self._init = True
def __new__(cls, client, original, entities, input_chat):
if isinstance(original, types.Message):
@@ -53,15 +55,6 @@ class Message:
else:
return cls
def __getattr__(self, item):
return getattr(self.original_message, item)
def __setattr__(self, name, value):
if not self._init or name in self.__dict__:
self.__dict__[name] = value
else:
setattr(self.original_message, name, value)
def __str__(self):
return str(self.original_message)