
    Pi                         d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ d dlmZmZ d dlmZ  e j        e          Zd	d
dededz  defdZdedefdZdS )    N)
HTTPStatus)ApiGetKernelRequest)registry)build_kaggle_client)KaggleApiHTTPErrorhandle_call)UtilityScriptHandleparse_utility_script_handle)EXTRA_CONSOLE_BLOCKF)force_downloadhandler   returnc                   t          |           }t                              d|                                 di t                     t          j        |d|          \  }}t          |          s-t                              d|j         d|j	         d           |S |t          j        vrt          j                            |           t                              d	| d
           |S )ap  
    Downloads the utility script and adds the directory path to the system path.

    Args:
        handle: (string) the notebook handle under https://kaggle.com/code.
        force_download: (bool) Optional flag to force download motebook output, even if it's cached.


    Returns:
        A string representing the path to the requested notebook output files.
    zDownloading Utility Script: z ...)extraN)pathr   z
Notebook 'z' by user 'z?' is not a Utility Script
 and will not be added to system pathzAdded z to system path)r
   loggerinfoto_urlr   r   notebook_output_resolver_is_notebook_utility_scriptnotebookownersysr   append)r   r   hutility_script_path_s        m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/kagglehub/utility_scripts.pyutility_script_installr      s     	$F++A
KK?qxxzz???G^J]G^K___%>qt\jkkk&q)) #6 6 6 6 6 6	
 	
 	
 #"#(**+,,,
KK=,===>>>    r   c                    	 t                      5 t                      | j        _        | j        _        t          fd          }d|j        j        v cd d d            S # 1 swxY w Y   d S # t          $ rY}|j
        K|j
        j        t          j        k    r1t                              d| j         d| j         d           Y d }~dS |d }~ww xY w)Nc                  B     j         j                                      S )N)kernelskernels_api_client
get_kernel)
api_clientrs   r   <lambda>z-_is_notebook_utility_script.<locals>.<lambda>5   s    :+=+P+[+[\]+^+^ r    zutility scriptzCould not find 'z' metadata by user 'z'.F)r   r   r   	user_namer   kernel_slugr   metadatacategory_idsr   responsestatus_coder   	NOT_FOUNDr   r   )r   r-   er&   r'   s      @@r   r   r   /   sE    "" 	Fj#%%A'AKJAM"#^#^#^#^#^__H#x'8'EE	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F    :!aj&<
@T&T&TKKV17VV
VVVWWW55555IsB   A3 AA&A3 &A**A3 -A*.A3 3
C=ACCC)loggingr   httpr   +kagglesdk.kernels.types.kernels_api_servicer   	kagglehubr   kagglehub.clientsr   kagglehub.exceptionsr   r   kagglehub.handler	   r
   kagglehub.loggerr   	getLogger__name__r   strboolr   r    r    r   <module>r>      s    



       K K K K K K       1 1 1 1 1 1 @ @ @ @ @ @ @ @ M M M M M M M M 0 0 0 0 0 0		8	$	$ JO   3 4$; SV    >#6 4      r    