mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-13 04:56:35 +03:00
Get rid of all remaining async properties
This commit is contained in:
parent
ae47fd37b0
commit
efc9f4c414
|
@ -204,8 +204,7 @@ class ChatAction(EventBuilder):
|
||||||
return await self._client.delete_messages(
|
return await self._client.delete_messages(
|
||||||
await self.input_chat, [self.action_message], *args, **kwargs)
|
await self.input_chat, [self.action_message], *args, **kwargs)
|
||||||
|
|
||||||
@property
|
async def get_pinned_message(self):
|
||||||
async def pinned_message(self):
|
|
||||||
"""
|
"""
|
||||||
If ``new_pin`` is ``True``, this returns the
|
If ``new_pin`` is ``True``, this returns the
|
||||||
`telethon.tl.custom.message.Message` object that was pinned.
|
`telethon.tl.custom.message.Message` object that was pinned.
|
||||||
|
@ -230,48 +229,77 @@ class ChatAction(EventBuilder):
|
||||||
return self._pinned_message
|
return self._pinned_message
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def added_by(self):
|
def added_by(self):
|
||||||
"""
|
"""
|
||||||
The user who added ``users``, if applicable (``None`` otherwise).
|
The user who added ``users``, if applicable (``None`` otherwise).
|
||||||
"""
|
"""
|
||||||
if self._added_by and not isinstance(self._added_by, types.User):
|
if self._added_by and not isinstance(self._added_by, types.User):
|
||||||
aby = self._entities.get(utils.get_peer_id(self._added_by))
|
aby = self._entities.get(utils.get_peer_id(self._added_by))
|
||||||
if not aby:
|
if aby:
|
||||||
aby = await self._client.get_entity(self._added_by)
|
|
||||||
self._added_by = aby
|
self._added_by = aby
|
||||||
|
|
||||||
return self._added_by
|
return self._added_by
|
||||||
|
|
||||||
|
async def get_added_by(self):
|
||||||
|
"""
|
||||||
|
Returns `added_by` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if not self.added_by and self._added_by:
|
||||||
|
self._added_by = await self._client.get_entity(self._added_by)
|
||||||
|
|
||||||
|
return self._added_by
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def kicked_by(self):
|
def kicked_by(self):
|
||||||
"""
|
"""
|
||||||
The user who kicked ``users``, if applicable (``None`` otherwise).
|
The user who kicked ``users``, if applicable (``None`` otherwise).
|
||||||
"""
|
"""
|
||||||
if self._kicked_by and not isinstance(self._kicked_by, types.User):
|
if self._kicked_by and not isinstance(self._kicked_by, types.User):
|
||||||
kby = self._entities.get(utils.get_peer_id(self._kicked_by))
|
kby = self._entities.get(utils.get_peer_id(self._kicked_by))
|
||||||
if kby:
|
if kby:
|
||||||
kby = await self._client.get_entity(self._kicked_by)
|
|
||||||
self._kicked_by = kby
|
self._kicked_by = kby
|
||||||
|
|
||||||
return self._kicked_by
|
return self._kicked_by
|
||||||
|
|
||||||
|
async def get_kicked_by(self):
|
||||||
|
"""
|
||||||
|
Returns `kicked_by` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if not self.kicked_by and self._kicked_by:
|
||||||
|
self._kicked_by = await self._client.get_entity(self._kicked_by)
|
||||||
|
|
||||||
|
return self._kicked_by
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def user(self):
|
def user(self):
|
||||||
"""
|
"""
|
||||||
The first user that takes part in this action (e.g. joined).
|
The first user that takes part in this action (e.g. joined).
|
||||||
|
|
||||||
Might be ``None`` if the information can't be retrieved or
|
Might be ``None`` if the information can't be retrieved or
|
||||||
there is no user taking part.
|
there is no user taking part.
|
||||||
"""
|
"""
|
||||||
if await self.users:
|
if self.users:
|
||||||
return self._users[0]
|
return self._users[0]
|
||||||
|
|
||||||
@property
|
async def get_user(self):
|
||||||
async def input_user(self):
|
"""
|
||||||
|
Returns `user` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if self.users or await self.get_users():
|
||||||
|
return self._users[0]
|
||||||
|
|
||||||
|
def input_user(self):
|
||||||
"""
|
"""
|
||||||
Input version of the ``self.user`` property.
|
Input version of the ``self.user`` property.
|
||||||
"""
|
"""
|
||||||
if await self.input_users:
|
if self.input_users:
|
||||||
|
return self._input_users[0]
|
||||||
|
|
||||||
|
async def get_input_user(self):
|
||||||
|
"""
|
||||||
|
Returns `input_user` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if self.input_users or await self.get_input_users():
|
||||||
return self._input_users[0]
|
return self._input_users[0]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
@ -283,7 +311,7 @@ class ChatAction(EventBuilder):
|
||||||
return utils.get_peer_id(self._user_peers[0])
|
return utils.get_peer_id(self._user_peers[0])
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def users(self):
|
def users(self):
|
||||||
"""
|
"""
|
||||||
A list of users that take part in this action (e.g. joined).
|
A list of users that take part in this action (e.g. joined).
|
||||||
|
|
||||||
|
@ -294,6 +322,22 @@ class ChatAction(EventBuilder):
|
||||||
return []
|
return []
|
||||||
|
|
||||||
if self._users is None:
|
if self._users is None:
|
||||||
|
self._users = [
|
||||||
|
self._entities[utils.get_peer_id(peer)]
|
||||||
|
for peer in self._user_peers
|
||||||
|
if utils.get_peer_id(peer) in self._entities
|
||||||
|
]
|
||||||
|
|
||||||
|
return self._users
|
||||||
|
|
||||||
|
async def get_users(self):
|
||||||
|
"""
|
||||||
|
Returns `users` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if not self._user_peers:
|
||||||
|
return []
|
||||||
|
|
||||||
|
if self._users is None or len(self._users) != len(self._user_peers):
|
||||||
have, missing = [], []
|
have, missing = [], []
|
||||||
for peer in self._user_peers:
|
for peer in self._user_peers:
|
||||||
user = self._entities.get(utils.get_peer_id(peer))
|
user = self._entities.get(utils.get_peer_id(peer))
|
||||||
|
@ -312,7 +356,7 @@ class ChatAction(EventBuilder):
|
||||||
return self._users
|
return self._users
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def input_users(self):
|
def input_users(self):
|
||||||
"""
|
"""
|
||||||
Input version of the ``self.users`` property.
|
Input version of the ``self.users`` property.
|
||||||
"""
|
"""
|
||||||
|
@ -321,11 +365,18 @@ class ChatAction(EventBuilder):
|
||||||
for peer in self._user_peers:
|
for peer in self._user_peers:
|
||||||
try:
|
try:
|
||||||
self._input_users.append(
|
self._input_users.append(
|
||||||
await self._client.get_input_entity(peer)
|
self._client.session.get_input_entity(peer)
|
||||||
)
|
)
|
||||||
except (TypeError, ValueError):
|
except ValueError:
|
||||||
pass
|
pass
|
||||||
return self._input_users
|
return self._input_users or []
|
||||||
|
|
||||||
|
async def get_input_users(self):
|
||||||
|
"""
|
||||||
|
Returns `input_users` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
# TODO Maybe we could re-fetch the message
|
||||||
|
return self.input_users
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def user_ids(self):
|
def user_ids(self):
|
||||||
|
|
|
@ -88,10 +88,9 @@ class MessageRead(EventBuilder):
|
||||||
"""
|
"""
|
||||||
return self._message_ids
|
return self._message_ids
|
||||||
|
|
||||||
@property
|
async def get_messages(self):
|
||||||
async def messages(self):
|
|
||||||
"""
|
"""
|
||||||
The list of `telethon.tl.custom.message.Message`
|
Returns the list of `telethon.tl.custom.message.Message`
|
||||||
**which contents'** were read.
|
**which contents'** were read.
|
||||||
|
|
||||||
Use :meth:`is_read` if you need to check whether a message
|
Use :meth:`is_read` if you need to check whether a message
|
||||||
|
|
|
@ -148,16 +148,24 @@ class UserUpdate(EventBuilder):
|
||||||
self.uploading = self.video = True
|
self.uploading = self.video = True
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def user(self):
|
def user(self):
|
||||||
"""Alias around the chat (conversation)."""
|
"""Alias for `chat` (conversation)."""
|
||||||
return await self.chat
|
return self.chat
|
||||||
|
|
||||||
|
async def get_user(self):
|
||||||
|
"""Alias for `get_chat` (conversation)."""
|
||||||
|
return await self.get_chat()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def input_user(self):
|
def input_user(self):
|
||||||
"""Alias around the input chat."""
|
"""Alias for `input_chat`."""
|
||||||
return await self.input_chat
|
return self.input_chat
|
||||||
|
|
||||||
|
async def get_input_user(self):
|
||||||
|
"""Alias for `get_input_chat`."""
|
||||||
|
return await self.get_input_chat()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def user_id(self):
|
def user_id(self):
|
||||||
"""Alias around `chat_id`."""
|
"""Alias for `chat_id`."""
|
||||||
return self.chat_id
|
return self.chat_id
|
||||||
|
|
|
@ -79,7 +79,7 @@ class Dialog:
|
||||||
self.unread_count = dialog.unread_count
|
self.unread_count = dialog.unread_count
|
||||||
self.unread_mentions_count = dialog.unread_mentions_count
|
self.unread_mentions_count = dialog.unread_mentions_count
|
||||||
|
|
||||||
self.draft = Draft(client, dialog.peer, dialog.draft)
|
self.draft = Draft._from_dialog(client, dialog)
|
||||||
|
|
||||||
self.is_user = isinstance(self.entity, types.User)
|
self.is_user = isinstance(self.entity, types.User)
|
||||||
self.is_group = (
|
self.is_group = (
|
||||||
|
|
|
@ -5,7 +5,7 @@ from ..functions.messages import SaveDraftRequest
|
||||||
from ..types import UpdateDraftMessage, DraftMessage
|
from ..types import UpdateDraftMessage, DraftMessage
|
||||||
from ...errors import RPCError
|
from ...errors import RPCError
|
||||||
from ...extensions import markdown
|
from ...extensions import markdown
|
||||||
from ...utils import Default
|
from ...utils import Default, get_peer_id, get_input_peer
|
||||||
|
|
||||||
|
|
||||||
class Draft:
|
class Draft:
|
||||||
|
@ -24,9 +24,12 @@ class Draft:
|
||||||
reply_to_msg_id (`int`):
|
reply_to_msg_id (`int`):
|
||||||
The message ID that the draft will reply to.
|
The message ID that the draft will reply to.
|
||||||
"""
|
"""
|
||||||
def __init__(self, client, peer, draft):
|
def __init__(self, client, peer, draft, entity):
|
||||||
self._client = client
|
self._client = client
|
||||||
self._peer = peer
|
self._peer = peer
|
||||||
|
self._entity = entity
|
||||||
|
self._input_entity = get_input_peer(entity) if entity else None
|
||||||
|
|
||||||
if not draft or not isinstance(draft, DraftMessage):
|
if not draft or not isinstance(draft, DraftMessage):
|
||||||
draft = DraftMessage('', None, None, None, None)
|
draft = DraftMessage('', None, None, None, None)
|
||||||
|
|
||||||
|
@ -37,28 +40,57 @@ class Draft:
|
||||||
self.reply_to_msg_id = draft.reply_to_msg_id
|
self.reply_to_msg_id = draft.reply_to_msg_id
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _from_update(cls, client, update):
|
def _from_dialog(cls, client, dialog):
|
||||||
if not isinstance(update, UpdateDraftMessage):
|
return cls(client=client, peer=dialog.dialog.peer,
|
||||||
raise TypeError(
|
draft=dialog.dialog.draft, entity=dialog.entity)
|
||||||
'You can only create a new `Draft` from a corresponding '
|
|
||||||
'`UpdateDraftMessage` object.'
|
|
||||||
)
|
|
||||||
|
|
||||||
return cls(client=client, peer=update.peer, draft=update.draft)
|
@classmethod
|
||||||
|
def _from_update(cls, client, update, entities=None):
|
||||||
|
assert isinstance(update, UpdateDraftMessage)
|
||||||
|
return cls(client=client, peer=update.peer, draft=update.draft,
|
||||||
|
entity=(entities or {}).get(get_peer_id(update.peer)))
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def entity(self):
|
def entity(self):
|
||||||
"""
|
"""
|
||||||
The entity that belongs to this dialog (user, chat or channel).
|
The entity that belongs to this dialog (user, chat or channel).
|
||||||
"""
|
"""
|
||||||
return await self._client.get_entity(self._peer)
|
return self._entity
|
||||||
|
|
||||||
@property
|
@property
|
||||||
async def input_entity(self):
|
def input_entity(self):
|
||||||
"""
|
"""
|
||||||
Input version of the entity.
|
Input version of the entity.
|
||||||
"""
|
"""
|
||||||
return await self._client.get_input_entity(self._peer)
|
if not self._input_entity:
|
||||||
|
try:
|
||||||
|
self._input_entity =\
|
||||||
|
self._client.session.get_input_entity(self._peer)
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return self._input_entity
|
||||||
|
|
||||||
|
async def get_entity(self):
|
||||||
|
"""
|
||||||
|
Returns `entity` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
if not self.entity and await self.get_input_entity():
|
||||||
|
try:
|
||||||
|
self._entity =\
|
||||||
|
self._client.get_entity(self._input_entity)
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return self._entity
|
||||||
|
|
||||||
|
async def get_input_entity(self):
|
||||||
|
"""
|
||||||
|
Returns `input_entity` but will make an API call if necessary.
|
||||||
|
"""
|
||||||
|
# We don't actually have an API call we can make yet
|
||||||
|
# to get more info, but keep this method for consistency.
|
||||||
|
return self.input_entity
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def text(self):
|
def text(self):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user