
    LMh/                     j   d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZmZmZmZ d dlmZ d d	lmZ d d
lmZ dZdZdZdZ e j        e          Z G d dee                   Z  G d dee                   Z! G d dee                   Z" G d dee                   Z#dS )    N)Optional)DEFAULT_CONNECT_TIMEOUTKaggleJwtClient)is_kaggle_cache_disabled)is_in_kaggle_notebook)BackendError)CompetitionHandleDatasetHandleModelHandleNotebookHandle)EXTRA_CONSOLE_BLOCK)PackageScope)ResolverKAGGLE_CACHE_MOUNT_FOLDERAttachDatasourceUsingJwtRequest   z/kaggle/inputc                   p    e Zd ZdefdZ	 d
dddedee         dee         deeee	         f         fd	Z
dS )CompetitionKaggleCacheResolverreturnc                 F    t                      rdS t                      rdS dS NFTr   r   self___s      _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/kagglehub/kaggle_cache_resolver.pyis_supportedz+CompetitionKaggleCacheResolver.is_supported!   s-    #%% 	5 "" 	4u    NFforce_downloadhpathr!   c                   t                      }|r#t                              di t                     d|j        i}|                    t          d|it          t          f          }d|vrd}t          |          t          j        t          t                    }| d|d          }	t          j                            |	          st                              d	|	 d
           nT|r+t                              d| d| di t                     n't                              d| di t                     t          j                            |	          s3t!          j        d           t          j                            |	          3|rB|	 d| }
t          j                            |
          sd| d|	 d}t%          |          |
d fS |	d fS )NWIgnoring `force_download` argument when running inside the Kaggle notebook environment.extraCompetitionSlugcompetitionReftimeout	mountSlug.'result.mountSlug' field missing from response/Mounting files to ...Attaching 'z' from competition '' to your Kaggle notebook...zAttaching competition '   'zf' is not present in the competition files.You can acces the other files othe attached competition at ')r   loggerinfor   competitionpostATTACH_DATASOURCE_REQUEST_NAMEr   ATTACH_DATASOURCE_READ_TIMEOUTr   osgetenv&KAGGLE_CACHE_MOUNT_FOLDER_ENV_VAR_NAME!DEFAULT_KAGGLE_CACHE_MOUNT_FOLDERr#   existstimesleep
ValueError)r   r"   r#   r!   clientcompetition_refresultmsgbase_mount_pathcached_pathcached_filepaths              r   _resolvez'CompetitionKaggleCacheResolver._resolve(   s=    !"" 	KKi-,-     q}
 * / -.LM  
 
 f$$BCs###)$JLmnn(@@6++>@@w~~k** 	KK=[===>>>> 		KKWdWWWWW-,-     
 KKI!III-,-     '..-- 	JqMMM '..-- 	 	)!,55t55O7>>/22 &b b bS^b b b  !oo%"D((D  r   N)__name__
__module____qualname__boolr   r	   r   strtupleintrJ    r   r   r   r       s             ;?3!di3! 3! 3!"3!*23-3!S[\`Sa3!	sHSM!	"3! 3! 3! 3! 3! 3!r   r   c                   p    e Zd ZdefdZ	 d
dddedee         dee         deeee	         f         fd	Z
dS )DatasetKaggleCacheResolverr   c                 F    t                      rdS t                      rdS dS r   r   r   s      r   r   z'DatasetKaggleCacheResolver.is_supported_   -    #%% 	5 "" 	4ur   NFr    r"   r#   r!   c                   |r#t                               di t                     t                      }|j        |j        d}|                                rt          |j                  |d<   n(t          j
        |          }|t          |          |d<   |                    t          d|it          t          f          }d|vrd}t          |          t!          j        t$          t&                    }	|	 d	|d          }
|                    d
          }t           j                            |
          st                               d|
 d           nT|r+t                               d| d| di t                     n't                               d| di t                     t           j                            |
          s3t/          j        d           t           j                            |
          3|rB|
 d	| }t           j                            |          sd| d|
 d}t3          |          ||fS |
|fS )Nr%   r&   )	OwnerSlugDatasetSlugVersionNumber
datasetRefr*   r,   r-   r.   versionNumberr/   r0   r1   z' from dataset 'r2   zAttaching dataset 'r3   r4   z^' is not present in the dataset files.You can acces the other files othe attached dataset at ')r5   r6   r   r   ownerdatasetis_versionedrP   versionr   get_versionr8   r9   r   r:   r   r;   r<   r=   r>   getr#   r?   r@   rA   rB   )r   r"   r#   r!   rC   dataset_refversion_from_package_scoperE   rF   rG   rH   ra   rI   s                r   rJ   z#DatasetKaggleCacheResolver._resolveh   s     	KKi-,-     !""9
 
 >> 	O+.qy>>K(( *6)A!)D)D&)5/23M/N/NO,*k -.LM  
 
 f$$BCs###)$JLmnn(@@6++>@@**_--w~~k** 	KK=[===>>>> 		KKSdSSASSS-,-     
 KKEaEEE-,-    
 '..-- 	JqMMM '..-- 	  	,!,55t55O7>>/22 &^ ^ ^OZ^ ^ ^  !oo%"G++G##r   rK   )rL   rM   rN   rO   r   r
   r   rP   rQ   rR   rJ   rS   r   r   rU   rU   ^   s             7;>$`e>$ >$ >$>$&.sm>$OWX\~>$	sHSM!	">$ >$ >$ >$ >$ >$r   rU   c                   p    e Zd ZdefdZ	 d
dddedee         dee         deeee	         f         fd	Z
dS )ModelKaggleCacheResolverr   c                 F    t                      rdS t                      rdS dS r   r   r   s      r   r   z%ModelKaggleCacheResolver.is_supported   rW   r   NFr    r"   r#   r!   c                   |r#t                               di t                     t                      }|j        |j        |j        |j        d}|                                rt          |j
                  |d<   n(t          j        |          }|t          |          |d<   |                    t          d|it          t           f          }d|vrd}t#          |          t%          j        t(          t*                    }	|	 d	|d          }
|                    d
          }t$          j                            |
          st                               d|
 d           nT|r+t                               d| d| di t                     n't                               d| di t                     t$          j                            |
          s3t3          j        d           t$          j                            |
          3|rB|
 d	| }t$          j                            |          sd| d|
 d}t7          |          ||fS |
|fS )Nr%   r&   )rY   	ModelSlug	FrameworkInstanceSlugr[   modelRefr*   r,   r-   r.   r]   r/   r0   r1   z' from model 'r2   Attaching model 'r3   r4   z^' is not present in the model files. You can access the other files of the attached model at ')r5   r6   r   r   r^   model	framework	variationr`   rP   ra   r   rb   r8   r9   r   r:   r   r;   r<   r=   r>   rc   r#   r?   r@   rA   rB   )r   r"   r#   r!   rC   	model_refre   rE   rF   rG   rH   ra   rI   s                r   rJ   z!ModelKaggleCacheResolver._resolve   s     	KKi-,-     !""K	
 
	 >> 	M),QYIo&& *6)A!)D)D&)5-01K-L-L	/**I -.LM  
 
 f$$BCs###)$JLmnn(@@6++>@@**_--w~~k** 	KK=[===>>>> 		KKQdQQ!QQQ-,-     
 KKCACCC-,-    
 '..-- 	JqMMM '..-- 	  	,!,55t55O7>>/22 &_ _ _P[_ _ _  !oo%"G++G##r   rK   )rL   rM   rN   rO   r   r   r   rP   rQ   rR   rJ   rS   r   r   rg   rg      s             59@$^c@$ @$ @$@$$,SM@$MUVZ^@$	sHSM!	"@$ @$ @$ @$ @$ @$r   rg   c                   p    e Zd ZdefdZ	 d
dddedee         dee         deeee	         f         fd	Z
dS )!NotebookOutputKaggleCacheResolverr   c                 F    t                      rdS t                      rdS dS r   r   r   s      r   r   z.NotebookOutputKaggleCacheResolver.is_supported   rW   r   NFr    r"   r#   r!   c                   |r#t                               di t                     t                      }|j        |j        d}|                                rt          |j                  |d<   n(t          j
        |          }|t          |          |d<   |                    t          d|it          t          f          }d|vrd}t          |          t!          j        t$          t&                    }	|	 d	|d          }
|                    d
          }t           j                            |
          st                               d|
 d           nT|r+t                               d| d| di t                     n't                               d| di t                     t           j                            |
          s3t/          j        d           t           j                            |
          3|rB|
 d	| }t           j                            |          sd| d|
 d}t3          |          ||fS |
|fS )Nr%   r&   )rY   
KernelSlugr[   	kernelRefr*   r,   r-   r.   r]   r/   r0   r1   z' from Notebook 'r2   rn   r3   r4   zj' is not present in the notebook output files.You can access the other files of the attached notebook at ')r5   r6   r   r   r^   notebookr`   rP   ra   r   rb   r8   r9   r   r:   r   r;   r<   r=   r>   rc   r#   r?   r@   rA   rB   )r   r"   r#   r!   rC   
kernel_refre   rE   rF   rG   rH   ra   rI   s                r   rJ   z*NotebookOutputKaggleCacheResolver._resolve   s     	KKi-,-     !""*
 

 >> 	N*-ai..J'' *6)A!)D)D&)5.12L.M.M
?+*Z -.LM  
 
 f$$BCs###)$JLmnn(@@6++>@@**_--w~~k** 	KK=[===>>>> 		KKTdTTQTTT-,-     
 KKCACCC-,-    
 '..-- 	JqMMM '..-- 	  	,!,55t55O7>>/22 &b b bS^b b b  !oo%"G++G##r   rK   )rL   rM   rN   rO   r   r   r   rP   rQ   rR   rJ   rS   r   r   rt   rt      s             8<?$af?$ ?$ ?$?$'/}?$PXY]P^?$	sHSM!	"?$ ?$ ?$ ?$ ?$ ?$r   rt   )$loggingr;   r@   typingr   kagglehub.clientsr   r   kagglehub.configr   kagglehub.envr   kagglehub.exceptionsr   kagglehub.handler	   r
   r   r   kagglehub.loggerr   kagglehub.packagesr   kagglehub.resolverr   r=   r9   r:   r>   	getLoggerrL   r5   r   rU   rg   rt   rS   r   r   <module>r      s    				               6 5 5 5 5 5 / / / / / / - - - - - - Z Z Z Z Z Z Z Z Z Z Z Z 0 0 0 0 0 0 + + + + + + ' ' ' ' ' ')D &!B !# 
 %4 ! 
	8	$	$;! ;! ;! ;! ;!X.?%@ ;! ;! ;!|H$ H$ H$ H$ H$-!8 H$ H$ H$VJ$ J$ J$ J$ J$x4 J$ J$ J$ZI$ I$ I$ I$ I$(@ I$ I$ I$ I$ I$r   