yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-06 15:14:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pyrofork: Add missings business_connection_id

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2024-07-14 20:02:29 +07:00
parent 61398bc5a4
commit e999da461d
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
8 changed files with 121 additions and 56 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pyrogram/methods/messages/edit_message_caption.py
View file

@ -32,7 +32,8 @@ class EditMessageCaption:
parse_mode: Optional["enums.ParseMode"] = None, parse_mode: Optional["enums.ParseMode"] = None,
caption_entities: List["types.MessageEntity"] = None, caption_entities: List["types.MessageEntity"] = None,
invert_media: bool = False, invert_media: bool = False,
reply_markup: "types.InlineKeyboardMarkup" = None reply_markup: "types.InlineKeyboardMarkup" = None,
business_connection_id: str = None
) -> "types.Message": ) -> "types.Message":
"""Edit the caption of media messages. """Edit the caption of media messages.
@ -64,6 +65,9 @@ class EditMessageCaption:
reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*): reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
An InlineKeyboardMarkup object. An InlineKeyboardMarkup object.
business_connection_id (``str``, *optional*):
Unique identifier of the business connection.
Returns: Returns:
:obj:`~pyrogram.types.Message`: On success, the edited message is returned. :obj:`~pyrogram.types.Message`: On success, the edited message is returned.
@ -79,5 +83,6 @@ class EditMessageCaption:
parse_mode=parse_mode, parse_mode=parse_mode,
entities=caption_entities, entities=caption_entities,
invert_media=invert_media, invert_media=invert_media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=business_connection_id
) )

33
pyrogram/methods/messages/edit_message_media.py
View file

@ -37,7 +37,8 @@ class EditMessageMedia:
media: "types.InputMedia", media: "types.InputMedia",
reply_markup: "types.InlineKeyboardMarkup" = None, reply_markup: "types.InlineKeyboardMarkup" = None,
file_name: str = None, file_name: str = None,
invert_media: bool = False invert_media: bool = False,
business_connection_id: str = None
) -> "types.Message": ) -> "types.Message":
"""Edit animation, audio, document, photo or video messages. """Edit animation, audio, document, photo or video messages.
@ -69,6 +70,9 @@ class EditMessageMedia:
invert_media (``bool``, *optional*): invert_media (``bool``, *optional*):
Inverts the position of the media and caption. Inverts the position of the media and caption.
business_connection_id (``str``, *optional*):
Unique identifier of the business connection.
Returns: Returns:
:obj:`~pyrogram.types.Message`: On success, the edited message is returned. :obj:`~pyrogram.types.Message`: On success, the edited message is returned.
@ -273,17 +277,24 @@ class EditMessageMedia:
else: else:
media = utils.get_input_media_from_file_id(media.media, FileType.DOCUMENT) media = utils.get_input_media_from_file_id(media.media, FileType.DOCUMENT)
r = await self.invoke( rpc = raw.functions.messages.EditMessage(
raw.functions.messages.EditMessage( peer=await self.resolve_peer(chat_id),
peer=await self.resolve_peer(chat_id), id=message_id,
id=message_id, media=media,
media=media, reply_markup=await reply_markup.write(self) if reply_markup else None,
reply_markup=await reply_markup.write(self) if reply_markup else None, message=message,
message=message, entities=entities,
entities=entities, invert_media=invert_media
invert_media=invert_media
)
) )
if business_connection_id is not None:
r = await self.invoke(
raw.functions.InvokeWithBusinessConnection(
connection_id=business_connection_id,
query=rpc
)
)
else:
r = await self.invoke(rpc)
for i in r.updates: for i in r.updates:
if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)): if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)):

23
pyrogram/methods/messages/edit_message_reply_markup.py
View file

@ -30,6 +30,7 @@ class EditMessageReplyMarkup:
chat_id: Union[int, str], chat_id: Union[int, str],
message_id: int, message_id: int,
reply_markup: "types.InlineKeyboardMarkup" = None, reply_markup: "types.InlineKeyboardMarkup" = None,
business_connection_id: str = None
) -> "types.Message": ) -> "types.Message":
"""Edit only the reply markup of messages sent by the bot. """Edit only the reply markup of messages sent by the bot.
@ -48,6 +49,9 @@ class EditMessageReplyMarkup:
reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*): reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
An InlineKeyboardMarkup object. An InlineKeyboardMarkup object.
business_connection_id (``str``, *optional*):
Unique identifier of the business connection.
Returns: Returns:
:obj:`~pyrogram.types.Message`: On success, the edited message is returned. :obj:`~pyrogram.types.Message`: On success, the edited message is returned.
@ -62,13 +66,20 @@ class EditMessageReplyMarkup:
InlineKeyboardMarkup([[ InlineKeyboardMarkup([[
InlineKeyboardButton("New button", callback_data="new_data")]])) InlineKeyboardButton("New button", callback_data="new_data")]]))
""" """
r = await self.invoke( rpc = raw.functions.messages.EditMessage(
raw.functions.messages.EditMessage( peer=await self.resolve_peer(chat_id),
peer=await self.resolve_peer(chat_id), id=message_id,
id=message_id, reply_markup=await reply_markup.write(self) if reply_markup else None
reply_markup=await reply_markup.write(self) if reply_markup else None,
)
) )
if business_connection_id is not None:
r = await self.invoke(
raw.functions.InvokeWithBusinessConnection(
connection_id=business_connection_id,
query=rpc
)
)
else:
r = await self.invoke(rpc)
for i in r.updates: for i in r.updates:
if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)): if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)):

31
pyrogram/methods/messages/edit_message_text.py
View file

@ -35,7 +35,8 @@ class EditMessageText:
entities: List["types.MessageEntity"] = None, entities: List["types.MessageEntity"] = None,
disable_web_page_preview: bool = None, disable_web_page_preview: bool = None,
invert_media: bool = None, invert_media: bool = None,
reply_markup: "types.InlineKeyboardMarkup" = None reply_markup: "types.InlineKeyboardMarkup" = None,
business_connection_id: str = None
) -> "types.Message": ) -> "types.Message":
"""Edit the text of messages. """Edit the text of messages.
@ -70,6 +71,9 @@ class EditMessageText:
reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*): reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
An InlineKeyboardMarkup object. An InlineKeyboardMarkup object.
business_connection_id (``str``, *optional*):
Unique identifier of the business connection.
Returns: Returns:
:obj:`~pyrogram.types.Message`: On success, the edited message is returned. :obj:`~pyrogram.types.Message`: On success, the edited message is returned.
@ -85,16 +89,23 @@ class EditMessageText:
disable_web_page_preview=True) disable_web_page_preview=True)
""" """
r = await self.invoke( rpc = raw.functions.messages.EditMessage(
raw.functions.messages.EditMessage( peer=await self.resolve_peer(chat_id),
peer=await self.resolve_peer(chat_id), id=message_id,
id=message_id, no_webpage=disable_web_page_preview or None,
no_webpage=disable_web_page_preview or None, invert_media=invert_media,
invert_media=invert_media, reply_markup=await reply_markup.write(self) if reply_markup else None,
reply_markup=await reply_markup.write(self) if reply_markup else None, **await utils.parse_text_entities(self, text, parse_mode, entities)
**await utils.parse_text_entities(self, text, parse_mode, entities)
)
) )
if business_connection_id is not None:
r = await self.invoke(
raw.functions.InvokeWithBusinessConnection(
connection_id=business_connection_id,
query=rpc
)
)
else:
r = await self.invoke(rpc)
for i in r.updates: for i in r.updates:
if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)): if isinstance(i, (raw.types.UpdateEditMessage, raw.types.UpdateEditChannelMessage)):

41
pyrogram/methods/messages/stop_poll.py
View file

@ -29,7 +29,8 @@ class StopPoll:
self: "pyrogram.Client", self: "pyrogram.Client",
chat_id: Union[int, str], chat_id: Union[int, str],
message_id: int, message_id: int,
reply_markup: "types.InlineKeyboardMarkup" = None reply_markup: "types.InlineKeyboardMarkup" = None,
business_connection_id: str = None
) -> "types.Poll": ) -> "types.Poll":
"""Stop a poll which was sent by you. """Stop a poll which was sent by you.
@ -50,6 +51,9 @@ class StopPoll:
reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*): reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
An InlineKeyboardMarkup object. An InlineKeyboardMarkup object.
business_connection_id (``str``, *optional*):
Unique identifier of the business connection.
Returns: Returns:
:obj:`~pyrogram.types.Poll`: On success, the stopped poll with the final results is returned. :obj:`~pyrogram.types.Poll`: On success, the stopped poll with the final results is returned.
@ -60,20 +64,27 @@ class StopPoll:
""" """
poll = (await self.get_messages(chat_id, message_id)).poll poll = (await self.get_messages(chat_id, message_id)).poll
r = await self.invoke( rpc = raw.functions.messages.EditMessage(
raw.functions.messages.EditMessage( peer=await self.resolve_peer(chat_id),
peer=await self.resolve_peer(chat_id), id=message_id,
id=message_id, media=raw.types.InputMediaPoll(
media=raw.types.InputMediaPoll( poll=raw.types.Poll(
poll=raw.types.Poll( id=int(poll.id),
id=int(poll.id), closed=True,
closed=True, question="",
question="", answers=[]
answers=[] )
) ),
), reply_markup=await reply_markup.write(self) if reply_markup else None
reply_markup=await reply_markup.write(self) if reply_markup else None
)
) )
if business_connection_id is not None:
r = await self.invoke(
raw.functions.InvokeWithBusinessConnection(
connection_id=business_connection_id,
query=rpc
)
)
else:
r = await self.invoke(rpc)
return types.Poll._parse(self, r.updates[0]) return types.Poll._parse(self, r.updates[0])

25
pyrogram/types/bots_and_keyboards/callback_query.py
View file

@ -215,7 +215,8 @@ class CallbackQuery(Object, Update):
text=text, text=text,
parse_mode=parse_mode, parse_mode=parse_mode,
disable_web_page_preview=disable_web_page_preview, disable_web_page_preview=disable_web_page_preview,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )
else: else:
return await self._client.edit_inline_text( return await self._client.edit_inline_text(
@ -223,7 +224,8 @@ class CallbackQuery(Object, Update):
text=text, text=text,
parse_mode=parse_mode, parse_mode=parse_mode,
disable_web_page_preview=disable_web_page_preview, disable_web_page_preview=disable_web_page_preview,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )
async def edit_message_caption( async def edit_message_caption(
@ -254,7 +256,12 @@ class CallbackQuery(Object, Update):
Raises: Raises:
RPCError: In case of a Telegram RPC error. RPCError: In case of a Telegram RPC error.
""" """
return await self.edit_message_text(caption, parse_mode, reply_markup=reply_markup) return await self.edit_message_text(
caption,
parse_mode,
reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
)
async def edit_message_media( async def edit_message_media(
self, self,
@ -284,13 +291,15 @@ class CallbackQuery(Object, Update):
chat_id=self.message.chat.id, chat_id=self.message.chat.id,
message_id=self.message.id, message_id=self.message.id,
media=media, media=media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )
else: else:
return await self._client.edit_inline_media( return await self._client.edit_inline_media(
inline_message_id=self.inline_message_id, inline_message_id=self.inline_message_id,
media=media, media=media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )
async def edit_message_reply_markup( async def edit_message_reply_markup(
@ -316,10 +325,12 @@ class CallbackQuery(Object, Update):
return await self._client.edit_message_reply_markup( return await self._client.edit_message_reply_markup(
chat_id=self.message.chat.id, chat_id=self.message.chat.id,
message_id=self.message.id, message_id=self.message.id,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )
else: else:
return await self._client.edit_inline_reply_markup( return await self._client.edit_inline_reply_markup(
inline_message_id=self.inline_message_id, inline_message_id=self.inline_message_id,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=getattr(self.message, "business_connection_id", None)
) )

12
pyrogram/types/messages_and_media/message.py
View file

@ -4121,7 +4121,8 @@ class Message(Object, Update):
entities=entities, entities=entities,
disable_web_page_preview=disable_web_page_preview, disable_web_page_preview=disable_web_page_preview,
invert_media=invert_media, invert_media=invert_media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=self.business_connection_id
) )
edit = edit_text edit = edit_text
@ -4181,7 +4182,8 @@ class Message(Object, Update):
parse_mode=parse_mode, parse_mode=parse_mode,
caption_entities=caption_entities, caption_entities=caption_entities,
invert_media=invert_media, invert_media=invert_media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=self.business_connection_id
) )
async def edit_media( async def edit_media(
@ -4228,7 +4230,8 @@ class Message(Object, Update):
message_id=self.id, message_id=self.id,
media=media, media=media,
invert_media=invert_media, invert_media=invert_media,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=self.business_connection_id
) )
async def edit_reply_markup(self, reply_markup: "types.InlineKeyboardMarkup" = None) -> "Message": async def edit_reply_markup(self, reply_markup: "types.InlineKeyboardMarkup" = None) -> "Message":
@ -4263,7 +4266,8 @@ class Message(Object, Update):
return await self._client.edit_message_reply_markup( return await self._client.edit_message_reply_markup(
chat_id=self.chat.id, chat_id=self.chat.id,
message_id=self.id, message_id=self.id,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=self.business_connection_id
) )
async def forward( async def forward(

3
pyrogram/types/messages_and_media/poll.py
View file

@ -250,5 +250,6 @@ class Poll(Object, Update):
return await self._client.stop_poll( return await self._client.stop_poll(
chat_id=self.chat.id, chat_id=self.chat.id,
message_id=self.id, message_id=self.id,
reply_markup=reply_markup reply_markup=reply_markup,
business_connection_id=self.business_connection_id
) )