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

Raise ConnectionError if client is not started

Browse Source
This commit is contained in:
Dan 2018-03-19 03:37:43 +01:00
parent 0f2dc6c624
commit 299d6aca5c
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyrogram/client/client.py
View File

@ -154,7 +154,7 @@ class Client:
self.auth_key = None self.auth_key = None
self.user_id = None self.user_id = None
self.rnd_id = None self.rnd_id = MsgId
self.peers_by_id = {} self.peers_by_id = {}
self.peers_by_username = {} self.peers_by_username = {}
@ -167,6 +167,7 @@ class Client:
self.session = None self.session = None
self.is_started = None
self.is_idle = None self.is_idle = None
self.updates_queue = Queue() self.updates_queue = Queue()
@ -195,6 +196,7 @@ class Client:
) )
self.session.start() self.session.start()
self.is_started = True
if self.user_id is None: if self.user_id is None:
if self.token is None: if self.token is None:
@ -210,8 +212,6 @@ class Client:
else: else:
self.send(functions.updates.GetState()) self.send(functions.updates.GetState())
self.rnd_id = MsgId
for i in range(self.UPDATES_WORKERS): for i in range(self.UPDATES_WORKERS):
Thread(target=self.updates_worker, name="UpdatesWorker#{}".format(i + 1)).start() Thread(target=self.updates_worker, name="UpdatesWorker#{}".format(i + 1)).start()
@ -227,6 +227,7 @@ class Client:
"""Use this method to manually stop the Client. """Use this method to manually stop the Client.
Requires no parameters. Requires no parameters.
""" """
self.is_started = False
self.session.stop() self.session.stop()
for _ in range(self.UPDATES_WORKERS): for _ in range(self.UPDATES_WORKERS):
@ -754,12 +755,15 @@ class Client:
Raises: Raises:
:class:`pyrogram.Error` :class:`pyrogram.Error`
""" """
r = self.session.send(data) if self.is_started:
r = self.session.send(data)
self.fetch_peers(getattr(r, "users", [])) self.fetch_peers(getattr(r, "users", []))
self.fetch_peers(getattr(r, "chats", [])) self.fetch_peers(getattr(r, "chats", []))
return r return r
else:
raise ConnectionError("client '{}' is not started".format(self.session_name))
def load_config(self): def load_config(self):
parser = ConfigParser() parser = ConfigParser()