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

enh: reset requests.Session feature

Browse Source 
Need for proxy changing and other reconnection stuff
This commit is contained in:
uburuntu 2018-09-06 12:42:44 +03:00
parent 2b3c86b647
commit bab9f7bbb9
2 changed files with 6 additions and 7 deletions

4
telebot/apihelper.py

@ -27,8 +27,8 @@ CONNECT_TIMEOUT = 3.5
READ_TIMEOUT = 9999 READ_TIMEOUT = 9999
def _get_req_session(): def _get_req_session(reset=False):
return util.per_thread('req_session', lambda: requests.session()) return util.per_thread('req_session', lambda: requests.session(), reset)
def _make_request(token, method_name, method='get', params=None, files=None, base_url=API_URL): def _make_request(token, method_name, method='get', params=None, files=None, base_url=API_URL):

9
telebot/util.py

@ -248,13 +248,12 @@ def extract_arguments(text):
return result.group(2) if is_command(text) else None return result.group(2) if is_command(text) else None
def per_thread(key, construct_value): def per_thread(key, construct_value, reset=False):
try: if reset or not hasattr(thread_local, key):
return getattr(thread_local, key)
except AttributeError:
value = construct_value() value = construct_value()
setattr(thread_local, key, value) setattr(thread_local, key, value)
return value
return getattr(thread_local, key)
def generate_random_token(): def generate_random_token():