3 uQhZ@snddlZddlmZmZddlmZddlmZddlmZddl m Z dd l m Z m Z mZGd d d ZdS) N)TLObjecttypes)SaveDraftRequest) DraftMessage)RPCError)markdown)get_input_peerget_peer get_peer_idc@seZdZdZddZeddZeddZdd Zd d Z ed d Z eddZ eddZ ddfdfddZ dffddZddZddZddZdd ZdS)!Drafta Custom class that encapsulates a draft on the Telegram servers, providing an abstraction to change the message conveniently. The library will return instances of this class when calling :meth:`get_drafts()`. Args: date (`datetime`): The date of the draft. link_preview (`bool`): Whether the link preview is enabled or not. reply_to_msg_id (`int`): The message ID that the draft will reply to. cCs||_t||_||_|r"t|nd|_| s:t|t rJtddddd}tj |j |j |_ |j |_ |j|_|j |_t|jtjr|jjnd|_dS)N)_clientr _peer_entityr _input_entity isinstancerr Zunparsemessageentities_text _raw_textdate no_webpage link_previewreply_torInputReplyToMessagereply_to_msg_id)selfcliententityZdraftr!y"|jjjt|jddj|_Wntk r<YnX|jS)z. Input version of the entity. F)Zadd_mark)rrZ_mb_entity_cachegetr rZ_as_input_peerAttributeError)rr!r!r" input_entity1szDraft.input_entityc sL|j rF|jIdHrFy|jj|jIdH|_Wntk rDYnX|jS)zJ Returns `entity` but will make an API call if necessary. N)r get_input_entityr get_entityrr ValueError)rr!r!r"r(?s zDraft.get_entitycs|jS)zP Returns `input_entity` but will make an API call if necessary. )r&)rr!r!r"r'LszDraft.get_input_entitycCs|jS)z The markdown text contained in the draft. It will be empty if there is no text (and hence no draft is set). )r)rr!r!r"textTsz Draft.textcCs|jS)z The raw (text without formatting) contained in the draft. It will be empty if there is no text (thus draft not set). )r)rr!r!r"raw_text\szDraft.raw_textcCs|j S)zM Convenience bool to determine if the draft is empty or not. )r)rr!r!r"is_emptydszDraft.is_emptyNrcs|dkr|j}|dkr|j}|dkr*|j}|jj||IdH\}}|jt|j|| |dkr^dntj||dIdH}|r||_||_ ||_||_t j j t j j d|_|S)aT Changes the draft message on the Telegram servers. The changes are reflected in this object. :param str text: New text of the draft. Preserved if left as None. :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. :return bool: `True` on success. Nr)Zpeerrrrr)tz)rrrrZ_parse_message_textrrrrrdatetimenowtimezoneutcr)rr*r parse_moderr+rresultr!r!r" set_messageks(zDraft.set_messageTcs*|jj|j|j|j|j||dIdHdS)z Sends the contents of this draft to the dialog. This is just a wrapper around ``send_message(dialog.input_entity, *args, **kwargs)``. )rrr2Z clear_draftN)r send_messagerr*rr)rclearr2r!r!r"sends z Draft.sendcs|jddIdHS)zD Deletes this draft, and returns `True` on success. r)r*N)r4)rr!r!r"deletesz Draft.deletecCsLy |j}Wn&tk r0}z |}WYdd}~XnXd|j||j|j|jdS)Nr )_r*r rrr)r rr*rrr)rr er!r!r"to_dicts z Draft.to_dictcCstj|jS)N)r pretty_formatr;)rr!r!r"__str__sz Draft.__str__cCstj|jddS)Nr)indent)rr<r;)rr!r!r" stringifyszDraft.stringify)__name__ __module__ __qualname____doc__r#propertyr r&r(r'r*r+r,r4r7r8r;r=r?r!r!r!r"r s        . r )r.rrrZfunctions.messagesrrerrorsr extensionsr utilsr r r r r!r!r!r"s