mirror/pyTelegramBotAPI
1
0
Fork 0
mirror of https://github.com/eternnoir/pyTelegramBotAPI.git synced 2023-08-10 21:12:57 +03:00
Code Releases Activity

Possibility to use alternative serializer

Browse Source 
With apihelper.CUSTOM_SERIALIZER you can replace pickle with other "dumper" like dill.
This commit is contained in:
Badiboy 2020-04-11 13:42:34 +03:00
parent 2c385bf077
commit ec86182f62
2 changed files with 11 additions and 2 deletions

10
telebot/__init__.py

@ -76,7 +76,10 @@ class Saver:
os.makedirs(dirs, exist_ok=True) os.makedirs(dirs, exist_ok=True)
with open(filename + ".tmp", file_mode) as file: with open(filename + ".tmp", file_mode) as file:
pickle.dump(handlers, file) if (apihelper.CUSTOM_SERIALIZER is None):
pickle.dump(handlers, file)
else:
apihelper.CUSTOM_SERIALIZER.dump(handlers, file)
if os.path.isfile(filename): if os.path.isfile(filename):
os.remove(filename) os.remove(filename)
@ -87,7 +90,10 @@ class Saver:
def return_load_handlers(filename, del_file_after_loading=True): def return_load_handlers(filename, del_file_after_loading=True):
if os.path.isfile(filename) and os.path.getsize(filename) > 0: if os.path.isfile(filename) and os.path.getsize(filename) > 0:
with open(filename, "rb") as file: with open(filename, "rb") as file:
handlers = pickle.load(file) if (apihelper.CUSTOM_SERIALIZER is None):
handlers = pickle.load(file)
else:
handlers = apihelper.CUSTOM_SERIALIZER.load(file)
if del_file_after_loading: if del_file_after_loading:
os.remove(filename) os.remove(filename)

3
telebot/apihelper.py

@ -18,6 +18,7 @@ from telebot import types
from telebot import util from telebot import util
logger = telebot.logger logger = telebot.logger
proxy = None proxy = None
API_URL = None API_URL = None
@ -26,6 +27,8 @@ FILE_URL = None
CONNECT_TIMEOUT = 3.5 CONNECT_TIMEOUT = 3.5
READ_TIMEOUT = 9999 READ_TIMEOUT = 9999
CUSTOM_SERIALIZER = None
ENABLE_MIDDLEWARE = False ENABLE_MIDDLEWARE = False