TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-18 13:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

First step of Client conversion using asyncio

Browse Source
This commit is contained in:
Dan 2018-06-13 20:00:19 +02:00
parent e2546e77ae
commit e333e8dada
1 changed files with 52 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
pyrogram/client/client.py
View File

@ -34,7 +34,6 @@ from configparser import ConfigParser
from datetime import datetime
from hashlib import sha256, md5
from signal import signal, SIGINT, SIGTERM, SIGABRT
from threading import Thread
from pyrogram.api import functions, types
from pyrogram.api.core import Object
@ -169,7 +168,7 @@ class Client(Methods, BaseClient):
self._proxy["enabled"] = True
self._proxy.update(value)
def start(self, debug: bool = False):
async def start(self, debug: bool = False):
"""Use this method to start the Client after creating it.
Requires no parameters.
@ -200,7 +199,7 @@ class Client(Methods, BaseClient):
client=self
)
self.session.start()
await self.session.start()
self.is_started = True
if self.user_id is None:
@ -224,66 +223,66 @@ class Client(Methods, BaseClient):
self.send(functions.messages.GetPinnedDialogs())
self.get_dialogs_chunk(0)
else:
self.send(functions.updates.GetState())
await self.send(functions.updates.GetState())
for i in range(self.UPDATES_WORKERS):
self.updates_workers_list.append(
Thread(
target=self.updates_worker,
name="UpdatesWorker#{}".format(i + 1)
)
)
self.updates_workers_list[-1].start()
for i in range(self.DOWNLOAD_WORKERS):
self.download_workers_list.append(
Thread(
target=self.download_worker,
name="DownloadWorker#{}".format(i + 1)
)
)
self.download_workers_list[-1].start()
self.dispatcher.start()
# for i in range(self.UPDATES_WORKERS):
# self.updates_workers_list.append(
# Thread(
# target=self.updates_worker,
# name="UpdatesWorker#{}".format(i + 1)
# )
# )
#
# self.updates_workers_list[-1].start()
#
# for i in range(self.DOWNLOAD_WORKERS):
# self.download_workers_list.append(
# Thread(
# target=self.download_worker,
# name="DownloadWorker#{}".format(i + 1)
# )
# )
#
# self.download_workers_list[-1].start()
#
# self.dispatcher.start()
mimetypes.init()
Syncer.add(self)
# Syncer.add(self)
def stop(self):
async def stop(self):
"""Use this method to manually stop the Client.
Requires no parameters.
"""
if not self.is_started:
raise ConnectionError("Client is already stopped")
Syncer.remove(self)
self.dispatcher.stop()
for _ in range(self.DOWNLOAD_WORKERS):
self.download_queue.put(None)
for i in self.download_workers_list:
i.join()
self.download_workers_list.clear()
for _ in range(self.UPDATES_WORKERS):
self.updates_queue.put(None)
for i in self.updates_workers_list:
i.join()
self.updates_workers_list.clear()
for i in self.media_sessions.values():
i.stop()
self.media_sessions.clear()
# Syncer.remove(self)
# self.dispatcher.stop()
#
# for _ in range(self.DOWNLOAD_WORKERS):
# self.download_queue.put(None)
#
# for i in self.download_workers_list:
# i.join()
#
# self.download_workers_list.clear()
#
# for _ in range(self.UPDATES_WORKERS):
# self.updates_queue.put(None)
#
# for i in self.updates_workers_list:
# i.join()
#
# self.updates_workers_list.clear()
#
# for i in self.media_sessions.values():
# i.stop()
#
# self.media_sessions.clear()
self.is_started = False
self.session.stop()
await self.session.stop()
def add_handler(self, handler, group: int = 0):
"""Use this method to register an update handler.
@ -812,7 +811,7 @@ class Client(Methods, BaseClient):
self.stop()
def send(self, data: Object):
async def send(self, data: Object):
"""Use this method to send Raw Function queries.
This method makes possible to manually call every single Telegram API method in a low-level manner.
@ -829,7 +828,7 @@ class Client(Methods, BaseClient):
if not self.is_started:
raise ConnectionError("Client has not been started")
r = self.session.send(data)
r = await self.session.send(data)
self.fetch_peers(getattr(r, "users", []))
self.fetch_peers(getattr(r, "chats", []))