B fdj@srddlZddlZddlmZddlmZmZmZmZddlZddlm Z ddl m Z m Z m Z dZGdddZdS) N)datetime)UnionOptionalCallableBinaryIO)types)FileIdFileType PHOTO_TYPESz downloads/c @sHeZdZeddddfdedefeeeeee eee fddd Z dS) DownloadMediaFTNzpyrogram.Clientz types.Message)selfmessage file_name in_memoryblockprogress progress_argsreturnc sd}t|tjr>x0|D]}t||d} | dk rPqWtdn|} t| trR| } n| j} t| } | j } t| dd} t| dd}t| dd}t| d d}t j |\}}|p| pd}t j |s|j|pt}|s||}| tkrd }n| tjkr|pd }nn| tjtjtjfkr$|p d }nL| tjkr<|p8d }n4| tjkrT|pPd}n| tjkrl|phd}nd}dt| j j|ptd| |}|!| ||||||f}|r|IdHSt"#$|dS)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()) ) audiodocumentZphotoZstickerZ animationvideoZvoiceZ video_noteZnew_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 ValueErrorstrZfile_idrdecode file_typeospathsplitisabsZ PARENT_DIRDEFAULT_DOWNLOAD_DIRguess_extensionr r ZVOICEZVIDEOZ ANIMATIONZ VIDEO_NOTEZDOCUMENTZSTICKERZAUDIOformatnamelowerrnowstrftimeZrnd_idZhandle_downloadasyncioZget_event_loopZ create_task)r rrrrrrZavailable_mediakindZmediaZ file_id_strZ file_id_objr"Zmedia_file_namerrr directoryZguessed_extension extension downloaderr r L/tmp/pip-unpacked-wheel-rcokkf2l/pyrogram/methods/messages/download_media.pydownload_media s\Z                       zDownloadMedia.download_media) __name__ __module__ __qualname__r'rr boolrtuplerrr4r r r r3r s r )r.r#rtypingrrrrZpyrogramrZpyrogram.file_idrr r r'r r r r r3s