o -j@srddlZddlZddlmZddlmZmZmZmZddlZddlm Z ddl m Z m Z m Z dZGdddZdS) N)datetime)UnionOptionalCallableBinaryIO)types)FileIdFileType PHOTO_TYPESz downloads/c@sXeZdZeddddfdddedd efd ed ed ed edede eee ffddZ dS) DownloadMediaFTNselfzpyrogram.Clientmessagez types.Messagez types.Story file_name in_memoryblockprogress progress_argsreturnc sd}t|tjst|tjr%|D]}t||d} | durnqtdn|} t| tr/| } n| j} t | } | j } t| dd} t| dd}t| dd}t| d d}t j |\}}|p_| p_d}|rvt j |}|d d}|rt|d vrvd}t j |s|j|pt}|s||}| tvrd }n9| tjkr|pd }n/| tjtjtjfvr|pd}n | tjkr|pd}n| tjkr|pd}n | tjkr|pd}nd}dt| j j|pt !"d|#|}|$| ||||||f}|r|IdHSt%&'|dS)aFDownload the media from a message. .. include:: /_includes/usable-by/users-bots.rst Parameters: message (:obj:`~pyrogram.types.Message` | :obj:`~pyrogram.types.Story` | ``str``): Pass a Message/Story 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_typedate).z..z.jpgz.oggz.mp4z.zipz.webpz.mp3z.unknownz {}_{}_{}{}z%Y-%m-%d_%H-%M-%S)( isinstancerMessageStorygetattr ValueErrorstrfile_idrdecode file_typeospathsplitbasenamereplaceisabs PARENT_DIRDEFAULT_DOWNLOAD_DIRguess_extensionr r VOICEVIDEO ANIMATION VIDEO_NOTEDOCUMENTSTICKERAUDIOformatnamelowerrnowstrftimernd_idhandle_downloadasyncioget_event_loop create_task)r rrrrrravailable_mediakindmedia file_id_str file_id_objr,media_file_namerr r! directoryguessed_extension extension downloaderr r e/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/pyrogram/methods/messages/download_media.pydownload_media!spZ                       zDownloadMedia.download_media) __name__ __module__ __qualname__r4rr)boolrtuplerrrRr r r rQr s. r )rDr-rtypingrrrrpyrogramrpyrogram.file_idrr r r4r r r r rQs