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

Merge pull request #575 from uburuntu/reset_session

Reset requests.Session feature
This commit is contained in:
FrankWang 2018-09-07 15:56:59 +08:00 committed by GitHub
commit 07b82dc9b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 7 deletions

View File

@ -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):

View File

@ -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():