U ;g @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 sj4IdH|dkr.W5QIdHRdSfddd}t|ttfrXt|d}nt|tjrj|}ntdt|dd}| d t j | } | d | d krtd j jrd nd } | | d d krtd| dtt| |} | dk} | rdnd} |dk }|p}| s.|s.tnd}tjIdHjIdHjIdHddfddt| D}tdz֐z2 IdH| |||!|}|s| s|sd"dd|#D}q| rt$j%j&j'||| |d}nt$j%j&j(|||d}|IdH|rLWW.W5QIdHRdS| sb|sb|)||d7}|rt*j+|t,||| | f|}t-.|r|IdHnj/0j1|IdHqWnDt2k rYnt3k r}zt45|W5d}~XYnZX| r8t$j6j7|| |dWBW5QIdHRSt$j6j8|| ||dWW5QIdHRSW5|D]}dIdHqjtj|IdHIdHt|ttfr|XW5QIdHRXdS)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 sbIdH}|dkrdSz||IdHWqtk rZ}zt|W5d}~XYqXqdS)N)getZinvoke Exceptionlog exception)sessiondatae)queuer c/opt/bot/vods_filmes/script/venv/lib/python3.8/site-packages/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_moderangeasyncioQueueputZgatherstopclosestartreadjoindigestrZ functionsuploadZSaveBigFilePartZ SaveFilePartupdate functoolspartialmininspectiscoroutinefunctionr Zrun_in_executorexecutorrrrrtypesZ InputFileBigZ InputFile)r r rrrrZ part_sizefp file_name file_sizeZfile_size_limit_mibr)Zis_bigZ workers_countZis_missing_partZmd5_sumworkersr"chunkZrpcfuncrr )rr rrr save_file's:             zSaveFile.save_file)NrNr ) __name__ __module__ __qualname__rr.rr9rtuplerUr r r rr &s r )r=rHrKr0loggingr:r5hashlibrpathlibrtypingrrrZpyrogramrrZpyrogram.sessionr getLoggerrVrr r r r rs