
    niy                         d dl Z d dlZd dlmZ d dlmZ dZdZdZd Z	 G d d	e          Z
e
j        d
e
j        de
j        de
j        de
j        de
j        diZde
fdZd ZdefdZd Zd ZdS )    N)Enum)PathKAGGLE_KERNEL_RUN_TYPEKAGGLE_DATA_PROXY_URLKAGGLE_API_V1_TOKENc                  4    t          j        t                    S N)logging	getLogger__name__     Q/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/kagglesdk/kaggle_env.py
get_loggerr      s    X&&&r   c                   &    e Zd ZdZdZdZdZdZdZdS )	KaggleEnvr                  N)	r   
__module____qualname__LOCALSTAGINGADMINQAPRODTESTr   r   r   r   r      s,        EGE	
BDDDDr   r   zhttp://localhostzhttps://staging.kaggle.comzhttps://admin.kaggle.comzhttps://qa.kaggle.comzhttps://api.kaggle.comzhttp://localhost:7777envc                     t           |          S r	   )_env_to_endpointr    s    r   get_endpointr$   "   s    C  r   c                  ,   t          j        d          } | | dk    rt          j        S | dk    rt          j        S | dk    rt          j        S | dk    rt          j        S | dk    rt          j        S | dk    rt          j        S t          d|  d	          )
NKAGGLE_API_ENVIRONMENTr   	LOCALHOSTr   r   r   r   z/Unrecognized value in KAGGLE_API_ENVIRONMENT: "")
osgetenvr   r   r   r   r   r   r   	Exceptionr#   s    r   get_envr,   &   s    
),
-
-C
{cVmm~
k
g~~
i  
d{{|
f}}~
LcLLL
M
MMr   returnc                      t          j        t                    Gt          j        t                    ,t	                                          dt           d           dS dS dS )Nz!Can't use the Kaggle Cache. The 'z"' environment variable is not set.FT)r)   r*   KAGGLE_NOTEBOOK_ENV_VAR_NAME"KAGGLE_DATA_PROXY_URL_ENV_VAR_NAMEr   warningr   r   r   is_in_kaggle_notebookr2   7   sl    	y-..:9788@LL  _:_ _ _   5t5r   c                     | sd S t          |           }|                                sd S |                                                                }|sd S t	                                          d|  d           |S )NzUsing access token from file: "r(   )r   exists	read_textstripr   debug)path
token_pathtoken_values      r   _get_access_token_from_filer;   D   s     tdJ t&&((..00K tLL@@@@AAAr   c                  R   t                      r<t          t          j                            t
                              } | r	| t
          fS t          j                            d          }|rWt          |                                          rt          |          dfS t                      	                    d           |dfS t          t          j
                            d                    }|r|dfS t          t          j
                            d                    }|r|dfS dS )NKAGGLE_API_TOKENz=Using access token from KAGGLE_API_TOKEN environment variablez~/.kaggle/access_tokenaccess_tokenz~/.kaggle/access_token.txt)NN)r2   r;   r)   environgetKAGGLE_API_V1_TOKEN_PATHr   r4   r   r7   r8   
expanduser)tokenr>   s     r   get_access_token_from_envrD   T   s#    5+BJNN;S,T,TUU 	5344:>>"455L 2$$&& 	S/==?QRRK	
 	
 	
 011.rw/A/ABZ/[/[\\L .n-- /rw/A/AB^/_/_``L .n--<r   )r
   r)   enumr   pathlibr   r/   r0   rA   r   r   r   r   r   r   r   r   r"   r$   r,   boolr2   r;   rD   r   r   r   <module>rH      s1    				            7 %< "0 ' ' '        O'3O/L)N,N+ !i ! ! ! !N N N"
t 
 
 
 
       r   