TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-28 00:56:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't call InitConnection on CDN DCs

Browse Source
This commit is contained in:
Dan 2017-12-19 11:38:15 +01:00
parent a9087a2a8c
commit 14632c9bb1
1 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
pyrogram/session/session.py
View File

@ -68,12 +68,14 @@ class Session:
notice_displayed = False
def __init__(self, dc_id: int, test_mode: bool, auth_key: bytes, api_id: str):
def __init__(self, dc_id: int, test_mode: bool, auth_key: bytes, api_id: str, is_cdn: bool = False):
if not Session.notice_displayed:
print("Pyrogram v{}, {}".format(__version__, __copyright__))
print("Licensed under the terms of the " + __license__, end="\n\n")
Session.notice_displayed = True
self.is_cdn = is_cdn
self.connection = Connection(DataCenter(dc_id, test_mode))
self.api_id = api_id
@ -107,6 +109,8 @@ class Session:
self.total_bytes = 0
def start(self):
terms = None
while True:
try:
self.connection.connect()
@ -128,19 +132,20 @@ class Session:
self.next_salt_thread = Thread(target=self.next_salt, name="NextSaltThread")
self.next_salt_thread.start()
terms = self._send(
functions.InvokeWithLayer(
layer,
functions.InitConnection(
self.api_id,
self.DEVICE_MODEL,
self.SYSTEM_VERSION,
self.APP_VERSION,
"en", "", "en",
functions.help.GetTermsOfService(),
if not self.is_cdn:
terms = self._send(
functions.InvokeWithLayer(
layer,
functions.InitConnection(
self.api_id,
self.DEVICE_MODEL,
self.SYSTEM_VERSION,
self.APP_VERSION,
"en", "", "en",
functions.help.GetTermsOfService(),
)
)
)
)
).text
if self.ping_thread is not None:
self.ping_thread.join()
@ -161,7 +166,7 @@ class Session:
log.debug("Session started")
return terms.text
return terms
def stop(self):
self.is_connected.clear()