U hZ@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 d"ddZ d#ddZddZddZddZd d!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|_|r6t|tsFtddddd}t |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!z"|jjt|jdd|_Wntk r<YnX|jS)z. Input version of the entity. F)Zadd_mark)rrZ_mb_entity_cachegetr rZ_as_input_peerAttributeErrorr$r!r!r" input_entity1s zDraft.input_entitycsJ|jsD|IdHrDz|j|jIdH|_Wntk rBYnX|jS)zJ Returns `entity` but will make an API call if necessary. N)r get_input_entityr get_entityrr ValueErrorr$r!r!r"r)?szDraft.get_entitycs|jS)zP Returns `input_entity` but will make an API call if necessary. )r'r$r!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). rr$r!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). )rr$r!r!r"raw_text\szDraft.raw_textcCs|j S)zM Convenience bool to determine if the draft is empty or not. r+r$r!r!r"is_emptydszDraft.is_emptyNrr!c s|dkr|j}|dkr|j}|dkr*|j}|j||IdH\}}|t|j|| |dkr^dnt||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)``. )rrr4Z clear_draftN)r send_messagerr,rr)rclearr4r!r!r"sendsz Draft.sendcs|jddIdHS)zD Deletes this draft, and returns `True` on success. r)r,N)r6r$r!r!r"deletesz Draft.deletec CsLz |j}Wn&tk r0}z|}W5d}~XYnXd|j||j|j|jdS)Nr )_r,r rrr)r rr,rrr)rr er!r!r"to_dicts z Draft.to_dictcCst|S)NrZ pretty_formatr=r$r!r!r"__str__sz Draft.__str__cCstj|ddS)Nr)indentr>r$r!r!r" stringifyszDraft.stringify)Nrr!N)Tr!)__name__ __module__ __qualname____doc__r#propertyr r'r)r(r,r-r.r6r9r:r=r?rAr!r!r!r"r s0       0 r )r0rrrZfunctions.messagesrrerrorsr extensionsr utilsr r r r r!r!r!r"s