a f’i;9ã@s dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z Gdd„dejƒZe e¡Zd)dd „Zd d „Zd d „Zdd„Zddœdd„Zdd„Zd*dd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„ZGd#d$„d$e ƒZ!Gd%d&„d&ej"ƒZ#d'd(„Z$dS)+z6Various helpers not related to the Telegram API itselféN)ÚPath)Úsha1c@seZdZdZdZdZdS)Ú _EntityTyperééN)Ú__name__Ú __module__Ú __qualname__ÚUSERÚCHATÚCHANNEL©r r úE/home/ec2-user/.local/lib/python3.9/site-packages/telethon/helpers.pyrsrTcCstjt d¡|ddS)zEGenerates a random long integer (8 bytes), which is optionally signedéÚlittle)ÚsignedÚ byteorder)ÚintÚ from_bytesÚosÚurandom©rr r rÚgenerate_random_longsrcCs"tj |¡}|rtj|dddS)z(Ensures that the parent directory existsT)Úexist_okN)rÚpathÚdirnameÚmakedirs)Ú file_pathÚparentr r rÚensure_parent_dir_exists s rcCsd dd„|Dƒ¡S)NÚc ssP|]H}dt|ƒkrdkrDnn"d dd„t d| d¡¡Dƒ¡n|VqdS)iiÿÿr css|]}t|ƒVqdS©N)Úchr)Ú.0Úyr r rÚ +óz*add_surrogate...z.)r(©Útextr r rÚ add_surrogate's ür0cCs| dd¡ d¡S)Nzutf-16Ú surrogatepass)r+Údecoder.r r rÚ del_surrogate0sr3)ÚlengthcCsh|durt|ƒ}d|ko&t|ƒknofd||dkoFdknofd||kobdkSS)zS `True` if ``index`` is within a surrogate (before and after it, not at!). Nruí €uí¯¿uí¿¿)Úlen)r/Úindexr4r r rÚwithin_surrogate4sÿýr7cCsè|s | ¡St|ƒ}| ¡}|t|ƒ}| ¡}t|ƒ}ttt|ƒƒƒD]š}||}|jdkrf||=qH|j|j|krª|j|kr|j|8_q²|j|j||_d|_n||=qH|j|j|krÄqH|j|krÖ||=qH||j|_qH|S)zý Strips whitespace from the given surrogated text modifying the provided entities, also removing any empty (0-length) entities. This assumes that the length of entities is greater or equal to 0, and that no entity is out of bounds. r)Ústripr5ÚlstripÚrstripÚreversedÚranger4Úoffset)r/ÚentitiesZlen_oriZ left_offsetZ len_finalÚiÚer r rÚ strip_textBs2      rAccs<|r|dus|dks|d7}d}||kr8|d7}|Vq dS)z¿ Generates an integer sequence starting from 1. If `retries` is not a zero or a positive integer value, the sequence will be infinite, otherwise it will end at `retries + 1`. Nrrr )ÚretriesZ force_retryÚattemptr r rÚ retry_rangeŒs rDcÃst |¡r|IdHS|SdSr!)ÚinspectÚ isawaitable)Úvaluer r rÚ _maybe_awaitŸs  rHc Ës¼| ¡D]®\}}|sq| ¡z|IdHWqtjy@YqtyPYqty}z*|jdkr|| d|t|ƒ|¡WYd}~qd}~0t y´| d|t|ƒ|¡Yq0qdS)zL Helper to cancel one or more tasks gracefully, logging exceptions. N)z"yield from wasn't used with futurez4Unhandled exception from %s after cancelling %s (%s)) ÚitemsÚcancelÚasyncioÚCancelledErrorÚ RuntimeErrorÚAssertionErrorÚargsÚ exceptionÚtypeÚ Exception)ÚlogÚtasksÚnameÚtaskr@r r rÚ_cancel¦s&   ÿ  ÿrWcCs8t|dƒr|j}n|jj}| ¡r*tdƒ‚| | ¡¡S)za Helps to cut boilerplate on async context managers that offer synchronous variants. Úloopz[You must use "async with" if the event loop is running (i.e. you are inside an "async def"))ÚhasattrrXÚ_clientÚ is_runningrMÚrun_until_completeÚ __aenter__)ÚselfrXr r rÚ _sync_enterËs ÿr_cGs*t|dƒr|j}n|jj}| |j|Ž¡S)NrX)rYrXrZr\Ú __aexit__)r^rOrXr r rÚ _sync_exitÞs racCsz|jdvrtd |¡ƒ‚Wn ty<td |¡ƒ‚Yn0|jj}d|vrTtjSd|vrbtjSd|vrptj Sd|vr~tjStd |¡ƒ‚dS)N)i‡VÔl¶9lF?ÓLiýò@iwy¡-l”]^ i¹aFl—&4)z {} does not have any entity typez2{} is not a TLObject, cannot determine entity typeZUserZChatZChannelZSelf) ZSUBCLASS_OF_IDÚ TypeErrorÚformatÚAttributeErrorÚ __class__rrr r r )ÚentityrUr r rÚ _entity_typeçs  rgcCs„|jdddd}|jdddd}t||ƒ ¡}t||ƒ ¡}t||ƒ ¡}||dd…}|dd…||dd …}||fS) z7Generates the key data corresponding to the given nonceérTré Né éé)Úto_bytesrÚdigest)Z server_nonceZ new_nonceZhash1Zhash2Zhash3ÚkeyZivr r rÚgenerate_key_data_from_noncesrpcs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)Ú TotalListaµ A list with an extra `total` property, which may not match its `len` since the total represents the total amount of items *available* somewhere else, not the items *in this list*. Examples: .. code-block:: python # Telethon returns these lists in some cases (for example, # only when a chunk is returned, but the "total" count # is available). result = await client.get_messages(chat, limit=10) print(result.total) # large number print(len(result)) # 10 print(result[0]) # latest message for x in result: # show the 10 messages print(x.text) cstƒj|i|¤Žd|_dS)Nr)ÚsuperÚ__init__Útotal©r^rOÚkwargs©rer rrs8szTotalList.__init__cCsd d dd„|Dƒ¡|j¡S)Nú[{}, total={}]ú, css|]}t|ƒVqdSr!)Ústrr,r r rr%>r&z$TotalList.__str__..©rcr(rt©r^r r rÚ__str__<sÿzTotalList.__str__cCsd d dd„|Dƒ¡|j¡S)Nrxrycss|]}t|ƒVqdSr!)Úreprr,r r rr%Br&z%TotalList.__repr__..r{r|r r rÚ__repr__@sÿzTotalList.__repr__)rrr Ú__doc__rsr}rÚ __classcell__r r rwrrq!s rqc@sÆeZdZdZddœdd„Zdd„Zdd „Zed d „ƒZed d „ƒZ dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„ZdS).Ú _FileStreama: Proxy around things that represent a file and need to be used as streams which may or not need to be closed. This will handle `pathlib.Path`, `str` paths, in-memory `bytes`, and anything IO-like (including `aiofiles`). It also provides access to the name and file size (also necessary). N)Ú file_sizecCs8t|tƒrt| ¡ƒ}||_d|_||_d|_d|_dSr!) Ú isinstancerrzÚabsoluteÚ_fileÚ_nameÚ_sizeÚ_streamÚ _close_stream)r^Úfilerƒr r rrsOs  z_FileStream.__init__cÃsˆt|jtƒrDtj |j¡|_tj |j¡|_t |jdƒ|_ d|_ |St|jt ƒrtt |jƒ|_t |j¡|_ d|_ |Stt|jddƒƒsŽtdƒ‚t|jddƒ|_|j|_ d|_ |jdur„tt|jddƒƒràt|j ¡ƒIdH}nd}|rJt|j ¡ƒIdH}t|j dtj¡ƒIdHt|j ¡ƒIdH|_t|j |tj¡ƒIdHn:t d ¡t|j ¡ƒIdH}t |ƒ|_t |¡|_ d|_ |S) NÚrbTÚreadz,file description should have a `read` methodrUFÚseekablerzRCould not determine file size beforehand so the entire file will be read in-memory)r„r†rzrrÚbasenamer‡ÚgetsizerˆÚopenr‰rŠÚbytesr5ÚioÚBytesIOÚcallableÚgetattrrbrHrŽÚtellÚseekÚSEEK_ENDÚSEEK_SETÚ_logÚwarningr)r^rŽÚposÚdatar r rr]YsB    ÿ  z_FileStream.__aenter__cÃs$|jr |jr t|j ¡ƒIdHdSr!)rŠr‰rHÚclose)r^Úexc_typeÚexc_valÚexc_tbr r rr`…s z_FileStream.__aexit__cCs|jSr!)rˆr|r r rrƒ‰sz_FileStream.file_sizecCs|jSr!)r‡r|r r rrUsz_FileStream.namecOs|jj|i|¤ŽSr!)r‰rrur r rr’r&z_FileStream.readcOs|jj|i|¤ŽSr!)r‰Úreadintorur r rr£“r&z_FileStream.readintocOs|jj|i|¤ŽSr!)r‰Úwriterur r rr¤”r&z_FileStream.writecOs|jj|i|¤ŽSr!)r‰Úfilenorur r rr¥•r&z_FileStream.filenocOs|jj|i|¤ŽSr!)r‰Úflushrur r rr¦–r&z_FileStream.flushcOs|jj|i|¤ŽSr!)r‰Úisattyrur r rr§—r&z_FileStream.isattycOs|jj|i|¤ŽSr!)r‰Úreadablerur r rr¨˜r&z_FileStream.readablecOs|jj|i|¤ŽSr!)r‰Úreadlinerur r rr©™r&z_FileStream.readlinecOs|jj|i|¤ŽSr!)r‰Ú readlinesrur r rrªšr&z_FileStream.readlinescOs|jj|i|¤ŽSr!)r‰r˜rur r rr˜›r&z_FileStream.seekcOs|jj|i|¤ŽSr!)r‰rŽrur r rrŽœr&z_FileStream.seekablecOs|jj|i|¤ŽSr!)r‰r—rur r rr—r&z_FileStream.tellcOs|jj|i|¤ŽSr!)r‰Útruncaterur r rr«žr&z_FileStream.truncatecOs|jj|i|¤ŽSr!)r‰Úwritablerur r rr¬Ÿr&z_FileStream.writablecOs|jj|i|¤ŽSr!)r‰Ú writelinesrur r rr­ r&z_FileStream.writelinescOsdSr!r rur r rrŸ¦sz_FileStream.close)rrr r€rsr]r`ÚpropertyrƒrUrr£r¤r¥r¦r§r¨r©rªr˜rŽr—r«r¬r­rŸr r r rr‚Es0  ,  r‚cCsBtjdkr6z t ¡WSty2t ¡ ¡YS0nt ¡SdS)N)éé)ÚsysÚ version_inforKÚget_running_looprMÚget_event_loop_policyÚget_event_loopr r r rr³«s    r³)T)T)%r€rKr“Úenumrr)rEÚloggingÚ functoolsr±ÚpathlibrÚhashlibrÚEnumrÚ getLoggerrr›rrr0r3r7rArDrHrWr_rargrpÚlistrqÚIOBaser‚r³r r r rÚs:     J % ($f