o
Ďi
@ s d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ eeZdZd Zd
Zed eejddf fd d
Zd efddZded dfddZddded e dB fddZ d ded dfddZ ddded e fddZ d e dB fd d Z!dS ) N) Generator)contextmanager)ApiGetModelRequest)build_kaggle_client)get_kaggle_credentialsset_kaggle_api_tokenset_kaggle_credentials)UnauthenticatedErrori z 
Create an API token from your Kaggle
settings page and paste it below.
z
Thank Youreturnc c sB t } t| }t| z | V W t| dS t| w )zCapture output that is logged using the logger.
Example:
```py
>>> output = capture_logger_output()
>>> logger.info("This is an info message")
>>> logger.error("This is an error message")
>>> print(output)
```
N)ioStringIOlogging
StreamHandler_logger
addHandler
removeHandler)bufferhandler r J/home/jeff/fluffinator/venv/lib/python3.10/site-packages/kagglehub/auth.py_capture_logger_output s
r c C sR z ddl m} | j}|jD ] }|jdkr W dS qW dS ttfy( Y dS w )zReturn `True` if code is executed in a notebook (Jupyter, Kaggle Notebook, Colab, QTconsole).
Taken from https://stackoverflow.com/a/39662359.
Adapted to make it work with Google colab as well.
r ) get_ipythonZMQInteractiveShellTF)IPythonr __class____mro____name__ NameErrorModuleNotFoundError)r shell_class parent_classr r r _is_in_notebook5 s
r! validate_credentialsc s d}zddl m} ddlm W n ty t|w jddddd }jd
d jd d }j t
| t g|d
| dt ddf fdd }|
| dS )zRPrompt the user for their Kaggle token and save it in a widget (Jupyter or Colab).z;You need the `ipywidgets` module: `pip install ipywidgets`.r )displaywidgetsflexcolumncenterz50%)r# flex_flow align_itemswidthz
API Token:) descriptionLogin)layout_r
Nc
s j }d_ dg _z$t }t| du r t W d n1 s'w Y | }W n tyG } z
t|}W Y d }~nd }~ww fdd| dD _d S )N z
Connecting...Tc s g | ] }| r |qS r )stripLabel).0liner$ r r
s s zB_notebook_login..on_click_login_button..
)
valuer2 childrenr r _validate_credentials_helpergetvalue Exceptionstrsplit)r/ tokencapturedmessageerrorlogin_token_widget token_widgetr" r% r r on_click_login_button` s z._notebook_login..on_click_login_button)IPython.displayr#
ipywidgetsr% ImportErrorLayoutPasswordButtonVBoxHTML NOTEBOOK_LOGIN_TOKEN_HTML_START NOTEBOOK_LOGIN_TOKEN_HTML_ENDr< on_click)r"
library_errorr#
box_layout login_buttonrE r rB r _notebook_loginG s* rT TverboserV c C s t Q}| }| |ddt }|j| }d|v r2| r't d |d W d S d|v rD|d t
krD| rCt d n| rKt d W d d S 1 sWw Y d S )Nzapi.v1.DiagnosticsServiceHellouserNamez*Kaggle credentials successfully validated.codezInvalid Kaggle credentials. You can obtain a Kaggle API token on your [Kaggle settings page](https://www.kaggle.com/settings/account).z3Unable to validate Kaggle credentials at this time.)
r http_client
_init_session_prepare_requestr _sessionsendjsonr infoINVALID_CREDENTIALS_ERRORrA warning)rV
api_clientrZ requestresponser r r r9 x s(
$r9 c C s6 t r t| dS td}t| | sdS t dS )z@Prompt the user for their Kaggle API token and save it globally.Nz9Enter your Kaggle API token (input will not be visible): )r! rT getpassr r9 )r" r> r r r login s
rg c
C sL t st zt| d}|rd|iW S t ty% } zt |d}~ww )zv
Return a dictionary with the username of the authenticated Kaggle user or raise an error if unauthenticated.
rU usernameN)r r r9 r; )rV rh er r r whoami s
rj c C s& zt ddd W S ty Y dS w )zWReturns the username of the authenticated logged-in user if configured, otherwise None.FrU rh N)rj r; r r r r get_username s
rk )T)"rf r r
collections.abcr
contextlibr (kagglesdk.models.types.model_api_servicer kagglehub.clientsr kagglehub.configr r r kagglehub.exceptionsr getLoggerr r ra rN rO r r boolr! rT r< r9 rg dictrj rk r r r r s*
1