^jVddlZddlZddlmZddlmZddlZddlmZddlm Z Gdd e eZ ed Gd d Z ddejde dzde dzfdZe dddde de de de dzde dzde f dZGddZeZdZdee e fdee e ffdZdZy)N) dataclass)Enum) constants)validate_hf_hub_argsceZdZdZdZy) XetTokenTypereadwriteN)__name__ __module__ __qualname__READWRITEe/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/huggingface_hub/utils/_xet.pyr r s D Err T)frozenc"eZdZUeed<eed<y) XetFileData file_hash refresh_routeN)r rrstr__annotations__rrrrrs Nrrresponseendpointreturnc|y |jtj}tj|jvr!|jtjd}n|jtj }||ntj}|jtjr>|jtjjd|jd}t||S#t $rYywxYw)a^ Parse XET file metadata from an HTTP response. This function extracts XET file metadata from the HTTP headers or HTTP links of a given response object. If the required metadata is not found, it returns `None`. Args: response (`httpx.Response`): The HTTP response object containing headers dict and links dict to extract the XET metadata from. Returns: `Optional[XetFileData]`: An instance of `XetFileData` containing the file hash and refresh route if the metadata is found. Returns `None` if the required metadata is missing. Nurl/)rr) headersrHUGGINGFACE_HEADER_X_XET_HASH$HUGGINGFACE_HEADER_LINK_XET_AUTH_KEYlinks&HUGGINGFACE_HEADER_X_XET_REFRESH_ROUTEKeyErrorENDPOINT startswithHUGGINGFACE_CO_URL_HOMEreplacerstripr)rrrrs r!parse_xet_file_data_from_responser-s$$Y%L%LM  9 9X^^ K$NN9+Y+YZ[`aM$,,Y-]-]^M$/xY5G5GH A AB%--i.O.O.V.VWZ.[]e]l]lmp]qr #  sA7C:: DD)revisionr token_typerepo_id repo_typer.c|||ntj}|d|d|d|jd}|dk7s||d|z }|S)a Build the URL used to fetch or refresh a Xet access token for a given repo. Args: token_type (`XetTokenType`): Type of the token to request: `"read"` or `"write"`. repo_id (`str`): A namespace (user or an organization) and a repo name separated by a `/`. repo_type (`str`): Type of the repo (e.g. `"model"`, `"dataset"`, `"space"`, `"bucket"`). revision (`str`, `optional`): The revision of the repo to get the token for. endpoint (`str`, `optional`): The endpoint to use for the request. Defaults to the Hub endpoint. Returns: `str`: The fully-qualified URL of the token refresh endpoint. z/api/zs/z/xet-z-tokenbucketr!)rr(value)r/r0r1r.rr s rxet_connection_info_refresh_urlr5:s_4$/xY5G5GH JeI;b z7G7G6H OCH 4 8*~ Jrc"eZdZdZdZdZdZy)XetSessionHoldera7Holds an optional XetSession; supports safe re-creation after sigint_abort or fork. Thread-safe: a ``threading.Lock`` guards all state mutations, which matters for free-threaded Python (3.14t) where multiple threads can race on ``get()`` or ``sigint_abort()`` without the GIL serialising them. cRtj|_d|_d|_yN) threadingLock_lock_session _session_pidselfs r__init__zXetSessionHolder.__init__gs^^%  (,rc|j5tj}|j|j|k7rd|_|jddlm}||_||_|jcdddS#1swYyxYw)aReturn the current session, creating one if needed. Fork-safe: if the current process PID differs from the PID that created the session (i.e. we are in a forked child), the old session is discarded and a fresh session is created for this process. Nr) XetSession)r<osgetpidr=r>hf_xetrC)r@ current_pidrCs rgetzXetSessionHolder.getlspZZ !))+K}}(T->->+-M!% }}$- *  $/!== ! ! !s A'A>>Bc|j5|j) |jjd|_d|_dddy#t$rY"wxYw#1swYyxYw)zMAbort the current session and clear it so the next get() creates a fresh one.N)r<r= sigint_abort Exceptionr>r?s rrJzXetSessionHolder.sigint_abortsg ZZ )}}(MM..0!% $(! ) )!  ) )s- AA A AAAAA$N)r rr__doc__rArHrJrrrr7r7_s- !0 )rr7c*tjS)a*Return the global :class:`hf_xet.XetSession`, creating it on first call. The session is shared across all calls within a process, just as the HTTP client returned by :func:`~huggingface_hub.utils._http.get_session` is shared. It is created lazily and is fork-safe and thread-safe. )_GLOBAL_XET_HOLDERrHrrrget_xet_sessionrOs  ! ! ##rr"c~|jDcic]\}}|jdk7s||c}}Scc}}w)zReturn a copy of headers with the authorization header removed. Xet storage requests use a short-lived xet access token for auth, so the Hub authorization header must not be forwarded to xet storage endpoints. authorization)itemslower)r"keyr4s rxet_headers_without_authrUs3 *1 [:3CIIK?r^s ! "3  $  #PT* `knr`r F   !!! ! Dj ! Dj ! !!H.).)b&'$\d38n\c3h\&r