B fd@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z ddl mZddlmZddlmZmZddlmZddlmZdd lmZmZmZmZmZddlZdd lmZm Z dd lm!Z!dd lm"Z"dd lm#Z#ddl$m%Z%ddl&m'Z'ddl&m(Z(m)Z)m*Z*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:ddl;mZ>m?Z?m@Z@ddlAmAZAddlBmCZCddlDmEZEeFeGZHGddde/ZIGdd d ZJdS)!N)ThreadPoolExecutor)datetime timedelta)sha256) import_module)StringIOBytesIO) MimeTypes)Path)UnionListOptionalCallableAsyncGenerator) __version__ __license__)enums)raw)utils)aes)CDNFileHashMismatch)SessionPasswordNeededVolumeLocNotFoundChannelPrivate BadRequest)Handler)Methods)AuthSession) FileStorage MemoryStorage)UserTermsOfService)ainput) Dispatcher)FileIdFileTypeThumbnailSource) mime_types)Parser)MsgIdcseZdZdZdeZedeZ e de Z dZ eejdjZedZedepnddZeZd Zd ZeZee e!d d ee e e d d d d d d d d d eed e"j#j$d d e%j&d efe'e(e)e'fe'e'e'e'e'e*e+e*e'e'e*e'e'e'e)e'e+d e*e*e)e*e)dfdd Z,ddZ-ddZ.ddZ/ddZ0ddZ1e2dddZ3e4d ddd Z5e6e(e7j8j2e7j8j9e7j8j:fe*d!d"d#Z;d$d%Zd*d+Z?d6e@e)e)e)eAeBe4eCeDd fd-d.d/ZEe'e4e'd0d1d2ZFe'e4e'd3d4d5ZGZHS)7ClientaPyrogram Client, the main means for interacting with Telegram. Parameters: name (``str``): A name for the client, e.g.: "my_account". api_id (``int`` | ``str``, *optional*): The *api_id* part of the Telegram API key, as integer or string. E.g.: 12345 or "12345". api_hash (``str``, *optional*): The *api_hash* part of the Telegram API key, as string. E.g.: "0123456789abcdef0123456789abcdef". app_version (``str``, *optional*): Application version. Defaults to "Pyrogram x.y.z". device_model (``str``, *optional*): Device model. Defaults to *platform.python_implementation() + " " + platform.python_version()*. system_version (``str``, *optional*): Operating System version. Defaults to *platform.system() + " " + platform.release()*. lang_code (``str``, *optional*): Code of the language used on the client, in ISO 639-1 standard. Defaults to "en". ipv6 (``bool``, *optional*): Pass True to connect to Telegram using IPv6. Defaults to False (IPv4). proxy (``dict``, *optional*): The Proxy settings as dict. E.g.: *dict(scheme="socks5", hostname="11.22.33.44", port=1234, username="user", password="pass")*. The *username* and *password* can be omitted if the proxy doesn't require authorization. test_mode (``bool``, *optional*): Enable or disable login to the test servers. Only applicable for new sessions and will be ignored in case previously created sessions are loaded. Defaults to False. bot_token (``str``, *optional*): Pass the Bot API token to create a bot session, e.g.: "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11" Only applicable for new sessions. session_string (``str``, *optional*): Pass a session string to load the session in-memory. Implies ``in_memory=True``. in_memory (``bool``, *optional*): Pass True to start an in-memory session that will be discarded as soon as the client stops. In order to reconnect again using an in-memory session without having to login again, you can use :meth:`~pyrogram.Client.export_session_string` before stopping the client to get a session string you can pass to the ``session_string`` parameter. Defaults to False. phone_number (``str``, *optional*): Pass the phone number as string (with the Country Code prefix included) to avoid entering it manually. Only applicable for new sessions. phone_code (``str``, *optional*): Pass the phone code as string (for test numbers only) to avoid entering it manually. Only applicable for new sessions. password (``str``, *optional*): Pass the Two-Step Verification password as string (if required) to avoid entering it manually. Only applicable for new sessions. workers (``int``, *optional*): Number of maximum concurrent workers for handling incoming updates. Defaults to ``min(32, os.cpu_count() + 4)``. workdir (``str``, *optional*): Define a custom working directory. The working directory is the location in the filesystem where Pyrogram will store the session files. Defaults to the parent directory of the main script. plugins (``dict``, *optional*): Smart Plugins settings as dict, e.g.: *dict(root="plugins")*. parse_mode (:obj:`~pyrogram.enums.ParseMode`, *optional*): Set the global parse mode of the client. By default, texts are parsed using both Markdown and HTML styles. You can combine both syntaxes together. no_updates (``bool``, *optional*): Pass True to disable incoming updates. When updates are disabled the client can't receive messages or other updates. Useful for batch programs that don't need to deal with updates. Defaults to False (updates enabled and received). takeout (``bool``, *optional*): Pass True to let the client use a takeout session instead of a normal one, implies *no_updates=True*. Useful for exporting Telegram data. Methods invoked inside a takeout session (such as get_chat_history, download_media, ...) are less prone to throw FloodWait exceptions. Only available for users, bots will ignore this parameter. Defaults to False (normal session). sleep_threshold (``int``, *optional*): Set a sleep threshold for flood wait exceptions happening globally in this client instance, below which any request that raises a flood wait will be automatically invoked again after sleeping for the required amount of time. Flood wait exceptions requiring higher waiting times will be raised. Defaults to 10 seconds. hide_password (``bool``, *optional*): Pass True to hide the password when typing it during the login. Defaults to False, because ``getpass`` (the library used) is known to be problematic in some terminal environments. max_concurrent_transmissions (``bool``, *optional*): Set the maximum amount of concurrent transmissions (uploads & downloads). A value that is too high may result in network related issues. Defaults to 1. z Pyrogram  enrzT^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:joinchat/|\+))([\w-]+)$ i,r$NFzenums.ParseMode)nameapi_idapi_hash app_version device_modelsystem_version lang_codeipv6proxy test_mode bot_tokensession_string in_memory phone_number phone_codepasswordworkersworkdirplugins parse_mode no_updatestakeoutsleep_threshold hide_passwordmax_concurrent_transmissionscst||_|rt|nd|_||_||_||_||_| |_ ||_ | |_ | |_ | |_| |_| |_||_||_||_||_t||_||_||_||_||_||_||_||_t|jdd|_|jrt |j|j|_!n$|jrt |j|_!nt"|j|j|_!t#||_$t%|_&t'||_(d|_)i|_*t+,|_-t+.|j|_/t+.|j|_0d|_1d|_2d|_3d|_4d|_5t6d|_7d|_8t+9|_:t;<|_=t+>|_?dS)Nr)Zthread_name_prefixi')@super__init__r1intr2r3r4r5r6lowerr7r8r9r:r;r<r=r>r?r@rAr rBrCrDrErFrGrHrIrexecutorr storagerr% dispatcherr+Zrnd_idr*parsersessionZmedia_sessionsasyncioLockZmedia_sessions_lock SemaphoreZsave_file_semaphoreget_file_semaphoreZ is_connectedZis_initializedZ takeout_idZdisconnect_handlermeCacheZ message_cacheZupdates_watchdog_taskEventupdates_watchdog_eventrnowlast_update_timeZget_event_looploop)selfr1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrI) __class__3/tmp/pip-unpacked-wheel-rcokkf2l/pyrogram/client.pyrKsd         zClient.__init__cCs|S)N)start)r^r`r`ra __enter__-szClient.__enter__cGs&y |Wntk r YnXdS)N)stopConnectionError)r^argsr`r`ra__exit__0s zClient.__exit__cs|IdHS)N)rb)r^r`r`ra __aenter__6szClient.__aenter__cs,y|IdHWntk r&YnXdS)N)rdre)r^rfr`r`ra __aexit__9szClient.__aexit__cstxnyt|j|jIdHWntjk r6YnXPt|jt |jdkr| t j j IdHqWdS)N)seconds)rSwait_forrZwaitUPDATES_WATCHDOG_INTERVAL TimeoutErrorrr[r\rinvoker functionsupdatesZGetState)r^r`r`raupdates_watchdog?szClient.updates_watchdog)returnc s|jr||jIdHStdtdtdtdxy~|jsxr#rMZ send_coderZMESSAGErZ SentCodeTypeZAPPrwCALLZ FLASH_CALLZ FRAGMENT_SMSZ EMAIL_CODEtyper?Zsign_inZphone_code_hashrformatZget_password_hintr@rHZsend_recovery_codeZrecover_password Exceptionlog exceptionZcheck_password isinstancer!Zsign_upr"textZaccept_terms_of_serviceid) r^valueconfirmZ sent_codeeZsent_code_descriptionsZ signed_inZ email_patternZ recovery_codeZ first_name last_nameZ signed_upr`r`ra authorizeKs        0  zClient.authorize)rDcCs ||_dS)aSet the parse mode to be used globally by the client. When setting the parse mode with this method, all other methods having a *parse_mode* parameter will follow the global value by default. Parameters: parse_mode (:obj:`~pyrogram.enums.ParseMode`): By default, texts are parsed using both Markdown and HTML styles. You can combine both syntaxes together. Example: .. code-block:: python from pyrogram import enums # Default combined mode: Markdown + HTML await app.send_message("me", "1. **markdown** and html") # Force Markdown-only, HTML is disabled app.set_parse_mode(enums.ParseMode.MARKDOWN) await app.send_message("me", "2. **markdown** and html") # Force HTML-only, Markdown is disabled app.set_parse_mode(enums.ParseMode.HTML) await app.send_message("me", "3. **markdown** and html") # Disable the parser completely app.set_parse_mode(enums.ParseMode.DISABLED) await app.send_message("me", "4. **markdown** and html") # Bring back the default combined mode app.set_parse_mode(enums.ParseMode.DEFAULT) await app.send_message("me", "5. **markdown** and html") N)rD)r^rDr`r`raset_parse_modes$zClient.set_parse_mode)peersrsc srd}g}xP|D]F}t|ddr(d}qd}d}t|tjjr|j}|j}|jrZ|jn|j rp|j djnd}|j }|j rdnd} nt|tjj tjj fr|j }d}d} nt|tjjrt|j}|j}|jr|jn|j r|j djnd}|jr dnd } n4t|tjjrt|j}|j}|jrpeer_idrZ peer_typer`r`ra fetch_peerssF   zClient.fetch_peersc st|_t|tjjtjjfrt| |j IdH| |j IdHf}dd|j D}dd|j D}xR|j D]N}t t t |ddddddpt |dd}t |dd}t |dd}t|tjjrt|t|tjjr|r|j} t| tjjsy\|tjj j|t|IdHtjjtjj|jj|jjd gd |||d IdH} Wntk rpYn@Xt| tjj js|d d| j D|d d| j D|j j!"|||fqtWnt|tjj#tjj$fr~|tjj j%|j&|j'|j(ddIdH} | j)rZ|j j!"tjj*| j)d|j&|j'ddd| j Ddd| j Dfn"| j+r|j j!"| j+diifnBt|tjj,r|j j!"|jiifnt|tjj-rt|dS)NcSsi|] }||jqSr`)r).0ur`r`ra sz)Client.handle_updates..cSsi|] }||jqSr`)r)rcr`r`rar!smessager channel_idpts pts_count)Zmin_idZmax_id)ranges)rfilterrlimitcSsi|] }||jqSr`)r)rrr`r`rarHscSsi|] }||jqSr`)r)rrr`r`rarIs)rdateZqtsr)rrrcSsi|] }||jqSr`)r)rrr`r`rar\scSsi|] }||jqSr`)r)rrr`r`rar]s).rr[r\rrrZUpdatesZUpdatesCombinedanyruserschatsrqrZUpdateChannelTooLongr~infoZUpdateNewChannelMessagerZ MessageEmptyrorpZGetChannelDifferenceZ resolve_peerrrZChannelMessagesFilterZ MessageRangerrZChannelDifferenceEmptyupdaterPZ updates_queue put_nowaitZUpdateShortMessageZUpdateShortChatMessageZ GetDifferencerrrZ new_messagesZUpdateNewMessageZ other_updatesZ UpdateShortZUpdatesTooLong) r^rqrrrrrrrrZdiffr`r`rahandle_updatesst      zClient.handle_updatesc s|jIdHt|jIdHdk|jIdHdk|jIdHdk|jIdHdkg}|r(|jrr|jszt d|j|jIdH|j dIdH|j dIdH|j|jIdH|jt ||j IdH|jIdH IdHIdH|jdIdH|jdIdHn|jIdHs|jrX|j|jIdHnxydttdIdH}|dkrtdwZtd|dIdH}|dkr|j|IdHPWn,tk r}z t|Wdd}~XYnXqZWdS) Nz_The API key is required for new authorizations. More info: https://docs.pyrogram.org/start/authrz&Enter the api_id part of the API key: z Invalid valuezIs "z" correct? (y/N): ru)rOopenrr:auth_keyuser_idZis_botr2r3AttributeErrordc_idrrcreaterLr#ryrMr})r^Z session_emptyrrrr`r`ra load_sessiongsD $  zClient.load_sessionc Cs>|jrD|j}x6dD](}||grdd|j|D||<qWndS|ddr:|d}|dg}|dg}d }|sZxtt|d d d D]}d |jj |j f}t |} xt |  D]} yfx`t| | jD]P\} } t| trt| tr|| | td |jt| j| | ||d7}qWWqtk rLYqXqWqWnLxH|D]>\}} |d |}d}y t |} Wn*tk rtd|j|wbYnXdt| krtd|j|qb| dkrt |  } d}x| D]} ylxft| | jD]V\} } t| tr t| tr || | td |jt| j| | ||d7}q WWn2tk r|rtd|j| |YnXqWqbW|rxH|D]>\}} |d |}d}y t |} Wn*tk rtd|j|wYnXdt| kr(td|j|q| dkrBt |  } d}x| D]} ylxft| | jD]V\} } t| tr\t| tr\|| | td|jt| j| | ||d8}q\WWn2tk r|rtd|j| |YnXqHWqW|d kr*td|j||dkrdnd|ntd|j|dS)N)includeexcludecSs,g|]$}|d|ddp$dfqS)rr$N)split)rir`r`ra sz'Client.load_plugins..ZenabledTrootrrr./z*.pyz*[{}] [LOAD] {}("{}") in group {} from "{}"r$z-[%s] [LOAD] Ignoring non-existent module "%s"__path__z#[%s] [LOAD] Ignoring namespace "%s"Fz9[{}] [LOAD] Ignoring non-existent function "{}" from "{}"z/[%s] [UNLOAD] Ignoring non-existent module "%s"z%[%s] [UNLOAD] Ignoring namespace "%s"z,[{}] [UNLOAD] {}("{}") from group {} in "{}"z;[{}] [UNLOAD] Ignoring non-existent function "{}" from "{}"z.[{}] Successfully loaded {} plugin{} from "{}"sz[%s] No plugin loaded from "%s")rCcopygetsortedr replacerglobjoinparentpartsstemrvarskeysrhandlersrrrL add_handlerr~rr|r1r{__name__r} ImportErrorwarningdirZremove_handler)r^rCoptionrrrcountpath module_pathmoduler1handlerrrZwarn_non_existent_functionsr`r`ra load_pluginss                         zClient.load_pluginsc s0|\}}}}}}}|s$tj|ddndtjtddtj||d} |rVtnt| d} yJxD| ||dd||2y 3dH} Wn t k sX| | qxYWWnHt k r} z*|s| t| t| tjr| dSd} ~ XYn:X|r|| _| S| tj| d} t| | | SdS)NT)exist_okz\\rz.tempwbr)osmakedirsrabspathresubrrrget_fileStopAsyncIterationwrite BaseExceptioncloseremoverrSZCancelledErrorr1splitextshutilmove)r^packetfile_id directory file_namer= file_sizeprogress progress_argsZtemp_file_pathfilechunkr file_pathr`r`rahandle_download s*$.   zClient.handle_downloadr`)rrroffsetrrrsc Cs|j4IdHv|j}|tjkr|jdkrBtjj|j|jd}n8|jdkr`tjj |j d}ntjj t |j|jd}tjj ||j|jtjkd} nD|tjkrtjj|j|j|j|jd} ntjj|j|j|j|jd} d} t|pd} d} t|| } |j}t||||jIdHkr@t|||jIdHIdHn|jIdH|jIdHd d }z y| IdH||jIdHkr|!tj"j#j$|d IdH}|!tj"j#j%|j&|j'd IdH|j!tj"j(j)| | | d ddIdH}t*|tjj(j+rx|j'}|V| d7} | | 7} |rt,j-||dkrBt.| |n| |f|}t/0|rj|IdHn|j12|j3|IdHt4|| ks| | krP|j!tj"j(j)| | | d ddIdH}qWn\t*|tjj(j5r$t||jt||j|jIdHIdH|jIdHd d d}zy| IdHx|!tj"j(j6|j7| | dIdH}t*|tjj(j8ry&|!tj"j(j9|j7|j:dIdHWnt;k rPYnXq6|j'}t<=||j>t?|j@dd| dAdd}|!tj"j(jB|j7| dIdH}xLtC|D]@\}}||jD||jD|d}tEF|jGtH|Ikdq W|V| d7} | | 7} |rt,j-||dkrt.| |n| |f|}t/0|r|IdHn|j12|j3|IdHt4|| ks| | kr6Pq6WWn(tJk r}z|Wdd}~XYnXWd|KIdHXWnFtLjMk r@Yn.tJk rl}ztNO|Wdd}~XYnXWd|KIdHXWdQIdHRXdS)Nr)rr)chat_id)rr)rZphoto_idbig)rrfile_referenceZ thumb_sizeiiT)is_media)r)rbytes)locationrr)rGr$)rZis_cdn) file_tokenrr)r request_tokenr0r)rrz$h.hash == sha256(cdn_chunk).digest())PrV file_typer'Z CHAT_PHOTOrrrZ InputPeerUserZchat_access_hashZ InputPeerChatZInputPeerChannelrrZInputPeerPhotoFileLocationZmedia_idZthumbnail_sourcer(ZCHAT_PHOTO_BIGZPHOTOZInputPhotoFileLocationrrZthumbnail_sizeZInputDocumentFileLocationabsrrrOrr:rrrbrorpauthZExportAuthorizationZImportAuthorizationrruploadZGetFilerFile functoolspartialrinspectiscoroutinefunctionr]Zrun_in_executorrNlenZFileCdnRedirectZ GetCdnFilerZCdnFileReuploadNeededZReuploadCdnFilerrrZctr256_decryptZencryption_key bytearrayZ encryption_ivto_bytesZGetCdnFileHashes enumeraterrcheckhashrdigestr}rdpyrogramZStopTransmissionr~r)r^rrrrrrrrrcurrenttotal chunk_sizeZ offset_bytesrrRZ exported_authrrfuncZ cdn_sessionZr2Zdecrypted_chunkhashesrhZ cdn_chunkrr`r`rar)s            8 (     zClient.get_file)filenamerscCs|j|dS)Nr) mimetypes guess_type)r^rr`r`raguess_mime_typeszClient.guess_mime_type) mime_typerscCs |j|S)N)rguess_extension)r^rr`r`rarszClient.guess_extension)rrrNr`)Ir __module__ __qualname____doc__rZ APP_VERSIONplatformpython_implementationpython_versionZ DEVICE_MODELsystemreleaseZSYSTEM_VERSIONZ LANG_CODEr sysargvrZ PARENT_DIRrcompileZINVITE_LINK_RErr cpu_countZWORKERSZWORKDIRrmZMAX_CONCURRENT_TRANSMISSIONSr rreadfprr)rZ ParseModeDEFAULTrZSLEEP_THRESHOLDstrr rLbooldictrKrcrgrhrirrr!rr rr rrrrrrrrrr&rtuplerrrrr __classcell__r`r`)r_rar,@stt  NI v&*0P3r Qr,c@s*eZdZedddZddZddZdS) rX)capacitycCs||_i|_dS)N)r3store)r^r3r`r`rarK szCache.__init__cCs|j|dS)N)r4r)r^keyr`r`ra __getitem__ szCache.__getitem__cCs^||jkr|j|=||j|<t|j|jkrZx,t|jddD]}|jtt|j=q@WdS)Nrr$)r4r r3rangenextiter)r^r5r_r`r`ra __setitem__s   zCache.__setitem__N)rr r!rLrKr6r;r`r`r`rarXsrX)KrSrr loggingrr#rrr(Zconcurrent.futures.threadrrrhashlibr importlibriorrrr pathlibr typingr r r rrrrrrrrZpyrogram.cryptorZpyrogram.errorsrrrrrZpyrogram.handlers.handlerrZpyrogram.methodsrZpyrogram.sessionrrZpyrogram.storagerr Zpyrogram.typesr!r"Zpyrogram.utilsr#rPr%rr&r'r(r)rQr*Zsession.internalsr+ getLoggerrr~r,rXr`r`r`rasZ                  O