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

Add an option to allow not saving/processing found entities

Browse Source
This commit is contained in:
Lonami Exo 2017-10-01 13:32:27 +02:00
parent e4c5396c04
commit 9bccd17881
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/tl/session.py
View File

@ -39,6 +39,7 @@ class Session:
self.system_lang_code = session.system_lang_code self.system_lang_code = session.system_lang_code
self.lang_pack = session.lang_pack self.lang_pack = session.lang_pack
self.report_errors = session.report_errors self.report_errors = session.report_errors
self.save_entities = session.save_entities
else: # str / None else: # str / None
self.session_user_id = session_user_id self.session_user_id = session_user_id
@ -51,6 +52,7 @@ class Session:
self.system_lang_code = self.lang_code self.system_lang_code = self.lang_code
self.lang_pack = '' self.lang_pack = ''
self.report_errors = True self.report_errors = True
self.save_entities = True
# Cross-thread safety # Cross-thread safety
self._seq_no_lock = Lock() self._seq_no_lock = Lock()
@ -78,16 +80,19 @@ class Session:
with self._save_lock: with self._save_lock:
with open('{}.session'.format(self.session_user_id), 'w') as file: with open('{}.session'.format(self.session_user_id), 'w') as file:
json.dump({ out_dict = {
'port': self.port, 'port': self.port,
'salt': self.salt, 'salt': self.salt,
'layer': self.layer, 'layer': self.layer,
'server_address': self.server_address, 'server_address': self.server_address,
'auth_key_data': 'auth_key_data':
b64encode(self.auth_key.key).decode('ascii') b64encode(self.auth_key.key).decode('ascii')
if self.auth_key else None, if self.auth_key else None
'entities': list(self._input_entities.items()) }
}, file) if self.save_entities:
out_dict['entities'] = list(self._input_entities.items())
json.dump(out_dict, file)
def delete(self): def delete(self):
"""Deletes the current session file""" """Deletes the current session file"""
@ -184,7 +189,7 @@ class Session:
"""Adds new input entities to the local database of them. """Adds new input entities to the local database of them.
Unknown types will be ignored. Unknown types will be ignored.
""" """
if not entities: if not entities or not self.save_entities:
return False return False
new = {} new = {}