U h;9@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 4/tmp/pip-unpacked-wheel-c81u5j2r/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|dddS)z(Ensures that the parent directory existsT)exist_okN)rpathdirnamemakedirs) file_pathparentr r rensure_parent_dir_exists s rcCsddd|DS)Nc ssP|]H}dt|krdkrDnn"dddtd|dDn|VqdS)iir css|]}t|VqdSN)chr).0yr r r +sz*add_surrogate...z.)r'textr r r add_surrogate's r/cCs|dddS)Nzutf-16 surrogatepass)r*decoder-r r r del_surrogate0sr2)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.indexr3r r rwithin_surrogate4sr6cCs|s |St|}|}|t|}|}t|}ttt|D]}||}|jdkrf||=qH|j|j|kr|j|kr|j|8_q|j|j||_d|_n||=qH|j|j|krqH|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)stripr4lstriprstripreversedranger3offset)r.entitiesZlen_oriZ left_offsetZ len_finalier r r strip_textBs2      r@ccs<|r|dks|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_retryattemptr r r retry_ranges rCcst|r|IdHS|SdSr!)inspect isawaitable)valuer r r _maybe_awaits  rGc s|D]\}}|sq|z|IdHWqtjk rBYqtk rTYqtk r}z"|jdkr|d|t||W5d}~XYqt k r|d|t||YqXqdS)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)) itemscancelasyncioZCancelledError RuntimeErrorAssertionErrorargs exceptiontype Exception)logZtasksnameZtaskr?r r r_cancels.  rScCs8t|dr|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"))hasattrrT_clientZ is_runningrKrun_until_complete __aenter__)selfrTr r r _sync_enters rZcGs*t|dr|j}n|jj}||j|S)NrT)rUrTrVrW __aexit__)rYrMrTr r r _sync_exits r\cCsz|jdkrtd|Wn"tk r>td|YnX|jj}d|krVtjSd|krdtjSd|krrtj Sd|krtjStd|dS)N)iVl9lF?Li@iwy-l]^ iaFl&4)z {} does not have any entity typez2{} is not a TLObject, cannot determine entity typeZUserZChatZChannelZSelf) ZSUBCLASS_OF_ID TypeErrorformatAttributeError __class__rrr r r )entityrRr r r _entity_types rbcCs|jdddd}|jdddd}t||}t||}t||}||dd}|dd||dd }||fS) z7Generates the key data corresponding to the given noncerTr N )to_bytesrdigest)Z server_nonceZ new_nonceZhash1Zhash2Zhash3keyZivr r rgenerate_key_data_from_noncesrkcs0eZdZdZfddZddZddZZS) 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__totalrYrMkwargsr`r rrn8szTotalList.__init__cCsdddd|D|jS)N[{}, total={}], css|]}t|VqdSr!)strr+r r rr%>sz$TotalList.__str__..r^r'rorYr r r__str__<szTotalList.__str__cCsdddd|D|jS)Nrsrtcss|]}t|VqdSr!)reprr+r r rr%Bsz%TotalList.__repr__..rvrwr r r__repr__@szTotalList.__repr__)rrr __doc__rnrxrz __classcell__r r rrrrl!s rlc@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|}||_d|_||_d|_d|_dSr!) isinstancerruabsolute_file_name_size_stream _close_stream)rYfiler~r r rrnOs  z_FileStream.__init__cst|jtrDtj|j|_tj|j|_t |jd|_ d|_ |St|jt rtt |j|_t|j|_ d|_ |Stt|jddstdt|jdd|_|j|_ d|_ |jdkrtt|jddrt|jIdH}nd}|rJt|jIdH}t|jdtjIdHt|jIdH|_t|j|tjIdHn:td t|jIdH}t ||_t||_ d|_ |S) NrbTreadz,file description should have a `read` methodrRFseekablerzRCould not determine file size beforehand so the entire file will be read in-memory)rrrurrbasenamergetsizeropenrrbytesr4ioBytesIOcallablegetattrr]rGrtellseekSEEK_ENDSEEK_SET_logwarningr)rYrposdatar r rrXYsB      z_FileStream.__aenter__cs$|jr |jr t|jIdHdSr!)rrrGclose)rYexc_typeexc_valexc_tbr r rr[s z_FileStream.__aexit__cCs|jSr!)rrwr r rr~sz_FileStream.file_sizecCs|jSr!)rrwr r rrRsz_FileStream.namecOs|jj||Sr!)rrrpr r rrz_FileStream.readcOs|jj||Sr!)rreadintorpr r rrrz_FileStream.readintocOs|jj||Sr!)rwriterpr r rrrz_FileStream.writecOs|jj||Sr!)rfilenorpr r rrrz_FileStream.filenocOs|jj||Sr!)rflushrpr r rrrz_FileStream.flushcOs|jj||Sr!)risattyrpr r rrrz_FileStream.isattycOs|jj||Sr!)rreadablerpr r rrrz_FileStream.readablecOs|jj||Sr!)rreadlinerpr r rrrz_FileStream.readlinecOs|jj||Sr!)r readlinesrpr r rrrz_FileStream.readlinescOs|jj||Sr!)rrrpr r rrrz_FileStream.seekcOs|jj||Sr!)rrrpr r rrrz_FileStream.seekablecOs|jj||Sr!)rrrpr r rrrz_FileStream.tellcOs|jj||Sr!)rtruncaterpr r rrrz_FileStream.truncatecOs|jj||Sr!)rwritablerpr r rrrz_FileStream.writablecOs|jj||Sr!)r writelinesrpr r rrrz_FileStream.writelinescOsdSr!r rpr r rrsz_FileStream.close)rrr r{rnrXr[propertyr~rRrrrrrrrrrrrrrrrrr r r rr}Es0  ,  r}cCsDtjdkr8z tWStk r4tYSXntSdS)N))sys version_inforJget_running_looprKZget_event_loop_policyZget_event_loopr r r rrs   r)T)T)%r{rJrenumrr(rDlogging functoolsrpathlibrhashlibrEnumr getLoggerrrrrr/r2r6r@rCrGrSrZr\rbrklistrlIOBaser}rr r r rs:     J % ($f