U h@s:ddlZddlmZmZddlmZGdddejZdS)N)errorsutils)typesc@seZdZdZdddddddZeddZdd Zed d Zd d Z eddZ eddZ eddZ eddZ ddZdS) ChatGetterz Helper base class that introduces the `chat`, `input_chat` and `chat_id` properties and `get_chat` and `get_input_chat` methods. N) input_chatchat broadcastcCs"||_||_||_||_d|_dS)N) _chat_peer _input_chat_chat _broadcast_client)selfZ chat_peerrrr rA/tmp/pip-unpacked-wheel-c81u5j2r/telethon/tl/custom/chatgetter.py__init__ s zChatGetter.__init__cCs|jS)a| Returns the :tl:`User`, :tl:`Chat` or :tl:`Channel` where this object belongs to. It may be `None` if Telegram didn't send the chat. If you only need the ID, use `chat_id` instead. If you need to call a method which needs this chat, use `input_chat` instead. If you're using `telethon.events`, use `get_chat()` instead. )r rrrrrs zChatGetter.chatcsj|jdkst|jddrd|IdHrdz|j|jIdH|_Wn"tk rb|IdHYnX|jS)a Returns `chat`, but will make an API call to find the chat unless it's already cached. If you only need the ID, use `chat_id` instead. If you need to call a method which needs this chat, use `get_input_chat()` instead. Nmin)r getattrget_input_chatrZ get_entityr ValueError _refetch_chatrrrrget_chat#s  zChatGetter.get_chatcCsV|jdkrP|jrP|jrPz$|jjtj|jdd|_Wntk rNYnX|jS)a This :tl:`InputPeer` is the input version of the chat where the message was sent. Similarly to `input_sender `, this doesn't have things like username or similar, but still useful in some cases. Note that this might not be available if the library doesn't have enough information available. NF)Zadd_mark) r r rZ_mb_entity_cachegetr get_peer_idZ_as_input_peerAttributeErrorrrrrr7s zChatGetter.input_chatcsx|jdkrr|jrr|jrrzD|j}|jd2z*3dHW}|j|kr*|j|_|j|_qXq*6Wnt j k rpYnX|jS)z| Returns `input_chat`, but will make an API call to find the input chat unless it's already cached. Nd) rchat_idrZ iter_dialogsidentityr Z input_entityr rZRPCError)rtargetdrrrrLs  zChatGetter.get_input_chatcCs|jrt|jSdS)a Returns the marked chat integer ID. Note that this value **will be different** from ``peer_id`` for incoming private messages, since the chat *to* which the messages go is to your own person, but the *chat* itself is with the one who sent the message. TL;DR; this gets the ID that you expect. If there is a chat in the object, `chat_id` will *always* be set, which is why you should use it instead of `chat.id `. N)r rrrrrrr_s zChatGetter.chat_idcCs|jrt|jtjSdS)z `True` if the message was sent as a private message. Returns `None` if there isn't enough information (e.g. on `events.MessageDeleted `). N)r isinstancerZPeerUserrrrr is_privatenszChatGetter.is_privatecCsV|jdkr$t|jdr$t|jj|_t|jtjrH|jdkr@dS|j St|jtj S)z True if the message was sent on a group or megagroup. Returns `None` if there isn't enough information (e.g. on `events.MessageDeleted `). Nr ) r hasattrrboolr r#r r PeerChannelZPeerChatrrrris_groupxs  zChatGetter.is_groupcCst|jtjS)z9`True` if the message was sent on a megagroup or channel.)r#r rr'rrrr is_channelszChatGetter.is_channelcsdS)zB Re-fetches chat information through other means. NrrrrrrszChatGetter._refetch_chat)N)__name__ __module__ __qualname____doc__rpropertyrrrrrr$r(r)rrrrrrs"      r)abcrrtlrABCrrrrrs