mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Tune session settings

Browse Source
This commit is contained in:
Dan 2018-04-12 08:29:39 +02:00
parent 1736a8ea9b
commit 862567a1ed
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pyrogram/session/session.py
View File

@ -61,7 +61,7 @@ class Session:
INITIAL_SALT = 0x616e67656c696361 INITIAL_SALT = 0x616e67656c696361
NET_WORKERS = 1 NET_WORKERS = 1
WAIT_TIMEOUT = 30 WAIT_TIMEOUT = 15
MAX_RETRIES = 5 MAX_RETRIES = 5
ACKS_THRESHOLD = 8 ACKS_THRESHOLD = 8
PING_INTERVAL = 5 PING_INTERVAL = 5
@ -308,7 +308,9 @@ class Session:
break break
try: try:
self._send(functions.PingDelayDisconnect(0, self.PING_INTERVAL + 15), False) self._send(functions.PingDelayDisconnect(
0, self.WAIT_TIMEOUT + 10
), False)
except (OSError, TimeoutError): except (OSError, TimeoutError):
pass pass
@ -338,7 +340,7 @@ class Session:
try: try:
self.current_salt = self._send(functions.GetFutureSalts(1)).salts[0] self.current_salt = self._send(functions.GetFutureSalts(1)).salts[0]
except (OSError, TimeoutError): except (OSError, TimeoutError, Error):
self.connection.close() self.connection.close()
break break
@ -395,12 +397,14 @@ class Session:
def send(self, data: Object): def send(self, data: Object):
for i in range(self.MAX_RETRIES): for i in range(self.MAX_RETRIES):
self.is_connected.wait() self.is_connected.wait(self.WAIT_TIMEOUT)
try: try:
return self._send(data) return self._send(data)
except (OSError, TimeoutError): except (OSError, TimeoutError):
(log.warning if i > 0 else log.info)("{}: {} Retrying {}".format(i, datetime.now(), type(data))) (log.warning if i > 2 else log.info)(
"{}: {} Retrying {}".format(i, datetime.now(), type(data))
)
continue continue
else: else:
return None return None