From 4f9b38765e0c820b7bfd38e8731a4e2c6280b1fd Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Thu, 23 Aug 2018 21:07:19 +0200 Subject: [PATCH] Add missing async/await keywords --- .../client/methods/chats/get_chat_member.py | 10 +++--- .../methods/messages/edit_message_media.py | 32 +++++++++---------- 2 files changed, 21 insertions(+), 21 deletions(-) diff --git a/pyrogram/client/methods/chats/get_chat_member.py b/pyrogram/client/methods/chats/get_chat_member.py index 51e07f91..bd43808f 100644 --- a/pyrogram/client/methods/chats/get_chat_member.py +++ b/pyrogram/client/methods/chats/get_chat_member.py @@ -21,7 +21,7 @@ from ...ext import BaseClient, utils class GetChatMember(BaseClient): - def get_chat_member(self, + async def get_chat_member(self, chat_id: int or str, user_id: int or str): """Use this method to get information about one member of a chat. @@ -41,11 +41,11 @@ class GetChatMember(BaseClient): Raises: :class:`Error ` """ - chat_id = self.resolve_peer(chat_id) - user_id = self.resolve_peer(user_id) + chat_id = await self.resolve_peer(chat_id) + user_id = await self.resolve_peer(user_id) if isinstance(chat_id, types.InputPeerChat): - full_chat = self.send( + full_chat = await self.send( functions.messages.GetFullChat( chat_id=chat_id.chat_id ) @@ -57,7 +57,7 @@ class GetChatMember(BaseClient): else: raise errors.UserNotParticipant elif isinstance(chat_id, types.InputPeerChannel): - r = self.send( + r = await self.send( functions.channels.GetParticipant( channel=chat_id, user_id=user_id diff --git a/pyrogram/client/methods/messages/edit_message_media.py b/pyrogram/client/methods/messages/edit_message_media.py index 086d8dc2..5993f1c9 100644 --- a/pyrogram/client/methods/messages/edit_message_media.py +++ b/pyrogram/client/methods/messages/edit_message_media.py @@ -31,7 +31,7 @@ from pyrogram.client.types import ( class EditMessageMedia(BaseClient): - def edit_message_media(self, + async def edit_message_media(self, chat_id: int or str, message_id: int, media, @@ -41,11 +41,11 @@ class EditMessageMedia(BaseClient): if isinstance(media, InputMediaPhoto): if os.path.exists(media.media): - media = self.send( + media = await self.send( functions.messages.UploadMedia( - peer=self.resolve_peer(chat_id), + peer=await self.resolve_peer(chat_id), media=types.InputMediaUploadedPhoto( - file=self.save_file(media.media) + file=await self.save_file(media.media) ) ) ) @@ -85,12 +85,12 @@ class EditMessageMedia(BaseClient): if isinstance(media, InputMediaVideo): if os.path.exists(media.media): - media = self.send( + media = await self.send( functions.messages.UploadMedia( - peer=self.resolve_peer(chat_id), + peer=await self.resolve_peer(chat_id), media=types.InputMediaUploadedDocument( mime_type=mimetypes.types_map[".mp4"], - file=self.save_file(media.media), + file=await self.save_file(media.media), attributes=[ types.DocumentAttributeVideo( supports_streaming=media.supports_streaming or None, @@ -139,12 +139,12 @@ class EditMessageMedia(BaseClient): if isinstance(media, InputMediaAudio): if os.path.exists(media.media): - media = self.send( + media = await self.send( functions.messages.UploadMedia( - peer=self.resolve_peer(chat_id), + peer=await self.resolve_peer(chat_id), media=types.InputMediaUploadedDocument( mime_type=mimetypes.types_map.get("." + media.media.split(".")[-1], "audio/mpeg"), - file=self.save_file(media.media), + file=await self.save_file(media.media), attributes=[ types.DocumentAttributeAudio( duration=media.duration, @@ -192,12 +192,12 @@ class EditMessageMedia(BaseClient): if isinstance(media, InputMediaAnimation): if os.path.exists(media.media): - media = self.send( + media = await self.send( functions.messages.UploadMedia( - peer=self.resolve_peer(chat_id), + peer=await self.resolve_peer(chat_id), media=types.InputMediaUploadedDocument( mime_type=mimetypes.types_map[".mp4"], - file=self.save_file(media.media), + file=await self.save_file(media.media), attributes=[ types.DocumentAttributeVideo( supports_streaming=True, @@ -245,9 +245,9 @@ class EditMessageMedia(BaseClient): ) ) - r = self.send( + r = await self.send( functions.messages.EditMessage( - peer=self.resolve_peer(chat_id), + peer=await self.resolve_peer(chat_id), id=message_id, reply_markup=reply_markup.write() if reply_markup else None, media=media, @@ -257,7 +257,7 @@ class EditMessageMedia(BaseClient): for i in r.updates: if isinstance(i, (types.UpdateEditMessage, types.UpdateEditChannelMessage)): - return utils.parse_messages( + return await utils.parse_messages( self, i.message, {i.id: i for i in r.users}, {i.id: i for i in r.chats}