U nhZ@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 unparsemessageentities_text _raw_textdate no_webpage link_previewreply_torInputReplyToMessagereply_to_msg_id)selfcliententitydraftr#b/var/www/html/peyman_registration/venvv2/lib/python3.8/site-packages/telethonv2/tl/custom/draft.py__init__s  zDraft.__init__cCs|jS)zQ The entity that belongs to this dialog (user, chat or channel). )rrr#r#r$r!*sz Draft.entitycCsD|js>z"|jjt|jdd|_Wntk r<YnX|jS)z. Input version of the entity. F)add_mark)rr_mb_entity_cachegetr r_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. r0r&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)peerrrrr)tz)rrrrZ_parse_message_textrrrrrdatetimenowtimezoneutcr)rr1r parse_moderr2rresultr#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)``. )rrr: clear_draftN)r send_messagerr1rr)rclearr:r#r#r$sendsz Draft.sendcs|jddIdHS)zD Deletes this draft, and returns `True` on success. r)r1N)r<r&r#r#r$deletesz Draft.deletec CsLz |j}Wn&tk r0}z|}W5d}~XYnXd|j||j|j|jdS)Nr )_r1r!rrr)r!rr1rrr)rr!er#r#r$to_dicts z Draft.to_dictcCst|S)Nr pretty_formatrDr&r#r#r$__str__sz Draft.__str__cCstj|ddS)Nr)indentrEr&r#r#r$ stringifyszDraft.stringify)Nrr#N)Tr#)__name__ __module__ __qualname____doc__r%propertyr!r,r.r-r1r2r3r<r@rArDrGrIr#r#r#r$r s0       0 r )r6rrrZfunctions.messagesrrerrorsr extensionsr utilsr r r r r#r#r#r$s