B fd @sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlZddlmZddlmZddlmZeeZGdd d ZdS) N)md5)PurePath)UnionBinaryIOCallable)StopTransmission)raw)Sessionc@s.eZdZddeeefeeeedddZ dS) SaveFileNrzpyrogram.Client)selfpathfile_id file_partprogress progress_argsc s~j4IdH^|dkrdSfddd}t|ttfrHt|d}nt|tjrZ|}ntdt|dd}| d t j | } | d | d krtd j jrd nd } | | d d krtd| dtt| |} | dk} | rdnd} |dk }|p}| s|stnd}tjIdHjIdHjIdHddfddt| D}tdzyIdH| ||x||}|s| s|sddd|D}P| rt j!j"j#||| |d}nt j!j"j$|||d}%|IdH|r"dS| s8|s8|&||d7}|rt'j(|t)||| | f|}t*+|r~|IdHnj,-j.|IdHqWWnDt/k rYnZt0k r}zt12|Wdd}~XYn.X| rt j3j4|| |dSt j3j5|| ||dSWdx|D]}%dIdHqWtj6|IdH7IdHt|ttfrh|8XWdQIdHRXdS)a Upload a file onto Telegram servers, without actually sending the message to anyone. Useful whenever an InputFile type is required. .. note:: This is a utility method intended to be used **only** when working with raw :obj:`functions ` (i.e: a Telegram API method you wish to use which is not available yet in the Client class as an easy-to-use method). .. include:: /_includes/usable-by/users-bots.rst Parameters: path (``str`` | ``BinaryIO``): The path of the file you want to upload that exists on your local machine or a binary file-like object with its attribute ".name" set for in-memory uploads. file_id (``int``, *optional*): In case a file part expired, pass the file_id and the file_part to retry uploading that specific chunk. file_part (``int``, *optional*): In case a file part expired, pass the file_id and the file_part to retry uploading that specific chunk. 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: ``InputFile``: On success, the uploaded file is returned in form of an InputFile object. Raises: RPCError: In case of a Telegram RPC error. Nc sfx`IdH}|dkrdSy||IdHWqtk r\}zt|Wdd}~XYqXqWdS)N)getZinvoke Exceptionlog exception)sessiondatae)queuer G/tmp/pip-unpacked-wheel-rcokkf2l/pyrogram/methods/advanced/save_file.pyworkeresz"SaveFile.save_file..workerirbzPInvalid file. Expected a file path as string or a binary (not text) file pointernamezfile.jpgrzFile size equals to 0 BiiizCan't upload files bigger than z MiBiT)Zis_mediacsg|]}jqSr )loopZ create_task).0_)r rrr r sz&SaveFile.save_file..cSs"g|]}t|dddqS)N)hexzfill)r!ir r rr#s)rrfile_total_partsbytes)rrr*)idpartsr)r+r,rZ md5_checksum)9Zsave_file_semaphore isinstancestrropenioIOBase ValueErrorgetattrseekosSEEK_ENDtellmeZ is_premiumintmathceilZrnd_idrr ZstorageZdc_idZauth_keyZ test_moderangeasyncioQueuestartreadjoindigestrZ functionsuploadZSaveBigFilePartZ SaveFilePartputupdate functoolspartialmininspectiscoroutinefunctionr Zrun_in_executorexecutorrrrrtypesZ InputFileBigZ InputFileZgatherstopclose)r r rrrrZ part_sizefp file_name file_sizeZfile_size_limit_mibr)Zis_bigZ workers_countZis_missing_partZmd5_sumworkerschunkZrpcfuncrr"r )rr rrr save_file's:                 zSaveFile.save_file)NrNr ) __name__ __module__ __qualname__rr.rr9rtuplerUr r r rr &sr )r=rFrIr0loggingr:r5hashlibrpathlibrtypingrrrZpyrogramrrZpyrogram.sessionr getLoggerrVrr r r r rs