2
0
mirror of https://github.com/pyrogram/pyrogram synced 2025-09-10 11:15:27 +00:00

Add ability to choose the amount of worker threads for the main session

This commit is contained in:
Dan
2018-01-26 11:41:09 +01:00
parent e542c73966
commit 7234edad5d
2 changed files with 31 additions and 7 deletions

View File

@@ -105,7 +105,8 @@ class Client:
phone_code: str or callable = None,
password: str = None,
first_name: str = None,
last_name: str = None):
last_name: str = None,
workers: int = 4):
self.session_name = session_name
self.test_mode = test_mode
@@ -115,6 +116,8 @@ class Client:
self.first_name = first_name
self.last_name = last_name
self.workers = workers
self.dc_id = None
self.auth_key = None
self.user_id = None
@@ -144,7 +147,14 @@ class Client:
self.load_config()
self.load_session(self.session_name)
self.session = Session(self.dc_id, self.test_mode, self.proxy, self.auth_key, self.config.api_id)
self.session = Session(
self.dc_id,
self.test_mode,
self.proxy,
self.auth_key,
self.config.api_id,
workers=self.workers
)
terms = self.session.start()
@@ -243,7 +253,14 @@ class Client:
self.dc_id = e.x
self.auth_key = Auth(self.dc_id, self.test_mode, self.proxy).create()
self.session = Session(self.dc_id, self.test_mode, self.proxy, self.auth_key, self.config.api_id)
self.session = Session(
self.dc_id,
self.test_mode,
self.proxy,
self.auth_key,
self.config.api_id,
workers=self.workers
)
self.session.start()
r = self.send(