mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-26 19:33:45 +03:00
129 lines
4.5 KiB
Python
129 lines
4.5 KiB
Python
from . import Draft
|
|
from .. import TLObject, types, functions
|
|
from ... import utils
|
|
|
|
|
|
class Dialog:
|
|
"""
|
|
Custom class that encapsulates a dialog (an open "conversation" with
|
|
someone, a group or a channel) providing an abstraction to easily
|
|
access the input version/normal entity/message etc. The library will
|
|
return instances of this class when calling :meth:`.get_dialogs()`.
|
|
|
|
Args:
|
|
dialog (:tl:`Dialog`):
|
|
The original ``Dialog`` instance.
|
|
|
|
pinned (`bool`):
|
|
Whether this dialog is pinned to the top or not.
|
|
|
|
message (:tl:`Message`):
|
|
The last message sent on this dialog. Note that this member
|
|
will not be updated when new messages arrive, it's only set
|
|
on creation of the instance.
|
|
|
|
date (`datetime`):
|
|
The date of the last message sent on this dialog.
|
|
|
|
entity (`entity`):
|
|
The entity that belongs to this dialog (user, chat or channel).
|
|
|
|
input_entity (:tl:`InputPeer`):
|
|
Input version of the entity.
|
|
|
|
id (`int`):
|
|
The marked ID of the entity, which is guaranteed to be unique.
|
|
|
|
name (`str`):
|
|
Display name for this dialog. For chats and channels this is
|
|
their title, and for users it's "First-Name Last-Name".
|
|
|
|
title (`str`):
|
|
Alias for `name`.
|
|
|
|
unread_count (`int`):
|
|
How many messages are currently unread in this dialog. Note that
|
|
this value won't update when new messages arrive.
|
|
|
|
unread_mentions_count (`int`):
|
|
How many mentions are currently unread in this dialog. Note that
|
|
this value won't update when new messages arrive.
|
|
|
|
draft (`telethon.tl.custom.draft.Draft`):
|
|
The draft object in this dialog. It will not be ``None``,
|
|
so you can call ``draft.set_message(...)``.
|
|
|
|
is_user (`bool`):
|
|
``True`` if the `entity` is a :tl:`User`.
|
|
|
|
is_group (`bool`):
|
|
``True`` if the `entity` is a :tl:`Chat`
|
|
or a :tl:`Channel` megagroup.
|
|
|
|
is_channel (`bool`):
|
|
``True`` if the `entity` is a :tl:`Channel`.
|
|
"""
|
|
def __init__(self, client, dialog, entities, messages):
|
|
# Both entities and messages being dicts {ID: item}
|
|
self._client = client
|
|
self.dialog = dialog
|
|
self.pinned = bool(dialog.pinned)
|
|
self.message = messages.get(dialog.top_message, None)
|
|
self.date = getattr(self.message, 'date', None)
|
|
|
|
self.entity = entities[utils.get_peer_id(dialog.peer)]
|
|
self.input_entity = utils.get_input_peer(self.entity)
|
|
self.id = utils.get_peer_id(self.entity) # ^ May be InputPeerSelf()
|
|
self.name = self.title = utils.get_display_name(self.entity)
|
|
|
|
self.unread_count = dialog.unread_count
|
|
self.unread_mentions_count = dialog.unread_mentions_count
|
|
|
|
self.draft = Draft._from_dialog(client, self)
|
|
|
|
self.is_user = isinstance(self.entity, types.User)
|
|
self.is_group = (
|
|
isinstance(self.entity, (types.Chat, types.ChatForbidden)) or
|
|
(isinstance(self.entity, types.Channel) and self.entity.megagroup)
|
|
)
|
|
self.is_channel = isinstance(self.entity, types.Channel)
|
|
|
|
async def send_message(self, *args, **kwargs):
|
|
"""
|
|
Sends a message to this dialog. This is just a wrapper around
|
|
``client.send_message(dialog.input_entity, *args, **kwargs)``.
|
|
"""
|
|
return await self._client.send_message(
|
|
self.input_entity, *args, **kwargs)
|
|
|
|
async def delete(self):
|
|
"""
|
|
Deletes the dialog from your dialog list. If you own the
|
|
channel this won't destroy it, only delete it from the list.
|
|
"""
|
|
if self.is_channel:
|
|
await self._client(functions.channels.LeaveChannelRequest(
|
|
self.input_entity))
|
|
else:
|
|
if self.is_group:
|
|
await self._client(functions.messages.DeleteChatUserRequest(
|
|
self.entity.id, types.InputPeerSelf()))
|
|
await self._client(functions.messages.DeleteHistoryRequest(
|
|
self.input_entity, 0))
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'_': 'Dialog',
|
|
'name': self.name,
|
|
'date': self.date,
|
|
'draft': self.draft,
|
|
'message': self.message,
|
|
'entity': self.entity,
|
|
}
|
|
|
|
def __str__(self):
|
|
return TLObject.pretty_format(self.to_dict())
|
|
|
|
def stringify(self):
|
|
return TLObject.pretty_format(self.to_dict(), indent=0)
|