From 3cce2ca1dbdb2237e74e28d1f746d60e69914c04 Mon Sep 17 00:00:00 2001
From: wulan17 <wulan17@nusantararom.org>
Date: Fri, 5 Apr 2024 19:32:59 +0700
Subject: [PATCH] Pyrofork: Add sender_business_bot field to class Message

Signed-off-by: wulan17 <wulan17@nusantararom.org>
---
 pyrogram/types/messages_and_media/message.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pyrogram/types/messages_and_media/message.py b/pyrogram/types/messages_and_media/message.py
index 47b70abca..5eafa14bd 100644
--- a/pyrogram/types/messages_and_media/message.py
+++ b/pyrogram/types/messages_and_media/message.py
@@ -77,6 +77,9 @@ class Message(Object, Update):
             The supergroup itself for messages from anonymous group administrators.
             The linked channel for messages automatically forwarded to the discussion group.
 
+        sender_business_bot (:obj:`~pyrogram.types.User`, *optional*):
+            Sender of the message, sent on behalf of a business bot.
+
         date (:py:obj:`~datetime.datetime`, *optional*):
             Date the message was sent.
 
@@ -400,6 +403,7 @@ def __init__(
         message_thread_id: int = None,
         from_user: "types.User" = None,
         sender_chat: "types.Chat" = None,
+        sender_business_bot: "types.User" = None,
         date: datetime = None,
         chat: "types.Chat" = None,
         topics: "types.ForumTopic" = None,
@@ -502,6 +506,7 @@ def __init__(
         self.message_thread_id = message_thread_id
         self.from_user = from_user
         self.sender_chat = sender_chat
+        self.sender_business_bot = sender_business_bot
         self.date = date
         self.chat = chat
         self.topics = topics
@@ -876,6 +881,7 @@ async def _parse(
             entities = [types.MessageEntity._parse(client, entity, users) for entity in message.entities]
             entities = types.List(filter(lambda x: x is not None, entities))
 
+            sender_business_bot = None
             forward_from = None
             forward_sender_name = None
             forward_from_chat = None
@@ -1027,6 +1033,9 @@ async def _parse(
 
             reactions = types.MessageReactions._parse(client, message.reactions)
 
+            if message.via_business_bot_id:
+                sender_business_bot = types.User._parse(client, users.get(message.via_business_bot_id, None))
+
             parsed_message = Message(
                 id=message.id,
                 message_thread_id=message_thread_id,
@@ -1035,6 +1044,7 @@ async def _parse(
                 topics=None,
                 from_user=from_user,
                 sender_chat=sender_chat,
+                sender_business_bot=sender_business_bot,
                 text=(
                     Str(message.message).init(entities) or None
                     if media is None or web_page_preview is not None