Add support for direct messages

A little update :)

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -300,6 +300,7 @@ def get_title_list(s: str) -> list:
  delete_chat_photo
  set_chat_title
  set_chat_description
+ set_chat_direct_messages_group
  set_chat_permissions
  pin_chat_message
  unpin_chat_message
@@ -312,6 +313,8 @@ def get_title_list(s: str) -> list:
  get_chat_members_count
  get_dialogs
  get_dialogs_count
+ get_direct_messages_topics_by_id
+ get_direct_messages_topics
  set_chat_username
  archive_chats
  unarchive_chats
@@ -693,6 +696,8 @@ def get_title_list(s: str) -> list:
  PaidMediaPreview
  PaidMessagesRefunded
  PaidMessagesPriceChanged
+ DirectMessagePriceChanged
+ DirectMessagesTopic
  PaymentForm
  ChatBoost
  ContactRegistered

compiler/errors/source/400_BAD_REQUEST.tsv

@@ -60,12 +60,13 @@ BROADCAST_CALLS_DISABLED	Broadcast calls disabled.
 BROADCAST_ID_INVALIDThe channel is invalid.
 BROADCAST_PUBLIC_VOTERS_FORBIDDENPolls with public voters cannot be sent in channels.
 BROADCAST_REQUIREDThis method can only be used with a channel.
-BUSINESS_BOT_MISSINGBusiness bot missing.
+BUSINESS_BOT_MISSINGThe business bot is missing.
 BUSINESS_CONNECTION_INVALIDThe business connection is invalid.
 BUSINESS_CONNECTION_NOT_ALLOWEDThis business connection is not allowed.
 BUSINESS_PEER_INVALIDMessages can't be set to the specified peer through the current [business connection](https://core.telegram.org/api/business#connected-bots).
 BUSINESS_PEER_USAGE_MISSINGYou cannot send a message to a user through a [business connection](https://core.telegram.org/api/business#connected-bots) if the user hasn't recently contacted us.
 BUSINESS_RECIPIENTS_EMPTYYou didn't set any flag in inputBusinessBotRecipients, thus the bot cannot work with *any* peer.
+BUSINESS_RECIPIENTS_INVALIDThe specified inputBusinessBotRecipients is invalid.
 BUSINESS_WORK_HOURS_EMPTYNo work hours were specified.
 BUSINESS_WORK_HOURS_PERIOD_INVALIDThe specified work hours are invalid, see [here »](https://core.telegram.org/api/business#opening-hours) for the exact requirements.
 BUTTON_COPY_TEXT_INVALIDThe specified [keyboardButtonCopy](https://core.telegram.org/constructor/keyboardButtonCopy).`copy_text` is invalid.
@@ -91,6 +92,7 @@ CHANNEL_BANNED	The channel is banned.
 CHANNEL_FORUM_MISSINGThis supergroup is not a forum.
 CHANNEL_ID_INVALIDThe specified supergroup ID is invalid.
 CHANNEL_INVALIDThe provided channel is invalid.
+CHANNEL_MONOFORUM_UNSUPPORTEDYou cannot use this method in direct chats.
 CHANNEL_PARICIPANT_MISSINGThe current user is not in the channel.
 CHANNEL_PRIVATEYou haven't joined this channel/supergroup.
 CHANNEL_TOO_BIGThis channel is too big.
@@ -150,6 +152,7 @@ DATE_EMPTY	The date argument is empty.
 DC_ID_INVALIDThe provided dc_id is invalid.
 DH_G_A_INVALIDThe g_a parameter is invalid.
 DOCUMENT_INVALIDThe specified document is invalid.
+EFFECT_ID_INVALIDThe specified effect id is invalid.
 EMAIL_HASH_EXPIREDThe email hash expired and cannot be used to verify it.
 EMAIL_INVALIDThe specified email is invalid.
 EMAIL_NOT_ALLOWEDThe specified email cannot be used to complete the operation.
@@ -219,6 +222,7 @@ FORUM_ENABLED	You can't execute the specified action because the group is a [for
 FRESH_CHANGE_ADMINS_FORBIDDENYou can't change administrator settings in this chat because your session was logged-in recently.
 FROM_MESSAGE_BOT_DISABLEDBots can't use fromMessage min constructors.
 FROM_PEER_INVALIDThe specified from_id is invalid.
+FROZEN_PARTICIPANT_MISSINGThe account is frozen and cannot access the chat participant.
 GAME_BOT_INVALIDYou cannot send that game with the current bot.
 GENERAL_MODIFY_ICON_FORBIDDENYou can't modify the icon of the "General" topic.
 GEO_POINT_INVALIDInvalid geo point provided.
@@ -354,7 +358,7 @@ PEER_TYPES_INVALID	The passed [keyboardButtonSwitchInline](https://core.telegram
 PERSISTENT_TIMESTAMP_EMPTYThe pts argument is empty.
 PERSISTENT_TIMESTAMP_INVALIDThe persistent timestamp is invalid.
 PHONE_CODE_EMPTYThe phone code is missing.
-PHONE_CODE_EXPIREDThe provided confirmation code has expired.
+PHONE_CODE_EXPIREDThe provided confirmation code has expired. Make sure you haven't sent it to anyone else and use the code on the same connection.
 PHONE_CODE_HASH_EMPTYThe phone code hash is missing.
 PHONE_CODE_INVALIDThe provided confirmation code is invalid.
 PHONE_HASH_EXPIREDAn invalid or expired phone code hash was provided.
@@ -389,6 +393,7 @@ POLL_OPTION_INVALID	IA poll option used invalid data (the data may be too long).
 POLL_QUESTION_INVALIDThe poll question is invalid.
 POLL_UNSUPPORTEDThis layer does not support polls in the invoked method.
 POLL_VOTE_REQUIREDCast a vote in the poll before calling this method.
+PRECHECKOUT_FAILEDPrecheckout failed.
 PREMIUM_ACCOUNT_REQUIREDA premium account is required to execute this action.
 PREMIUM_GIFTCODE_WAS_REFUNDEDThis gift code can't be redeemed because the giveaway organizer requested a refund.
 PRICING_CHAT_INVALIDThe pricing for the [subscription](https://core.telegram.org/api/subscriptions) is invalid, the maximum price is specified in the [`stars_subscription_amount_max` config key »](https://core.telegram.org/api/config#stars-subscription-amount-max).
@@ -425,6 +430,7 @@ REPLY_MARKUP_TOO_LONG	The specified reply markup is too long.
 REPLY_MESSAGES_TOO_MUCHEach shortcut can contain a maximum of [appConfig.`quick_reply_messages_limit`](https://core.telegram.org/api/config#quick-reply-messages-limit) messages, the limit was reached.
 REPLY_MESSAGE_ID_INVALIDThe specified reply message id is invalid.
 REPLY_TO_INVALIDThe specified `reply_to` field is invalid.
+REPLY_TO_MONOFORUM_PEER_INVALIDThe specified `reply_to` field is invalid.
 REPLY_TO_USER_INVALIDThe replied-to user is invalid.
 REQUEST_TOKEN_INVALIDThe master DC did not accept the `request_token` from the CDN DC. Continue downloading the file from the master DC using upload.getFile.
 RESET_REQUEST_MISSINGNo password reset is in progress.
@@ -439,6 +445,7 @@ RINGTONE_INVALID	The specified ringtone is invalid.
 RINGTONE_MIME_INVALIDThe MIME type for the ringtone is invalid.
 RSA_DECRYPT_FAILEDInternal RSA decryption failed.
 SAVED_DIALOGS_UNSUPPORTEDYou cannot use this method.
+SAVED_PEER_INVALIDThe saved peer is invalid.
 SCHEDULE_BOT_NOT_ALLOWEDBots are not allowed to schedule messages.
 SCHEDULE_DATE_INVALIDInvalid schedule date provided.
 SCHEDULE_DATE_TOO_LATEThe date you tried to schedule is too far in the future (more than one year).
@@ -471,6 +478,7 @@ STARGIFT_ALREADY_CONVERTED	The provided star gift already converted to stars.
 STARGIFT_ALREADY_UPGRADEDThis star gift was already upgraded before.
 STARGIFT_CONVERT_TOO_OLDThis gift can no longer be converted into stars.
 STARGIFT_INVALIDThe passed [inputInvoiceStarGift](https://core.telegram.org/constructor/inputInvoiceStarGift) is invalid.
+STARGIFT_MESSAGE_INVALIDThe provided message for this gift is invalid.
 STARGIFT_NOT_FOUNDThe specified star gift was not found.
 STARGIFT_NOT_OWNERYou can't control this gift because it doesn't belong to you.
 STARGIFT_RESELL_NOT_ALLOWEDYou can't buy this gift. Someone else may have already bought it.
@@ -480,8 +488,11 @@ STARGIFT_UPGRADE_UNAVAILABLE	Gift is not available for upgrade yet.
 STARGIFT_USAGE_LIMITEDThe gift is sold out.
 STARREF_AWAITING_ENDThe previous referral program was terminated less than 24 hours ago: further changes can be made after the date specified in userFull.starref_program.end_date.
 STARREF_HASH_REVOKEDThe specified affiliate link was already revoked.
+STARREF_MONTHS_TOO_LOWThe `duration_months` parameter must be between 0 and 36.
 STARREF_PERMILLE_INVALIDThe specified commission_permille is invalid: the minimum and maximum values for this parameter are contained in the [starref_min_commission_permille](https://core.telegram.org/api/config#starref-min-commission-permille) and [starref_max_commission_permille](https://core.telegram.org/api/config#starref-max-commission-permille) client configuration parameters.
 STARREF_PERMILLE_TOO_LOWThe specified commission_permille is too low: the minimum and maximum values for this parameter are contained in the [starref_min_commission_permille](https://core.telegram.org/api/config#starref-min-commission-permille) and [starref_max_commission_permille](https://core.telegram.org/api/config#starref-max-commission-permille) client configuration parameters.
+STARS_AMOUNT_INVALIDThe specified `paid_stars` amount is invalid.
+STARS_FORM_AMOUNT_MISMATCHThe stars form amount mismatch.
 STARS_INVOICE_INVALIDThe specified Telegram Star invoice is invalid.
 STARS_PAYMENT_REQUIREDTo import this chat invite link, you must first [pay for the associated Telegram Star subscription »](https://core.telegram.org/api/subscriptions#channel-subscriptions).
 START_PARAM_EMPTYThe start parameter is empty.

compiler/errors/source/403_FORBIDDEN.tsv

@@ -2,6 +2,7 @@ id	message
 ALLOW_PAYMENT_REQUIRED_XThe user has enabled paid messages with {value} stars.
 ANONYMOUS_REACTIONS_DISABLEDSorry, anonymous administrators cannot leave reactions or participate in polls.
 BOT_ACCESS_FORBIDDENYou can't access this bot.
+BOT_OWNER_REQUIREDYou must be the owner of the bot to perform this action.
 BROADCAST_FORBIDDENThis request can't be used in channels.
 CHANNEL_PUBLIC_GROUP_NAThe channel/supergroup is not available.
 CHAT_ACTION_FORBIDDENYou cannot execute this action.

pyrogram/enums/chat_type.py

@@ -39,5 +39,8 @@ class ChatType(AutoName):
  CHANNEL = auto()
  "Chat is a channel"
 
+ FORUM = auto()
+ "Chat is a forum"
+
  DIRECT = auto()
  "Chat is a direct with a channel"

pyrogram/enums/message_service_type.py

@@ -174,5 +174,8 @@ class MessageServiceType(AutoName):
  PAID_MESSAGES_REFUNDED = auto()
  "Refunded paid messages"
 
- PAID_MESSAGES_PRICE = auto()
+ PAID_MESSAGES_PRICE_CHANGED = auto()
  "Paid messages price"
+
+ DIRECT_MESSAGE_PRICE_CHANGED = auto()
+ "Direct message price"

pyrogram/filters.py

@@ -549,7 +549,7 @@ async def private_filter(_, __, m: Message):
 
 # region group_filter
 async def group_filter(_, __, m: Message):
- return bool(m.chat and m.chat.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP})
+ return bool(m.chat and m.chat.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP, enums.ChatType.FORUM})
 
 
 group = create(group_filter)
@@ -567,6 +567,17 @@ async def channel_filter(_, __, m: Message):
 """Filter messages sent in channels."""
 
 
+# endregion
+
+# region direct_filter
+async def direct_filter(_, __, m: Message):
+ return bool(m.chat and m.chat.type == enums.ChatType.DIRECT)
+
+
+direct = create(direct_filter)
+"""Filter messages sent in direct."""
+
+
 # endregion
 
 # region forum_filter

pyrogram/methods/bots/send_inline_bot_result.py

@@ -33,6 +33,7 @@ async def send_inline_bot_result(
  result_id: str,
  disable_notification: Optional[bool] = None,
  message_thread_id: Optional[int] = None,
+ direct_messages_chat_topic_id: int = None,
  reply_parameters: Optional["types.ReplyParameters"] = None,
  paid_message_star_count: int = None,
  schedule_date: datetime = None,
@@ -68,7 +69,11 @@ async def send_inline_bot_result(
 
  message_thread_id (``int``, *optional*):
  Unique identifier of a message thread to which the message belongs.
- For supergroups only.
+ For forums only.
+
+ direct_messages_chat_topic_id (``int``, *optional*):
+ Unique identifier of the topic in a channel direct messages chat administered by the current user.
+ For directs only only.
 
  reply_parameters (:obj:`~pyrogram.types.ReplyParameters`, *optional*):
  Describes reply parameters for the message that is being sent.
@@ -153,7 +158,8 @@ async def send_inline_bot_result(
  reply_to=await utils.get_reply_to(
  self,
  reply_parameters,
- message_thread_id
+ message_thread_id,
+ direct_messages_chat_topic_id
  ),
  schedule_date=utils.datetime_to_timestamp(schedule_date),
  allow_paid_stars=paid_message_star_count

pyrogram/methods/chats/__init__.py

@@ -43,6 +43,8 @@
 from .get_similar_channels import GetSimilarChannels
 from .get_suitable_discussion_chats import GetSuitableDiscussionChats
 from .get_dialogs import GetDialogs
+from .get_direct_messages_topics_by_id import GetDirectMessagesTopicsByID
+from .get_direct_messages_topics import GetDirectMessagesTopics
 from .get_dialogs_count import GetDialogsCount
 from .get_folders import GetFolders
 from .get_forum_topics import GetForumTopics
@@ -60,6 +62,7 @@
 from .restrict_chat_member import RestrictChatMember
 from .set_administrator_title import SetAdministratorTitle
 from .set_chat_description import SetChatDescription
+from .set_chat_direct_messages_group import SetChatDirectMessagesGroup
 from .set_chat_permissions import SetChatPermissions
 from .set_chat_discussion_group import SetChatDiscussionGroup
 from .set_chat_photo import SetChatPhoto
@@ -102,6 +105,7 @@ class Chats(
  SetChatTitle,
  SetChatTTL,
  SetChatDescription,
+ SetChatDirectMessagesGroup,
  PinChatMessage,
  UnpinChatMessage,
  PinForumTopic,
@@ -110,6 +114,8 @@ class Chats(
  UpdateColor,
  EditFolder,
  GetDialogs,
+ GetDirectMessagesTopicsByID,
+ GetDirectMessagesTopics,
  GetChatMembersCount,
  SetChatUsername,
  SetChatPermissions,

pyrogram/methods/chats/get_direct_messages_topics.py

@@ -0,0 +1,109 @@
+# Pyrogram - Telegram MTProto API Client Library for Python
+# Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+# This file is part of Pyrogram.
+#
+# Pyrogram is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pyrogram is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
+
+from typing import AsyncGenerator, Optional, Union
+
+import pyrogram
+from pyrogram import raw, types, utils
+
+
+class GetDirectMessagesTopics:
+ async def get_direct_messages_topics(
+ self: "pyrogram.Client",
+ chat_id: Union[int, str],
+ limit: int = 0,
+ exclude_pinned: Optional[bool] = None
+ ) -> AsyncGenerator["types.DirectMessagesTopic", None]:
+ """Get one or more topic from a direct messages channel chat.
+
+ .. include:: /_includes/usable-by/users.rst
+
+ Parameters:
+ chat_id (``int`` | ``str``):
+ Unique identifier (int) or username (str) of the target chat.
+
+ limit (``int``, *optional*):
+ Limits the number of topics to be retrieved.
+ By default, no limit is applied and all topics are returned.
+
+ exclude_pinned (``bool``, *optional*):
+ Exclude pinned topics.
+
+ Returns:
+ ``Generator``: A generator yielding :obj:`~pyrogram.types.DirectMessagesTopic` objects.
+
+ Example:
+ .. code-block:: python
+
+ # Iterate through all topics
+ async for topic in app.get_direct_messages_topics(chat_id):
+ print(topic)
+ """
+ current = 0
+ total = limit or (1 << 31) - 1
+ limit = min(100, total)
+
+ offset_date = 0
+ offset_id = 0
+ offset_peer = raw.types.InputPeerEmpty()
+
+ while True:
+ r = await self.invoke(
+ raw.functions.messages.GetSavedDialogs(
+ offset_date=offset_date,
+ offset_id=offset_id,
+ offset_peer=offset_peer,
+ limit=limit,
+ hash=0,
+ exclude_pinned=exclude_pinned,
+ parent_peer=await self.resolve_peer(chat_id)
+ )
+ )
+
+ users = {i.id: i for i in r.users}
+ chats = {i.id: i for i in r.chats}
+
+ messages = {}
+
+ for message in r.messages:
+ if isinstance(message, raw.types.MessageEmpty):
+ continue
+
+ messages[message.id] = await types.Message._parse(self, message, users, chats)
+
+ topics = []
+
+ for topic in r.dialogs:
+ topics.append(types.DirectMessagesTopic._parse(topic, messages, users, chats))
+
+ if not topics:
+ return
+
+ last = topics[-1]
+
+ offset_id = last.last_message.id
+ offset_date = utils.datetime_to_timestamp(last.last_message.date)
+ offset_peer = await self.resolve_peer(last.id)
+
+ for topic in topics:
+ yield topic
+
+ current += 1
+
+ if current >= total:
+ return