hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-18 02:56:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Rename and reorder some params in Draft for consistency (#673)

Browse Source
This commit is contained in:
Lonami Exo 2018-03-08 13:04:01 +01:00
parent 8cefb22e14
commit 9d46bb35c8
1 changed files with 26 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
telethon/tl/custom/draft.py
View File

@ -18,7 +18,7 @@ class Draft:
self._text = markdown.unparse(draft.message, draft.entities) self._text = markdown.unparse(draft.message, draft.entities)
self._raw_text = draft.message self._raw_text = draft.message
self.date = draft.date self.date = draft.date
self.no_webpage = draft.no_webpage self.link_preview = not draft.no_webpage
self.reply_to_msg_id = draft.reply_to_msg_id self.reply_to_msg_id = draft.reply_to_msg_id
@classmethod @classmethod
@ -47,48 +47,54 @@ class Draft:
def raw_text(self): def raw_text(self):
return self._raw_text return self._raw_text
def set_message(self, text, no_webpage=None, reply_to_msg_id=None, def set_message(self, text=None, reply_to=0, parse_mode='md',
parse_mode='md'): link_preview=None):
""" """
Changes the draft message on the Telegram servers. The changes are Changes the draft message on the Telegram servers. The changes are
reflected in this object. Changing only individual attributes like for reflected in this object.
example the ``reply_to_msg_id`` should be done by providing the current
values of this object, like so:
draft.set_message(
draft.text,
no_webpage=draft.no_webpage,
reply_to_msg_id=NEW_VALUE,
entities=draft.entities
)
:param str text: New text of the draft. :param str text: New text of the draft.
:param bool no_webpage: Whether to attach a web page preview. Preserved if left as None.
:param int reply_to_msg_id: Message id to reply to.
:param int reply_to: Message ID to reply to.
Preserved if left as 0, erased if set to None.
:param bool link_preview: Whether to attach a web page preview.
Preserved if left as None.
:param str parse_mode: The parse mode to be used for the text. :param str parse_mode: The parse mode to be used for the text.
:return bool: ``True`` on success. :return bool: ``True`` on success.
""" """
if text is None:
text = self._text
if reply_to == 0:
reply_to = self.reply_to_msg_id
if link_preview is None:
link_preview = self.link_preview
raw_text, entities = self._client._parse_message_text(text, parse_mode) raw_text, entities = self._client._parse_message_text(text, parse_mode)
result = self._client(SaveDraftRequest( result = self._client(SaveDraftRequest(
peer=self._peer, peer=self._peer,
message=raw_text, message=raw_text,
no_webpage=no_webpage, no_webpage=not link_preview,
reply_to_msg_id=reply_to_msg_id, reply_to_msg_id=reply_to,
entities=entities entities=entities
)) ))
if result: if result:
self._text = text self._text = text
self._raw_text = raw_text self._raw_text = raw_text
self.no_webpage = no_webpage self.link_preview = link_preview
self.reply_to_msg_id = reply_to_msg_id self.reply_to_msg_id = reply_to
return result return result
def send(self, clear=True, parse_mode='md'): def send(self, clear=True, parse_mode='md'):
self._client.send_message(self._peer, self.text, self._client.send_message(self._peer, self.text,
reply_to=self.reply_to_msg_id, reply_to=self.reply_to_msg_id,
link_preview=not self.no_webpage, link_preview=self.link_preview,
parse_mode=parse_mode, parse_mode=parse_mode,
clear_draft=clear) clear_draft=clear)