o -j&!@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlZddlmZddlmZddlmZeeZGdd d ZdS) N)md5)PurePath)UnionBinaryIOCallable)StopTransmission)raw)Sessionc @s@eZdZ    d dddeeefdeded ed ef d d Z dS)SaveFileNrselfzpyrogram.Clientpathfile_id file_partprogress progress_argsc s$j4IdH{|dur WdIdHdSfddd}t|ttfr/t|d}n t|tjr8|}ntdt|dd}| d tj | } | d | d krZtd j j r`d nd } | | d d krrtd| dtt| |} | dk} | rdnd} |du}|p}| s|stnd}tjIdHjIdHjIdHddfddt| D}tdzzIdH| || ||}|s| s|sddd|D}n| rtj j!j"||| |d}n tj j!j#|||d}$|IdH|rHWW|D] }$dIdHqtj%|IdH&IdHt|ttfr=|'WdIdHdS| sS|sS|(||d7}|rt)j*|t+||| | g|R}t,-|rw|IdHn j./j0|IdHqWn t1yt2y}z t34|WYd}~nd}~ww| rtj5j6|| |dW|D] }$dIdHqtj%|IdH&IdHt|ttfr|'WdIdHStj5j7|| ||dW|D] }$dIdHqtj%|IdH&IdHt|ttfr|'WdIdHSW|D] }$dIdHq&tj%|IdH&IdHt|ttfrM|'n+|D] }$dIdHqPtj%|IdH&IdHt|ttfrx|'wwWdIdHdS1IdHswYdS)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 sd IdH}|durdSz ||IdHWnty0}z t|WYd}~nd}~wwq)N)getinvoke Exceptionlog exception)sessiondatae)queuer `/var/www/kodo/Anonymous/send/lib/python3.10/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)is_mediacsg|] }jqSr )loop 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/r md5_checksum)8save_file_semaphore isinstancestrropenioIOBase ValueErrorgetattrseekSEEK_ENDtellme is_premiumintmathceilrnd_idrr storagedc_idauth_key test_moderangeasyncioQueuestartreadjoindigestr functionsuploadSaveBigFilePart SaveFilePartputgatherstopcloseupdate functoolspartialmininspectiscoroutinefunctionr"run_in_executorexecutorrrrrtypes InputFileBig InputFile)r r rrrr part_sizefp file_name file_sizefile_size_limit_mibr,is_big workers_countis_missing_partmd5_sumworkerschunkrpcr%funcrr )rr rrr save_file's :          ' T   &rh 0zSaveFile.save_file)NrNr ) __name__ __module__ __qualname__rr3rr>rtuplermr r r rr &s$ r )rGrVrYr5loggingr?hashlibrpathlibrtypingrrrpyrogramrrpyrogram.sessionr getLoggerrnrr r r r rs