N[ijxddlZddlZddlmZddlmZmZmZmZddlZddlm Z ddl m Z m Z m Z dZGddZy) N)datetime)UnionOptionalCallableBinaryIO)types)FileIdFileType PHOTO_TYPESz downloads/cZeZdZeddddfdddedefd ed ed ed ed ede eee ffdZ y) DownloadMediaFTNselfzpyrogram.Clientmessagez types.Message file_name in_memoryblockprogress progress_argsreturnc 2Kd}t|tjr#|D]}t||d} | nt d|} t| t r| } n | j } tj| } | j} t| dd} t| dd}t| dd}t| d d}tjj|\}}|xs| xsd}tjj|s|j|xstz }|s0|j!|}| t"vrd }n| t$j&k(r|xsd }n| t$j(t$j*t$j,fvr|xsd }nP| t$j.k(r|xsd }n6| t$j0k(r|xsd}n| t$j2k(r|xsd}nd}dj5t%| jj6j9|xst;j<j?d|jA|}|jC| ||||||f}|r |d{StEjFjI|y7(w)a!Download the media from a message. .. include:: /_includes/usable-by/users-bots.rst Parameters: message (:obj:`~pyrogram.types.Message` | ``str``): Pass a Message containing the media, the media itself (message.audio, message.video, ...) or a file id as string. file_name (``str``, *optional*): A custom *file_name* to be used instead of the one provided by Telegram. By default, all files are downloaded in the *downloads* folder in your working directory. You can also specify a path for downloading files in a custom location: paths that end with "/" are considered directories. All non-existent folders will be created automatically. in_memory (``bool``, *optional*): Pass True to download the media in-memory. A binary file-like object with its attribute ".name" set will be returned. Defaults to False. block (``bool``, *optional*): Blocks the code execution until the file has been downloaded. Defaults to True. progress (``Callable``, *optional*): Pass a callback function to view the file transmission progress. The function must take *(current, total)* as positional arguments (look at Other Parameters below for a detailed description) and will be called back each time a new file chunk has been successfully transmitted. progress_args (``tuple``, *optional*): Extra custom arguments for the progress callback function. You can pass anything you need to be available in the progress callback scope; for example, a Message object or a Client instance in order to edit the message with the updated progress status. Other Parameters: current (``int``): The amount of bytes transmitted so far. total (``int``): The total size of the file. *args (``tuple``, *optional*): Extra custom arguments as defined in the ``progress_args`` parameter. You can either keep ``*args`` or add every single extra argument in your function signature. Returns: ``str`` | ``None`` | ``BinaryIO``: On success, the absolute path of the downloaded file is returned, otherwise, in case the download failed or was deliberately stopped with :meth:`~pyrogram.Client.stop_transmission`, None is returned. Otherwise, in case ``in_memory=True``, a binary file-like object with its attribute ".name" set is returned. Raises: ValueError: if the message doesn't contain any downloadable media Example: Download media to file .. code-block:: python # Download from Message await app.download_media(message) # Download from file id await app.download_media(message.photo.file_id) # Keep track of the progress while downloading async def progress(current, total): print(f"{current * 100 / total:.1f}%") await app.download_media(message, progress=progress) Download media in-memory .. code-block:: python file = await app.download_media(message, in_memory=True) file_name = file.name file_bytes = bytes(file.getbuffer()) ) audiodocumentphotosticker animationvideovoice video_notenew_chat_photoNz3This message doesn't contain any downloadable mediar file_sizer mime_typedatez.jpgz.oggz.mp4z.zipz.webpz.mp3z.unknownz {}_{}_{}{}z%Y-%m-%d_%H-%M-%S)% isinstancerMessagegetattr ValueErrorstrfile_idr decode file_typeospathsplitisabs PARENT_DIRDEFAULT_DOWNLOAD_DIRguess_extensionr r VOICEVIDEO ANIMATION VIDEO_NOTEDOCUMENTSTICKERAUDIOformatnamelowerrnowstrftimernd_idhandle_downloadasyncioget_event_loop create_task)rrrrrrravailable_mediakindmedia file_id_str file_id_objr,media_file_namer"r#r$ directoryguessed_extension extension downloaders [/var/www/files/kim/lib/python3.12/site-packages/pyrogram/methods/messages/download_media.pydownload_mediazDownloadMedia.download_media sYt- gu}} -' Xt4$  X !!VWWE eS !K--KmmK0 )) !%b9E;2 E;3 ufd+!ww}}Y7 966B ww}}Y'9+L8LMI $ 4 4Y ? K'" hnn,-7 x~~x/A/A8CVCVWW-7 h///-7 h...-8 hnn,-7 & $++../44::<'112EF  I)) )Y 9hP] ^  ## #  " " $ 0 0 <$s1JH8J,J-)J) __name__ __module__ __qualname__r2rr)boolrtuplerrrPrrOr r s.!![=[=+,[=[= [=  [=  [=[= %X & '[=rVr )rBr-rtypingrrrrpyrogramrpyrogram.file_idr r r r2r rrVrOrZs4& 66::#\=\=rV