ei=dZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZmZe rdd lmZGd d eZdS)z?This module contains an object that represents a Telegram File.N) b64decode)Path) TYPE_CHECKINGBinaryIO)decrypt)TelegramObject) DEFAULT_NONE) is_local_file) FilePathInputJSONDictODVInput)FileCredentialsceZdZdZdZ ddddedededzdedzd edzf fd Zd efd Z d e d e fdZ de e e e dde dzdeedeedeedeed ef dZe e e e ddedeedeedeedeed df dZ de e e e dd edzdeedeedeedeed ef dZddZxZS)Filea This object represents a file ready to be downloaded. The file can be e.g. downloaded with :attr:`download_to_drive`. It is guaranteed that the link will be valid for at least 1 hour. When the link expires, a new one can be requested by calling :meth:`telegram.Bot.get_file`. Objects of this class are comparable in terms of equality. Two objects of this class are considered equal, if their :attr:`file_unique_id` is equal. .. versionchanged:: 20.0 ``download`` was split into :meth:`download_to_drive` and :meth:`download_to_memory`. Note: * Maximum file size to download is :tg-const:`telegram.constants.FileSizeLimit.FILESIZE_DOWNLOAD`. * If you obtain an instance of this class from :attr:`telegram.PassportFile.get_file`, then it will automatically be decrypted as it downloads when you call e.g. :meth:`download_to_drive`. Args: file_id (:obj:`str`): Identifier for this file, which can be used to download or reuse the file. file_unique_id (:obj:`str`): Unique identifier for this file, which is supposed to be the same over time and for different bots. Can't be used to download or reuse the file. file_size (:obj:`int`, optional): File size in bytes, if known. file_path (:obj:`str`, optional): File path. Use e.g. :meth:`download_to_drive` to get the file. Attributes: file_id (:obj:`str`): Identifier for this file, which can be used to download or reuse the file. file_unique_id (:obj:`str`): Unique identifier for this file, which is supposed to be the same over time and for different bots. Can't be used to download or reuse the file. file_size (:obj:`int`): Optional. File size in bytes, if known. file_path (:obj:`str`): Optional. File path. Use e.g. :meth:`download_to_drive` to get the file. ) _credentialsfile_id file_path file_sizefile_unique_idN api_kwargsrrrrrct|t||_t||_||_||_d|_|jf|_| dS)Nr) super__init__strrrrrr _id_attrs_freeze)selfrrrrr __class__s X/home/div/.hermes/hermes-agent/venv/lib/python3.11/site-packages/telegram/_files/file.pyrz File.__init__Uss J/// LL #&~#6#6%.%.48-/ returnc tjt|j}tjtj|j|jtj|j |j |j S)zPConvert any UTF-8 char in :obj:`File.file_path` into a url encoded ASCII string.) urllib_parseurlsplitrr urlunsplit SplitResultschemenetlocquotepathqueryfragment)rsress r _get_encoded_urlzFile._get_encoded_urlmsc$S%8%899&  $ T[,*` .. versionchanged:: 20.0 * :paramref:`custom_path` parameter now also accepts :class:`pathlib.Path` as argument. * Returns :class:`pathlib.Path` object in cases where previously a :obj:`str` was returned. * This method was previously called ``download``. It was split into :meth:`download_to_drive` and :meth:`download_to_memory`. .. versionchanged:: 21.7 Raises :exc:`RuntimeError` if :attr:`file_path` is not set. Note that files without a :attr:`file_path` could never be downloaded, as this attribute is mandatory for that operation. Args: custom_path (:class:`pathlib.Path` | :obj:`str` , optional): The path where the file will be saved to. If not specified, will be saved in the current working directory with :attr:`file_path` as file name or the :attr:`file_id` if :attr:`file_path` is not set. Keyword Args: read_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.read_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. write_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.write_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. connect_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.connect_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. pool_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.pool_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. Returns: :class:`pathlib.Path`: Returns the Path object the file was downloaded to. Raises: RuntimeError: If :attr:`file_path` is not set. 9No `file_path` available for this file. Can not download.Nz /decrypted_r6)r RuntimeErrorr r/rrr5 read_bytesrparentname write_bytesshutilcopyfileget_botrequestretrieve) rr;r7r8r9r: local_fileurlfile_to_decryptr0r+filenames r download_to_drivezFile.download_to_driveysD~ \Z[[ ["4>22  =ddd&;&;&=&=  $+ "4>22O''(B(B(D(DEEC&K((C 677-G/J^^__   S ! ! !K  "z " ODNC ,<,< = = = $$ $  7K((HH  7'' 'D00566HLLNN*33 %'+% 4           -'',,CS!!!r!outcK|jstdt|j}|rdn|}|rt |jnd}|r|} n7|j|||||d{V} |j r| | } | | dS)a Download this file into memory. :paramref:`out` needs to be supplied with a :obj:`io.BufferedIOBase`, the file contents will be saved to that object using the :obj:`out.write` method. .. seealso:: :wiki:`Working with Files and Media ` Hint: If you want to immediately read the data from ``out`` after calling this method, you should call ``out.seek(0)`` first. See also :meth:`io.IOBase.seek`. .. versionadded:: 20.0 .. versionchanged:: 21.7 Raises :exc:`RuntimeError` if :attr:`file_path` is not set. Note that files without a :attr:`file_path` could never be downloaded, as this attribute is mandatory for that operation. Args: out (:obj:`io.BufferedIOBase`): A file-like object. Must be opened for writing in binary mode. Keyword Args: read_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.read_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. write_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.write_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. connect_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.connect_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. pool_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.pool_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. Raises: RuntimeError: If :attr:`file_path` is not set. r=Nr6) rr>r r/rr?rErFrGrr5write) rrMr7r8r9r:rHrIr+r0s r download_to_memoryzFile.download_to_memorys`~ \Z[[ ["4>22  =ddd&;&;&=&='1;tDN###t  //##CC .77)+ /) 8C   -'',,C #r!cK|jstd|t}t|jr't |j}nI|j| ||||d{V}|j r)| | |n| ||S)a Download this file and return it as a bytearray. .. versionchanged:: 21.7 Raises :exc:`RuntimeError` if :attr:`file_path` is not set. Note that files without a :attr:`file_path` could never be downloaded, as this attribute is mandatory for that operation. Args: buf (:obj:`bytearray`, optional): Extend the given bytearray with the downloaded data. Keyword Args: read_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.read_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. .. versionadded:: 20.0 write_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.write_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. .. versionadded:: 20.0 connect_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.connect_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. .. versionadded:: 20.0 pool_timeout (:obj:`float` | :obj:`None`, optional): Value to pass to :paramref:`telegram.request.BaseRequest.post.pool_timeout`. Defaults to :attr:`~telegram.request.BaseRequest.DEFAULT_NONE`. .. versionadded:: 20.0 Returns: :obj:`bytearray`: The same object as :paramref:`buf` if it was specified. Otherwise a newly allocated :obj:`bytearray`. Raises: RuntimeError: If :attr:`file_path` is not set. r=Nr6) rr> bytearrayr rr?rErFrGr/rextendr5)rr0r7r8r9r: bytes_datas r download_as_bytearrayzFile.download_as_bytearray'sb~ \Z[[ [ ;++C  ( ( dn--88::JJ#||~~5>>%%'')+ /) ?  J   # JJt,,Z88 9 9 9 9 JJz " " " r! credentialsrc||_dS)zSets the passport credentials for the file. Args: credentials (:class:`telegram.FileCredentials`): The credentials. N)r)rrVs r set_credentialszFile.set_credentialsns (r!)NNr2)rVrr"N)__name__ __module__ __qualname____doc__ __slots__rintr rr/bytesr5r r r floatrrLrrPrRrUrX __classcell__)rs@r rr%s%%NI!% $ '+:  : tO0 #    dEdedddd -1h)5)5+7(4hhh"T)huo h   h "% huoh hhhh\)5)5+7(4BBB Buo B   B "% BuoB BBBBL!%E)5)5+7(4EEE  Euo E   E "% EuoE EEEEN((((((((r!r)r\rC urllib.parseparser$base64rpathlibrtypingrrtelegram._passport.credentialsrtelegram._telegramobjectrtelegram._utils.defaultvaluer telegram._utils.filesr telegram._utils.typesr r r telegramrrr!r rns#&FE ######********222222333333555555//////CCCCCCCCCC)((((((O(O(O(O(O(>O(O(O(O(O(r!