U h}@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlmZddlmZddlmZmZddlmZmZmZdd lmZzddlZddlZddlZWnek rdZYnXed d ed d eddeddeddeddeddeddeddeddedded d!ed"d#ed$d%ed&d'ed(d)ed*d+e d,Z e d-Z!e d.e j"Z#ed/d0Z$e%e&Z'dd2d3Z(d4d5Z)d6d7Z*d8d9Z+dd;d<Z,d=d>Z-d?d@Z.dAdBZ/dCdDZ0dEdFZ1dGdHZ2dIdJZ3dKddKdKdKdKddLdMdNZ4dOdPZ5dQdRZ6e,fdSdTZ7dUdVZ8dWdXZ9dddKdKdKdKddYdZd[Z:d\d]Z;d^d_ZdddeZ?dfdgZ@dhdiZAdjdkZBdldmZCdndoZDdpdqZEdrdsZFdtduZGddvdwZHdxdyZIdzd{ZJd|d}ZKd~dZLddZMddZNddZOddZPddZQddZRddZSddZTdd1ddddZUGdddZVddZWddZXdS)z Utilities for working with the Telegram API itself (such as handy methods to convert between an entity like a User, Chat, etc. into its Input version) N) namedtuple)guess_extension) GeneratorType)markdownhtml) add_surrogate del_surrogate strip_text)typesz image/pngz.pngz image/jpegz.jpegz image/webpz.webpz image/gifz.gifz image/bmpz.bmpz image/x-tgaz.tgaz image/tiffz.tiffzimage/vnd.adobe.photoshopz.psdz video/mp4z.mp4zvideo/quicktimez.movz video/aviz.aviz audio/mpegz.mp3z audio/m4az.m4az audio/aacz.aacz audio/oggz.oggz audio/flacz.flaczapplication/x-tgstickerz.tgszJ@|(?:https?://)?(?:www\.)?(?:telegram\.(?:me|dog)|t\.me)/(@|\+|joinchat/)?ztg://(join)\?invite=z ^[a-z](?:(?!__)\w){1,30}[a-z\d]$ZFileInfozdc_id location sizedccs6t|}|d8}|D]}t|gt||VqdS)z Turns the given iterable into chunks of the specified size, which is 100 by default since that's what Telegram uses the most. rN)iter itertoolschainislice)iterablesizeitheadr2/tmp/pip-unpacked-wheel-c81u5j2r/telethon/utils.pychunksHsrcCsft|tjrF|jr(|jr(d|j|jS|jr4|jS|jr@|jSdSnt|tjtjtjfrb|j SdS)z} Gets the display name for the given :tl:`User`, :tl:`Chat` or :tl:`Channel`. Returns an empty string otherwise. z{} {}) isinstancer User last_name first_nameformatChat ChatForbiddenChanneltitleentityrrrget_display_nameSs  r$cCszt|WdStk r:t|tjtjfr6YdSYnXt|tjrN|j}t|tjtj tj fr|j dkrrdSt |j p~dSdS)z8Gets the corresponding extension for any Telegram media.z.jpgapplication/octet-streamr) get_input_photo TypeErrorrr UserProfilePhoto ChatPhotoMessageMediaDocumentdocumentDocumentZ WebDocumentZWebDocumentNoProxy mime_typer)mediarrr get_extensionhs"   r/cCstdt|j|dS)Nz!Cannot cast {} to any kind of {}.)r'rtype__name__)r#targetrrr_raise_cast_failsr3TcCs0z|jdkr|WSWnNtk rb|rrErJrIrKrLr"rrrrYs(         rYcCs~z*|jdkr|WS|jdkr(t|WSWntk rHt|dYnXztt|WStk rnYnXt|ddS)z2Similar to :meth:`get_input_peer`, but for dialogsl9Dr4InputDialogPeerN)r6r rZr7r3r9r')dialogrrrget_input_dialog+s  r\cCsz|jdkr|WSWntk r2t|dYnXt|tjrVtj|j|j|j dSt|tj rjt St|tj rt |jSt|tjrt |jSt|ddS)z4Similar to :meth:`get_input_peer`, but for documentsh[f InputDocumentr?r<file_referenceN)r6r7r3rr r,r^r?r<r` DocumentEmptyZInputDocumentEmptyr*get_input_documentr+Messager.)r+rrrrb=s$        rbcCs(z|jdkr|WSWntk r2t|dYnXt|tjrF|j}t|tjjtj fr`|j }t|tjrtj |j |j |jdSt|tjrtSt|tjjr|j}t|tjrt|jSt|tjrt|jSt|tjtjtjfrt|j St|tjtjtjtjfrtSt|ddS)z1Similar to :meth:`get_input_peer`, but for photosc InputPhotor_N) r6r7r3rr rcr.photosPhotoMessageMediaPhotophotorer?r<r` PhotoEmptyInputPhotoEmptymessagesrSZ full_chat ChannelFullr&Z chat_photorQZ profile_photor rrrOr@rrCrirrrr&Vs:            r&cCsz*|jdkr|WS|jdkr(t|WSWntk rHt|dYnXt|}t|tjrht|St|tj r|t St|ddS)z6Similar to :meth:`get_input_peer`, but for chat photoslt-)l_NInputChatPhotoN) r6r ZInputChatUploadedPhotor7r3r&rrerorkZInputChatPhotoEmptyrnrrrget_input_chat_photo|s     rpcCsz|jdkr|WSWntk r2t|dYnXt|tjrRtj|j|jdSt|tj rft St|tj r|t |j St|tjrt |jSt|ddS)z5Similar to :meth:`get_input_peer`, but for geo pointsi%0 InputGeoPoint)latlongN)r6r7r3rr ZGeoPointrqrrrsZ GeoPointEmptyZInputGeoPointEmptyMessageMediaGeo get_input_geogeorcr.)rvrrrrus        ruF)is_photo attributesforce_document voice_note video_notesupports_streamingttlc CszH|jdkr|WS|jdkr,tj||dWS|jdkrFtj||dWSWntk rft|dYnXt|tjrtjt|j |j |p|j dSt|tj tj j tjfrtjt||dSt|tjrtjt|j|p|j dSt|tjtjfrtjt||dSt|tjtjfr\|r0tj||dSt||||||d \}} tj|| |||d St|tjrtjtj|jj|jjd d St|tjrtj |j!|j"|j#d dSt|tj$rtj%t&|j'dSt|tj(rtj)t&|j'|j*|j+|j,dSt|tj-r,tj.t&|j'|j/|j0|j1|j2d dSt|tj3rFt4|j5St|tj6tj7tj8tj9tj:tj;frrt<St|tj=rt>|j?||dSt|tj@r|jAjBr|jCjCstDddd|jCjCD} nd} tjE|jA| |jCjF|jCjGdSt|tjHrtE|St|ddS)a Similar to :meth:`get_input_peer`, but for media. If the media is :tl:`InputFile` and ``is_photo`` is known to be `True`, it will be treated as an :tl:`InputMediaUploadedPhoto`. Else, the rest of parameters will indicate how to treat it. lFurd) ttl_secondsr] InputMedia)r?spoilerr~)r?r~)filer~)rxryrzr{r|)rr-rxZ force_filer~r?r<)r?r) phone_numberrrZvcard) geo_point)rperiodheadingproximity_notification_radius)rr!addressprovidervenue_idZ venue_type)rwr}z6Cannot cast unanswered quiz to any kind of InputMedia.cSsg|]}|jr|jqSr)correctoption).0rrrr sz#get_input_media..N)pollcorrect_answerssolutionsolution_entities)Ir6r ZInputMediaPhotoZInputMediaDocumentr7r3rrhr&rirr~rgrfrjr*rbr+r,raZ InputFileZ InputFileBigZInputMediaUploadedPhotoget_attributesZInputMediaUploadedDocumentZMessageMediaGameZInputMediaGameZ InputGameIDZgamer?r<ZMessageMediaContactZInputMediaContactrrrrtZInputMediaGeoPointrurvZMessageMediaGeoLiveZInputMediaGeoLiverrrZMessageMediaVenueZInputMediaVenuer!rrrZMessageMediaDiceZInputMediaDiceZemoticonZMessageMediaEmptyZMessageMediaUnsupportedZChatPhotoEmptyZUserProfilePhotoEmptyr)r(ZInputMediaEmptyrcget_input_mediar.ZMessageMediaPollrZquizresultsr'ZInputMediaPollrrZPoll) r.rwrxryrzr{r|r}attrsmimerrrrrs             rcCsfzBt|trt|WS|jdkr(|WS|jdkr@t|jWSWntk rVYnXt|ddS)z:Similar to :meth:`get_input_peer`, but for input messages.iżT yrN)rintr InputMessageIDr6r?r7r3messagerrrget_input_message-s    rcCsVz2|jdkr|WS|jdkr0tj|j|jdWSWntk rPt|dYnXdS)z7Similar to :meth:`get_input_peer`, but for input calls.iaXi rInputGroupCallN)r6r rr?r<r7r3)callrrrget_input_group_call<s  rc Csp|sdS||}z|t|d}Wn:tk rfz ||}Wntk r`d}YnXYnX||fS)zE Returns ``(entity, input_entity)`` for the given entity ID. )NNrN)get resolve_idZ_as_input_peerr7r')Z entity_identitiescacher9r#r5rrr_get_entity_pairGs  rcCsn|dkr dSt|tr|St|tjr,|jSz|jdkr@|jWSWntk rVYnXtdt |dS)z7Similar to :meth:`get_input_peer`, but for message IDs.NrzInvalid message type: {}) rrr rr?r6r7r'rr0rrrrget_message_id\s    rc Cs&tsdSd}d}d}zzt|tr.t|d}n8t|trDt|}n"|}d}t |ddrb| }nd}|stWWdS| }t |dd}tj j tjj|d|g|d}tj|WW:Stk r}ztd ||j|W5d}~XYnXW5|r |r |n|r |r ||XdS) NTrbFseekablenamerzfile:)sourcetagsfilenamezFailed to analyze %s: %s %s)hachoircloseseekrstropenbytesioBytesIOgetattrrtellparserguessZ guessParserstreamZ InputIOStreammetadataZextractMetadata Exception_logwarning __class__)rrZ close_streamrposrrerrr _get_metadataqs@        (   r)rxr-ryrzr{r|thumbcCs<t|tr|n t|dd}|dkr0t|d}tjttj |i} t |rt |} | r| drr| d} n| dr| d} nd} tj|| dr| dnd| t| dr| djndd | tj<|st|rt |} | rNtj|| d r | d nd | d r$| d nd t| dr@| djnd |d } nx|rt |} d }d }| r| d r| d }| r| d r| d }tjd||||d} ntjdd d ||d} | | tj<|rtj| krd| tj_ntjddd| tj<|r"|D]}|| t|<q|s,d}t| |fS)zp Get a list of attributes for the given file and the mime type as a tuple ([attribute], mime_type). rZunnamedNrauthorZartistr!duration)voicer! performerrwidthrheight) round_messagewhrr|)rr|T)rr%)rrr mimetypes guess_typer ZDocumentAttributeFilenameospathbasenameis_audiorhasrDocumentAttributeAudiorsecondsis_videoDocumentAttributeVideorr0listvalues)rrxr-ryrzr{r|rrZ attr_dictmrdocZt_mrrarrrrs          rcssdStfdddDrsz&sanitize_parse_mode..)parseunparsecss|]}t|VqdSr)callablerrrrrsc@seZdZeddZdS)z'sanitize_parse_mode..CustomModecSstdSr)NotImplementedError)textrrrrrsz/sanitize_parse_mode..CustomMode.unparseN)r1 __module__ __qualname__ staticmethodrrrrr CustomModesr)ZmdrZhtmrzUnknown parse mode {}zInvalid parse mode type {}) allrrrrrrrlowerKeyError ValueErrorrr')rrrrrsanitize_parse_modes, rcCst|}|j|jfS)z Similar to :meth:`get_input_peer`, but for input messages. Note that this returns a tuple ``(dc_id, location)``, the ``dc_id`` being present if known. )_get_file_infodc_idlocation)rinforrrget_input_locationsrcCsz|jdkrtd|dWSWntk r:t|dYnXt|tjrN|j}t|tjrb|j }nt|tj rt|j }t|tj rt|j tj|j|j|jdd|jSt|tjrt|j tj|j|j|j|jdjdt|jdSt|ddS)Nib#ZInputFileLocationr)r?r<r`Z thumb_size)r6 _FileInfor7r3rr rcr.r*r+rhrir,rZInputDocumentFileLocationr?r<r`rrgZInputPhotoFileLocationsizesr0_photo_size_byte_count)rrrrr's<          rcCsNt|trtj|dSt|tjr,|jSt|ddrBt |j St |SdS)z Gets the extension for the given file, which can be either a str or an ``open()``'ed file (which has a ``.name`` attribute). rrN) rrrrsplitextpathlibPathsuffixr_get_extensionrr/rrrrrHs    rcCs0tdt|tj}|rdStt|tjSdS)zT Returns `True` if the file extension looks like an image file to Telegram. z \.(png|jpe?g)TN)rematchr IGNORECASErresolve_bot_file_idr rg)rrrrris_imageYsrcCstdt|tjS)zQ Returns `True` if the file extension looks like a gif file to Telegram. z\.gif)rrrrrrrris_gifdsrcCs\t|}|s8t|}|r2|dr2|ddSdSn d|}t|dpPddSdS)z2Returns `True` if the file has an audio mime type.r-zaudio/FrrrNrrrr startswithrrrextrrrrrksrcCs\t|}|s8t|}|r2|dr2|ddSdSn d|}t|dpPddSdS)z1Returns `True` if the file has a video mime type.r-zvideo/FrrrNrrrrrrysrcCst|ttttttfS)a  Returns `True` if the given object looks like a list. Checking ``if hasattr(obj, '__iter__')`` and ignoring ``str/bytes`` is not enough. Things like ``open()`` are also iterable (and probably many other things), so just support the commonly known list-like objects. )rrtuplesetdictranger)objrrr is_list_likesrcCs4t|trt|Stddt|}|r0|SdS)z:Parses the given phone, or returns `None` if it's invalid.z[+()\s-]rN)rrrrsubisdigit)Zphonerrr parse_phones  rcCsr|}t|pt|}|rT||d}t|d}|rJ|dfS|d}t|rj| dfSdSdS)a< Parses the given username or channel access hash, given a string, username or URL. Returns a tuple consisting of both the stripped, lowercase username and whether it is a joinchat/ hash (in which case is not lowercase'd). Returns ``(None, False)`` if the ``username`` or link is not valid. NrT/F)NF) strip USERNAME_REr TG_JOIN_REendboolgrouprstripVALID_USERNAME_REr)usernamerZ is_inviterrrparse_usernames    rcCsBt|}g}|D],}|j}|j|j}|t|||q|S)aK Gets the inner text that's surrounded by the given entities. For instance: text = 'hey!', entity = MessageEntityBold(2, 2) -> 'y!'. :param text: the original text. :param entities: the entity or entities that must be matched. :return: a single result or a list of the text surrounded by the entities. )roffsetlengthappendr )rrresultrstartrrrrget_inner_texts  r#c CsZz.t|tr$t|\}}||WS|jdkr4|WSt|tjjtjtjtj tj fr\|j WSt|tj rvt |jWSt|tjrt|jWSt|tjrt|jWS|jdkrt|jWSt|ddd}t|tjtjfrt|jWSt|tjr t|jWSt|tjtjfr.t |jWSWnttfk rJYnXt|ddS)NiV)io|}l|3F)rVrWZPeer)rrrr6r ZcontactsZ ResolvedPeerZInputNotifyPeerZTopPeerDialogZ DialogPeerrKrm PeerChannelr?rOPeerUserr@rTrEr9r>rJrArUrBrNrGr7r'r3)rKpidclsrrrget_peers@        r)cCst|tr|r|St|dSt|tjr4t|dz t|}Wntk r^t|dYnXt|tjrr|j St|tj rd|j krdksnt|j d|_ |r|j S|j Sd|j krdksnt|j d|_ |s|j Sd|j  SdS)a Convert the given peer into its marked ID by default. This "mark" comes from the "bot api" format, and with it the peer type can be identified back. User ID is left unmodified, chat ID is negated, and channel ID is "prefixed" with -100: * ``user_id`` * ``-chat_id`` * ``-100channel_id`` The original ID and the peer type class can be returned with a call to :meth:`resolve_id(marked_id)`. rz.int (you might want to use client.get_peer_id)rlc( J)N) rrrr r;r3r)r'r&rErTrUrG)rKZadd_markrrr get_peer_ids&      r+cCs@|dkr|tjfS| }|dkr2|d8}|tjfS|tjfSdS)zCGiven a marked ID, returns the original ID and its :tl:`Peer` type.rr*N)r r&r%rT)Z marked_idrrrrs  rcCsN|s|Sd}d}|D]0}|dkr2|||7}d}q||7}t|g}q||S)z, Decodes run-length-encoded `data`. r)datanewlastcurrrr _rle_decode's  r3cCsLd}d}|D]:}|s|d7}q |r8|dt|g7}d}|t|g7}q |S)Nr,rrr-r.)stringr0countr2rrr _rle_encode;s r6c Cs@zt|dt|dWStjttfk r:YdSXdS)a( Decodes a url-safe base64-encoded string into its bytes by first adding the stripped necessary padding characters. This is the way Telegram shares binary data as strings, such as Bot API-style file IDs or invite links. Returns `None` if the input string was not valid. =N)base64urlsafe_b64decodelenbinasciiErrorrr'r4rrr_decode_telegram_base64Js r?c Cs<zt|ddWStjttfk r6YdSXdS)z0 Inverse for `_decode_telegram_base64`. =asciiN)r9urlsafe_b64encoderdecoder<r=rr'r>rrr_encode_telegram_base64ZsrDc Csxtt|}|sdS|dd|d}}|dkr6dS|dkrJt|dksb|dkrft|dkrf|dkrtd|\}}}}ntd |\}}}}}d |krd ksndSg}|d ks|d kr|tjd|d kdnl|dks|dkr|tjddd|dkdn<|dkr2|tj dt dn|dkrJ|t tj ||dddd||dd S|dkr~t|dks|dkrtt|dkrt|dkrtd|\}}}}} } } n^t|dkrtd|\}}}}} } } }n2t|dkrtd|\ }}}}}} }} }ndSd |kr2d ks8ndS|sD|rHdnd } tj ||ddtj| dddd!g|dd"SdS)#a Given a Bot API-style `file_id `, returns the media it represents. If the `file_id ` is not valid, `None` is returned instead. Note that the `file_id ` does not have information such as image dimensions or file size, so these will be zero if present. For thumbnails, the photo ID and hash will always be zero. Nr)r8rEr8z.Nz LQz>LLQ)NNN)r) rrrr;rfromhexr?rTrUerrorr')linkZ link_hashZis_linkpayloadrrrresolve_invite_links    rlc CsfzDtdt|\}}}}|dkr.t| nt|}||||fWStjtfk r`YdSXdS)a} Resolves an inline message ID. Returns a tuple of ``(message id, peer, dc id, access hash)`` The ``peer`` may either be a :tl:`PeerUser` referencing the user who sent the message via the bot in a private conversation or small group chat, or a :tl:`PeerChannel` if the message was sent in a channel. The ``access_hash`` does not have any use yet. z}td|||dd}||O}td||||d<q0t|d |S) af Encodes the input `bytes` into a 5-bit byte-string to be used as a voice note's waveform. See `decode_waveform` for the reverse operation. Example .. code-block:: python chat = ... file = 'my.ogg' # Send 'my.ogg' with a ascending-triangle waveform await client.send_file(chat, file, attributes=[types.DocumentAttributeAudio( duration=7, voice=True, waveform=utils.encode_waveform(bytes(range(2 ** 5)) # 2**5 because 5-bit )] # Send 'my.ogg' with a square waveform await client.send_file(chat, file, attributes=[types.DocumentAttributeAudio( duration=7, voice=True, waveform=utils.encode_waveform(bytes((31, 31, 15, 15, 15, 15, 31, 31)) * 4) )] rIrMr.updater)rN)rr)rr)r)rr maprcompiler;r=rrrbr rrr r )rrrrrrZlast_entZ cur_limitsplitrxrZcur_textZnew_textZcur_entZnew_entrrrr split_texts:/    " &  rc@seZdZddZddZdS)AsyncClassWrappercCs ||_dSr)wrapped)selfrrrr__init__szAsyncClassWrapper.__init__cs,t|j|fdd}tr$|SSdS)Ncs"||}t|r|IdHS|Sr)inspect isawaitable)argsrvalrrrwrappers z.AsyncClassWrapper.__getattr__..wrapper)rrr)ritemrrrr __getattr__s   zAsyncClassWrapper.__getattr__N)r1rrrrrrrrrsrcCsXt|dks|ddkr|Std}d}|d|d<|d|d<t||dd |S) z Adds the JPG header and footer to a stripped image. Ported from https://github.com/telegramdesktop/tdesktop/blob/bec39d89e19670eb436dc794a8f20b657cb87c71/Telegram/SourceFiles/ui/image/image.cpp#L225 rJrrsoJFIFC(#(#!#-+(0s                           `(&   3 W! !   " 0  k0% )_