o XzjY9@s dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z GdddejZeeZd)dd Zd d Zd d ZddZddddZddZd)ddZddZddZddZddZdd Zd!d"ZGd#d$d$e Z!Gd%d&d&ej"Z#d'd(Z$dS)*z6Various helpers not related to the Telegram API itselfN)Path)sha1c@seZdZdZdZdZdS) _EntityTyperN)__name__ __module__ __qualname__USERCHATCHANNELr r K/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/helpers.pyrsrTcCstjtd|ddS)zEGenerates a random long integer (8 bytes), which is optionally signedlittle)signed byteorder)int from_bytesosurandomrr r rgenerate_random_longsrcCs&tj|}|rtj|dddSdS)z(Ensures that the parent directory existsT)exist_okN)rpathdirnamemakedirs) file_pathparentr r rensure_parent_dir_exists s rcCsddd|DS)Nc ssR|]$}dt|krdkr#nndddtd|dDn|VqdS)iir cs|]}t|VqdSN)chr).0yr r r +z*add_surrogate...z.)r)textr r r add_surrogate's r1cCs|dddS)Nzutf-16 surrogatepass)r,decoder/r r r del_surrogate0sr4)lengthcCsh|durt|}d|kot|kno3d||dko#dkno3d||ko1dkSS)zS `True` if ``index`` is within a surrogate (before and after it, not at!). Nruuu)len)r0indexr5r r rwithin_surrogate4sr8cCs|s|St|}|}|t|}|}t|}ttt|D]M}||}|jdkr3||=q$|j|j|krU|j|krH|j|8_n|j|j||_d|_n||=q$|j|j|krbq$|j|krk||=q$||j|_q$|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)stripr6lstriprstripreversedranger5offset)r0entitieslen_ori left_offset len_finalier r r strip_textBs2      rEccsH|r|dus|dks|d7}d}||kr"|d7}|V||ksdSdS)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 )retries force_retryattemptr r r retry_ranges rIcst|r |IdHS|Sr")inspect isawaitable)valuer r r _maybe_awaits  rMc s|D]V\}}|s q|z|IdHWqtjy!Yqty)YqtyI}z|jdkr?|d|t||WYd}~qd}~wt y[|d|t||YqwdS)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)) itemscancelasyncioCancelledError RuntimeErrorAssertionErrorargs exceptiontype Exception)logtasksnametaskrDr r r_cancels.      r\cCs8t|dr |j}n|jj}|rtd||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"))hasattrr]_client is_runningrRrun_until_complete __aenter__)selfr]r r r _sync_enters rdcGs*t|dr |j}n|jj}||j|S)Nr])r^r]r_ra __aexit__)rcrTr]r r r _sync_exits rfcCsz|jdvr td|Wntytd|w|jj}d|vr(tjSd|vr/tjSd|vr6tj Sd|vr=tjStd|)N)iVl9lF?Li@iwy-l]^ iaFl&4)z {} does not have any entity typez2{} is not a TLObject, cannot determine entity typeUserChatChannelSelf) SUBCLASS_OF_ID TypeErrorformatAttributeError __class__rrr r r )entityrZr r r _entity_types"   rqcCs|jdddd}|jdddd}t||}t||}t||}||dd}|dd||dd }||fS) z7Generates the key data corresponding to the given noncerTr N )to_bytesrdigest) server_nonce new_noncehash1hash2hash3keyivr r rgenerate_key_data_from_noncesrcs0eZdZdZfddZddZddZZS) 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) cstj|i|d|_dS)Nr)super__init__totalrcrTkwargsror rr8s zTotalList.__init__cCdddd|D|jS)N[{}, total={}], csr!r")strr-r r rr&>r'z$TotalList.__str__..rmr)rrcr r r__str__<zTotalList.__str__cCr)Nrrcsr!r")reprr-r r rr&Br'z%TotalList.__repr__..rrr r r__repr__@rzTotalList.__repr__)rrr __doc__rrr __classcell__r r rrr!s  rc@seZdZdZddddZddZdd Zed d Zed d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd 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|tr t|}||_d|_||_d|_d|_dSr") isinstancerrabsolute_file_name_size_stream _close_stream)rcfilerr r rrOs   z_FileStream.__init__cst|jtr#tj|j|_tj|j|_t |jd|_ d|_ |St|jt r;t |j|_t|j|_ d|_ |Stt|jddsHtdt|jdd|_|j|_ d|_ |jdurtt|jddrpt|jIdH}nd}|rt|jIdH}t|jdtjIdHt|jIdH|_t|j|tjIdH|Std t|jIdH}t ||_t||_ d|_ |S) NrbTreadz,file description should have a `read` methodrZFseekablerzRCould not determine file size beforehand so the entire file will be read in-memory)rrrrrbasenamergetsizeropenrrbytesr6ioBytesIOcallablegetattrrlrMrtellseekSEEK_ENDSEEK_SET_logwarningr)rcrposdatar r rrbYsF       z_FileStream.__aenter__cs.|jr|jrt|jIdHdSdSdSr")rrrMclose)rcexc_typeexc_valexc_tbr r rres z_FileStream.__aexit__cC|jSr")rrr r rrz_FileStream.file_sizecCrr")rrr r rrZrz_FileStream.namecO|jj|i|Sr")rrrr r rrz_FileStream.readcOrr")rreadintorr r rrrz_FileStream.readintocOrr")rwriterr r rrrz_FileStream.writecOrr")rfilenorr r rrrz_FileStream.filenocOrr")rflushrr r rrrz_FileStream.flushcOrr")risattyrr r rrrz_FileStream.isattycOrr")rreadablerr r rrrz_FileStream.readablecOrr")rreadlinerr r rrrz_FileStream.readlinecOrr")r readlinesrr r rrrz_FileStream.readlinescOrr")rrrr r rrrz_FileStream.seekcOrr")rrrr r rrrz_FileStream.seekablecOrr")rrrr r rrrz_FileStream.tellcOrr")rtruncaterr r rrrz_FileStream.truncatecOrr")rwritablerr r rrrz_FileStream.writablecOrr")r writelinesrr r rrrz_FileStream.writelinescOsdSr"r rr r rrsz_FileStream.close)rrr rrrbrepropertyrrZrrrrrrrrrrrrrrrrr r r rrEs2  ,   rc CsVztWSty*ztWYSty)t}t||YYSwwr")rPget_running_looprRget_event_loopnew_event_loopset_event_loop)r]r r rrs     r)T)%rrPrenumrr*rJlogging functoolssyspathlibrhashlibrEnumr getLoggerrrrrr1r4r8rErIrMr\rdrfrqrlistrIOBaserrr r r rs<      J% ($ f