U hTd@sddlZddlZddlZddlZddlZddlmZmZmZm Z ddlm Z m Z ddlm Z ddl mZmZmZddZejrd d lmZd ejd d dZGdddZdS)N)errorshelpersutilshints) MultiErrorRPCError) retry_range) TLRequesttypes functionscCstdS)Nz(You can only invoke requests, not types!) TypeErrorrr9/tmp/pip-unpacked-wheel-c81u5j2r/telethon/client/users.py r)TelegramClientF)earlytdcCs d|r dnd|||d|jjfS)Nz(Sleeping%s for %ds (%s) on %s flood waitz early)seconds) __class____name__)delayrequestrrrrr _fmt_floods  rc@seZdZd+ddddZd,ddddZd-ded d d d Zedej e d ddZ ded ddZ ded ddZ ddejdejdfdddZddddddZddddd Zd.ddee d"d#d$Zddd%d&Zddd'd(Zddd)d*ZdS)/ UserMethodsFNrselfcs|j|j||dIdHS)Nordered)_callZ_sender)rrr!flood_sleep_thresholdrrr__call__szUserMethods.__call__csH|jdk r |jtkr td|dkr.|j}t|r@t|n|g}t|rXt|n|}t|D]\}}t |t s|t | |tIdH|j |jkr&|j|j }t|t} | dkr|j|j dnT| |kr|jtjt| |ddt| IdH|j|j dntj|| d|jrdt|rHt|||<qdt|}qdd} d} t|_t|jD]} z|j ||d} t | trNg}g}| D]}z|IdH}WnBt!k r}z"|"||"dWYqW5d}~XYnX|j#$||"d|"|| d7} qt%d d |DrBt&|||n |WSn | IdH}|j#$||WSWqptj'tj(tj)tj*tj+tj,fk r}z0|} |jt-d |j.j|td IdHW5d}~XYqptjtj/tj0tj1fk r}z|} t|r|| }t |tj0s:t|j2|j|j <|j2dkrLd|_2|j2|jkr|jtjt|j2|t|j2IdHnW5d}~XYntj3tj4tj5fk r}zX|} |jtd |j6t |tj3tj4f}|r|7IdHr|8|j6IdHW5d}~XYnXqp|j9r6| dk r6| t:d;| dS)NzQThe asyncio event loop must not change after connection (see the FAQ for details)T)r)rcapturerr rcss|]}|dk VqdSNr.0xrrr Wsz$UserMethods._call..z)Telegram is having internal issues %s: %srzPhone migrated to %dz#Request was unsuccessful {} time(s))` which will use the cache), and then use `get_entity` with the result of the previous call. Similar limits apply to invite links, and you should use their ID instead. Using phone numbers (from people in your contact list), exact names, integer IDs or :tl:`Peer` rely on a `get_input_entity` first, which in turn needs the entity to be in cache, unless a :tl:`InputPeer` was passed. Unsupported types will raise ``TypeError``. If the entity can't be found, ``ValueError`` will be raised. Returns :tl:`User`, :tl:`Chat` or :tl:`Channel` corresponding to the input entity. A list will be returned if more than one was given. Example .. code-block:: python from telethon import utils me = await client.get_entity('me') print(utils.get_display_name(me)) chat = await client.get_input_entity('username') async for message in client.iter_messages(chat): ... # Note that you could have used the username directly, but it's # good to use get_input_entity if you will reuse it a lot. async for message in client.iter_messages('username'): ... # Note that for this to work the phone number must be in your contacts some_id = await client.get_peer_id('+34123456789') NcSsg|] }|jqSr)chat_idr(rrr Csz*UserMethods.get_entity..cSsi|]}tj|dd|qS)Fadd_mark)r get_peer_idr(rrr Is z*UserMethods.get_entity..Fr`css$|]}t|tjr|jr|VqdSr')r0r ZUserZis_self)r)urrrr+[s z)UserMethods.get_entity..r)rr-r0strr;get_input_entityrZ _EntityTypeUSERZCHATZCHANNELZ _entity_typer extendr rRrSmessagesZGetChatsRequestchatschannelsGetChannelsRequest itertoolschain_get_entity_from_stringr InputPeerSelfrbnextvalues) rr\Zsingleinputsr*listsrRrjrktmpcurrZ id_entityrJrrr get_entitysd8            zUserMethods.get_entityzhints.EntityLikeztypes.TypeInputPeer)rpeerrMcsz t|WStk r YnXz4t|ts8|jdkrT|jtj|dd WSWnt k rjYnX|dkr|t Sz|j |WStk rYnXt|trt||IdHSt|}t|t jr$|tjt j|jddgIdH}|rt|dt jst|dSnxt|t jr>t |jSt|t jrz6|tjt j |j!ddgIdH}t|j"dWSt#j$k rYnXtd%|t&|j'dS) a( Turns the given entity into its input entity version. Most requests use this kind of :tl:`InputPeer`, so this is the most suitable call to make for those cases. **Generally you should let the library do its job** and don't worry about getting the input entity first, but if you're going to use an entity often, consider making the call: Arguments entity (`str` | `int` | :tl:`Peer` | :tl:`InputPeer`): If a username or invite link is given, **the library will use the cache**. This means that it's possible to be using a username that *changed* or an old invite link (this only happens if an invite link for a small group chat is used after it was upgraded to a mega-group). If the username or ID from the invite link is not found in the cache, it will be fetched. The same rules apply to phone numbers (``'+34 123456789'``) from people in your contact list. If an exact name is given, it must be in the cache too. This is not reliable as different people can share the same name and which entity is returned is arbitrary, and should be used only for quick tests. If a positive integer ID is given, the entity will be searched in cached users, chats or channels, without making any call. If a negative integer ID is given, the entity will be searched exactly as either a chat (prefixed with ``-``) or as a channel (prefixed with ``-100``). If a :tl:`Peer` is given, it will be searched exactly in the cache as either a user, chat or channel. If the given object can be turned into an input entity directly, said operation will be done. Unsupported types will raise ``TypeError``. If the entity can't be found, ``ValueError`` will be raised. Returns :tl:`InputPeerUser`, :tl:`InputPeerChat` or :tl:`InputPeerChannel` or :tl:`InputPeerSelf` if the parameter is ``'me'`` or ``'self'``. If you need to get the ID of yourself, you should use `get_me` with ``input_peer=True``) instead. Example .. code-block:: python # If you're going to use "username" often in your code # (make a lot of calls), consider getting its input entity # once, and then using the "user" everywhere instead. user = await client.get_input_entity('username') # The same applies to IDs, chats or channels. chat = await client.get_input_entity(-123456789) VFr`rWrNr)rUzCould not find the input entity for {} ({}). Please read https://docs.telethon.dev/en/stable/concepts/entities.html to find out more details.)(rrVr r0intSUBCLASS_OF_IDrNrPrbrQAttributeErrorr rpr<rfr@reroZget_peerPeerUserr rRrSZ InputUserZuser_idZ UserEmptyZPeerChatZ InputPeerChatr^Z PeerChannelrkrlZ InputChannelZ channel_idrjrZChannelInvalidErrorrAtyper)rrxrRrkrrrrfbsTA     zUserMethods.get_input_entity)rrxcs"t||IdH\}}||Sr')rZ resolve_idrb)rrxrCclsrrr _get_peerszUserMethods._get_peerT)rrxrarMcst|trtj||dSz|jdkr4||IdH}Wn$tk rZ||IdH}YnXt|tjrz|j ddIdH}tj||dS)a Gets the ID for the given entity. This method needs to be ``async`` because `peer` supports usernames, invite-links, phone numbers (from people in your contact list), etc. If ``add_mark is False``, then a positive ID will be returned instead. By default, bot-API style IDs (signed) are returned. Example .. code-block:: python print(await client.get_peer_id('me')) r`)ry9NTrZ) r0r{rrbr|rfr}r rprX)rrxrarrrrbs   zUserMethods.get_peer_idc st|}|rhz6|tjdIdHjD]}|j|kr(|WSq(Wntjk rbt dYnXnf| dkr| IdHSt |\}}|r|tj |IdH}t|tjrt dnt|tjr|jSn|rz|tj|IdH}Wn6tjk r*}zt d||W5d}~XYnXzXtj|jddt|jtjrhtfd d |jDWStfd d |jDWSWntk rYnXz||j|IdHWSt k rYnXt d |dS) a Gets a full entity from the given string, which may be a phone or a username, and processes all the found entities on the session. The string may also be a user link, or a channel/chat invite link. This method has the side effect of adding the found users to the session database, so it can be queried later without API calls, if this option is enabled on the session. Returns the found entity, or raises TypeError if not found. rNzOCannot get entity by phone number as a bot (try using integer IDs, not strings)rzz^Cannot get entity from a channel (or group) that you are not part of. Join the group and retryzNo user has "{}" as usernameFr`c3s|]}|jkr|VqdSr'rTr(pidrrr+:s z6UserMethods._get_entity_from_string..c3s|]}|jkr|VqdSr'rr(rrrr+<s z,Cannot find any entity corresponding to "{}") rZ parse_phoner ZcontactsZGetContactsRequestrRphonerZBotMethodInvalidErrorr@lowerrXZparse_usernameriZCheckChatInviteRequestr0r Z ChatInviteZChatInviteAlreadyZchatZResolveUsernameRequestZUsernameNotOccupiedErrorrArbrxr~rqrj StopIterationrwr<rf) rstringruserusernameZ is_join_chatZinviterJrKrrrro sd         z#UserMethods._get_entity_from_stringcsjz>|jdkr&||jIdH|_|WS|jdkrs