3 uQh;9@s dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z GdddejZejeZd)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 4/tmp/pip-build-2nz6shyl/telethon/telethon/helpers.pyrsrTcCstjtjd|ddS)zEGenerates a random long integer (8 bytes), which is optionally signedlittle)signed byteorder)int from_bytesosurandom)rr r rgenerate_random_longsrcCs"tjj|}|rtj|dddS)z(Ensures that the parent directory existsT)exist_okN)rpathdirnamemakedirs) file_pathparentr r rensure_parent_dir_exists s rcCsdjdd|DS)NcssP|]H}dt|kodknrDdjddtjd|jdDn|VqdS)iircss|]}t|VqdS)N)chr).0yr r r +sz*add_surrogate...z.)r%)textr r r add_surrogate'sr+cCs|jddjdS)Nzutf-16 surrogatepass)r(decode)r*r r r del_surrogate0sr.)lengthcCsh|dkrt|}d|ko&t|knofd||dkoFdknofd||kobdkSS)zS `True` if ``index`` is within a surrogate (before and after it, not at!). Nruuu)len)r*indexr/r r rwithin_surrogate4s  r2cCs|s |jSt|}|j}|t|}|j}t|}xttt|D]}||}|jdkrh||=qJ|j|j|kr|j|kr|j|8_q|j|j||_d|_n||=qJ|j|j|krqJ|j|kr||=qJ||j|_qJW|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)stripr0lstriprstripreversedranger/offset)r*entitiesZlen_oriZ left_offsetZ len_finalier r r strip_textBs2      r<ccsB|r|dkp|dk r|d7}d}x||kr<|d7}|Vq$WdS)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_retryattemptr r r retry_ranges  r?cstj|r|IdHS|SdS)N)inspect isawaitable)valuer r r _maybe_awaits  rCcsx|jD]\}}|sq |jy|IdHWq tjk rDYq tk rVYq tk r}z$|jdkr|jd|t||WYdd}~Xq t k r|jd|t||Yq Xq WdS)zL Helper to cancel one or more tasks gracefully, logging exceptions. N"yield from wasn't used with futurez4Unhandled exception from %s after cancelling %s (%s))rD) itemscancelasyncioZCancelledError RuntimeErrorAssertionErrorargs exceptiontype Exception)logZtasksnameZtaskr;r r r_cancels"  rPcCs8t|dr|j}n|jj}|jr*td|j|jS)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"))hasattrrQ_clientZ is_runningrHrun_until_complete __aenter__)selfrQr r r _sync_enters rWcGs*t|dr|j}n|jj}|j|j|S)NrQ)rRrQrSrT __aexit__)rVrJrQr r r _sync_exits rYcCsy|jdkrtd j|Wn"tk r>td j|YnX|jj}d |krVtjSd |krdtjSd |krrtj Sd|krtjStd j|dS)NV9F?L@wy-]^ aF&4)z {} does not have any entity typez2{} is not a TLObject, cannot determine entity typeZUserZChatZChannelZSelf)rZr[r\r]r^r_r`ra) ZSUBCLASS_OF_ID TypeErrorformatAttributeError __class__rrr r r )entityrOr r r _entity_types.rgcCs|jdddd}|jdddd}t||j}t||j}t||j}||dd}|dd||dd }||fS) z7Generates the key data corresponding to the given noncerT)r N )to_bytesrdigest)Z server_nonceZ new_nonceZhash1Zhash2Zhash3keyZivr r rgenerate_key_data_from_noncesrpcs0eZdZdZfddZddZddZZS) 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||d|_dS)Nr)super__init__total)rVrJkwargs)rer rrs8szTotalList.__init__cCsdjdjdd|D|jS)Nz[{}, total={}]z, css|]}t|VqdS)N)str)r!r)r r rr#>sz$TotalList.__str__..)rcr%rt)rVr r r__str__<szTotalList.__str__cCsdjdjdd|D|jS)Nz[{}, total={}]z, css|]}t|VqdS)N)repr)r!r)r r rr#Bsz%TotalList.__repr__..)rcr%rt)rVr r r__repr__@szTotalList.__repr__)rrr __doc__rsrwry __classcell__r r )rerrq!s rqc@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|trt|j}||_d|_||_d|_d|_dS)N) isinstancerrvZabsolute_file_name_size_stream _close_stream)rVfiler}r r rrsOs  z_FileStream.__init__cst|jtrDtjj|j|_tjj|j|_t |jd|_ d|_ |St|jt rtt |j|_tj|j|_ d|_ |Stt|jddstdt|jdd|_|j|_ d|_ |jdkrtt|jddrt|jjIdH}nd}|rJt|jjIdH}t|jjdtjIdHt|jjIdH|_t|jj|tjIdHn:tjd t|jjIdH}t ||_tj||_ d|_ |S) NrbTreadz,file description should have a `read` methodrOFseekablerzRCould not determine file size beforehand so the entire file will be read in-memory)r~rrvrrbasenamergetsizeropenrrbytesr0ioBytesIOcallablegetattrrbrCrtellseekSEEK_ENDSEEK_SET_logwarningr)rVrposdatar r rrUYs@      z_FileStream.__aenter__cs$|jr |jr t|jjIdHdS)N)rrrCclose)rVexc_typeexc_valexc_tbr r rrXs z_FileStream.__aexit__cCs|jS)N)r)rVr r rr}sz_FileStream.file_sizecCs|jS)N)r)rVr r rrOsz_FileStream.namecOs|jj||S)N)rr)rVrJrur r rrsz_FileStream.readcOs|jj||S)N)rreadinto)rVrJrur r rrsz_FileStream.readintocOs|jj||S)N)rwrite)rVrJrur r rrsz_FileStream.writecOs|jj||S)N)rfileno)rVrJrur r rrsz_FileStream.filenocOs|jj||S)N)rflush)rVrJrur r rrsz_FileStream.flushcOs|jj||S)N)risatty)rVrJrur r rrsz_FileStream.isattycOs|jj||S)N)rreadable)rVrJrur r rrsz_FileStream.readablecOs|jj||S)N)rreadline)rVrJrur r rrsz_FileStream.readlinecOs|jj||S)N)r readlines)rVrJrur r rrsz_FileStream.readlinescOs|jj||S)N)rr)rVrJrur r rrsz_FileStream.seekcOs|jj||S)N)rr)rVrJrur r rrsz_FileStream.seekablecOs|jj||S)N)rr)rVrJrur r rrsz_FileStream.tellcOs|jj||S)N)rtruncate)rVrJrur r rrsz_FileStream.truncatecOs|jj||S)N)rwritable)rVrJrur r rrsz_FileStream.writablecOs|jj||S)N)r writelines)rVrJrur r rrsz_FileStream.writelinescOsdS)Nr )rVrJrur r rrsz_FileStream.close)rrr rzrsrUrXpropertyr}rOrrrrrrrrrrrrrrrrr r r rr|Es,  ,  r|c Cs>tjdkr2ytjStk r.tjjSXntjSdS)N)rr)sys version_inforGget_running_looprHZget_event_loop_policyZget_event_loopr r r rrs  r)T)T)%rzrGrenumrr&r@logging functoolsrZpathlibrhashlibrEnumr getLoggerrrrrr+r.r2r<r?rCrPrWrYrgrplistrqIOBaser|rr r r rs:     J % ($f