o XzjZ@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||_||_|rt|nd|_|rt|ts#tddddd}t |j |j |_ |j |_ |j|_|j |_t|jtjrG|jj|_dSd|_dS)N)_clientr _peer_entityr _input_entity isinstancerr unparsemessageentities_text _raw_textdate no_webpage link_previewreply_torInputReplyToMessagereply_to_msg_id)selfcliententitydraftr#S/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/tl/custom/draft.py__init__s  &zDraft.__init__cC|jS)zQ The entity that belongs to this dialog (user, chat or channel). )rrr#r#r$r!*sz Draft.entitycCsJ|js"z|jjt|jdd|_W|jSty!Y|jSw|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_entitycsR|js&|IdHr&z|j|jIdH|_W|jSty%Y|jSw|jS)zJ Returns `entity` but will make an API call if necessary. N)r!get_input_entityr get_entityrr ValueErrorr'r#r#r$r/?s zDraft.get_entitycs|jS)zP Returns `input_entity` but will make an API call if necessary. )r-r'r#r#r$r.LszDraft.get_input_entitycCr&)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$textTz Draft.textcCr&)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\r3zDraft.raw_textcCs|j S)zM Convenience bool to determine if the draft is empty or not. r1r'r#r#r$is_emptydszDraft.is_emptyNrr#c s|dur|j}|dkr|j}|dur|j}|j||IdH\}}|t|j|| |dur0dnt||dIdH}|rT||_||_ ||_||_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)rrrr_parse_message_textrrrrrdatetimenowtimezoneutcr)rr2r parse_moderr4rresultr#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_messagerr2rr)rclearr=r#r#r$sends  z Draft.sendcs|jddIdHS)zD Deletes this draft, and returns `True` on success. r)r2N)r?r'r#r#r$deletesz Draft.deletec CsNz|j}Wnty}z|}WYd}~nd}~wwd|j||j|j|jdS)Nr )_r2r!rrr)r!rr2rrr)rr!er#r#r$to_dicts z Draft.to_dictcCst|S)Nr pretty_formatrGr'r#r#r$__str__sz Draft.__str__cCstj|ddS)Nr)indentrHr'r#r#r$ stringifyszDraft.stringify)Nrr#N)Tr#)__name__ __module__ __qualname____doc__r%propertyr!r-r/r.r2r4r5r?rCrDrGrJrLr#r#r#r$r s.        0  r )r9rrrfunctions.messagesrrerrorsr extensionsr utilsr r r r r#r#r#r$s