o
    Ďiy                     @   s   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d Z	G d	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dZdd ZdefddZdd Zdd ZdS )    N)Enum)PathKAGGLE_KERNEL_RUN_TYPEKAGGLE_DATA_PROXY_URLKAGGLE_API_V1_TOKENc                   C   s
   t tS N)logging	getLogger__name__ r   r   P/home/jeff/fluffinator/venv/lib/python3.10/site-packages/kagglesdk/kaggle_env.py
get_logger   s   
r   c                   @   s$   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    r   zhttp://localhostzhttps://staging.kaggle.comzhttps://admin.kaggle.comzhttps://qa.kaggle.comzhttps://api.kaggle.comzhttp://localhost:7777envc                 C   s   t |  S r   )_env_to_endpointr   r   r   r   get_endpoint"   s   r   c                  C   sv   t d} | d u 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 | dkr3t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   r   r   r   get_env&   s   
r&   returnc                   C   s<   t td urt td u r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   r   is_in_kaggle_notebook7   s   r+   c                 C   sH   | 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_valuer   r   r   _get_access_token_from_fileD   s   r3   c                  C   s   t  rttjt} | r| tfS tjd}|r/t| r%t|dfS t 	d |dfS ttj
d}|r=|dfS ttj
d}|rK|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)r+   r3   r#   environgetKAGGLE_API_V1_TOKEN_PATHr   r,   r   r/   r0   
expanduser)tokenr5   r   r   r   get_access_token_from_envT   s&   r;   )r   r#   enumr   pathlibr   r(   r)   r8   r   r   r   r   r   r   r   r   r   r   r&   boolr+   r3   r;   r   r   r   r   <module>   s*    

