o
Ďij @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl
Z
d dlmZmZmZ d dlmZ d dlmZ d dlmZ G dd dZdS ) N)datetime timedeltatimezone) KaggleClient)KaggleCredentials)ExchangeOAuthTokenRequestc @ s e Zd ZdZdefddZG dd dZG dd dejj Z
e d d
Z d e
d ee
d
dfddZd ee
dedefddZdd ZdefddZd d ee
dedefdd Zd S ) KaggleOAuth kagglesdkclientc C s, || _ | | _d| _d | _tt| _d S NF) _client http_client _http_client_server_running_credslogging getLogger__name___logger)selfr
r R/home/jeff/fluffinator/venv/lib/python3.10/site-packages/kagglesdk/kaggle_oauth.py__init__ s
zKaggleOAuth.__init__c @ sJ e Zd Zdd ZddefddZddedefd d
Zd edefd d
ZdS )KaggleOAuth.OAuthStatec C s. t t | _tj | _tj| j| _ d S N)
struuiduuid4stater
OAuthState_generate_code_verifier
code_verifier_generate_code_challengecode_challenger r r r r s z KaggleOAuth.OAuthState.__init__ lengthc C s
t | S r )secrets
token_urlsafer& r r r _generate_state#
z&KaggleOAuth.OAuthState._generate_state@ returnc C s. d| kr
dkst d t dt| S )N* z;Code verifier length must be between 42 and 128 characters.)
ValueErrorr' r( r) r r r r & s
z.KaggleOAuth.OAuthState._generate_code_verifierr! c C s, | d}t| }t|d}|S )Nzutf-8)encodehashlibsha256digestbase64urlsafe_b64encodedecode)r! code_verifier_bytescode_challenge_bytescode_challenge_base64r r r r" + s
z/KaggleOAuth.OAuthState._generate_code_challengeN)r% )r, ) r
__module__ __qualname__r intr* r r r" r r r r r s
r c sH e Zd Zddddd d fddZd d
Zd d Zd
d Z ZS )z KaggleOAuth.OAuthCallbackHandlerNoauth oauth_state
on_successloggerr? r r@ r c s. || _ || _|| _|| _t j|i | d S r )_oauth _oauth_state _on_successr superr )r r? r@ rA rB argskwargs __class__r r r 2 s
z)KaggleOAuth.OAuthCallbackHandler.__init__c C s0 | j dkrd S z | W | d S | w )Nz /favicon.ico)path_handle_oauth_callback _stop_serverr$ r r r do_GETA s
z'KaggleOAuth.OAuthCallbackHandler.do_GETc C sD t j| j}t j|j}d|v rd|v r|d d }|d d }| jd | jd| | jd| || jj kr\|
d | dd | | j
d
| | d S | jd | jj d | |
d
| dd | | j
d d S | jd| j |
d
| dd | | j
d d S )Ncoder r z
Received OAuth Callback:z code : z state: z Content-typez text/htmls^ Authentication Successful!
You can close this window.
zInvalid state! Expected: z , Received: i s9 Authentication Failed!
z"
Received Invalid OAuth Callback: s\ Authentication Failed!
Invalid callback parameters.
)urllibparseurlparserK parse_qsqueryr debugrD r
send_response send_header end_headerswfilewriterE error)r
parsed_url query_paramsrO r r r r rL I s8
z7KaggleOAuth.OAuthCallbackHandler._handle_oauth_callbackc C s | j d S r )rC stop_serverr$ r r r rM i s z-KaggleOAuth.OAuthCallbackHandler._stop_server)r? r r@ r )r r; r< r rN rL rM
__classcell__r r rI r OAuthCallbackHandler1 s ra c C s0 t dv rdS dtjv rtjd dkrdS dS )N)WindowsDarwinTDISPLAY F)platformsystemosenvironr r r r _can_open_browserl s
z KaggleOAuth._can_open_browserrO scopesr@ r c C s` t }||_|j|_d|_| jjj|}t| j|j |j
t t
jt|jd |j|d| _d S )Nauthorization_code)seconds)r
refresh_token access_tokenaccess_token_expirationusernamerk )r rO r!
grant_typer security oauth_clientexchange_oauth_tokenr refreshToken accessTokenr nowr utcr
expires_inrq r )r rO rk r@ requestresponser r r _exchange_oauth_tokenv s z!KaggleOAuth._exchange_oauth_tokenno_launch_browserr- c
sd | ot }j }|r"tdd}d| }jd| t jj t j
|j j d}|rt
| td td| d d tffd
d fd d
}d_td|f|$}jd| d jrz| jssjd W d jS 1 sw Y jS td td| tddd t } | jS )Ni@ i(# zhttp://localhost:z!Will listen for the callback at: ) client_id redirect_uriscoper r# z&Your browser has been opened to visit:z z
rO c s | d S r )r} )rO )r@ rk r r r ru s z9KaggleOAuth._run_oauth_flow..exchange_oauth_tokenc s t j| jd|S )Nr> )r ra r )rG rH )ru r@ r r r handler_factory s z4KaggleOAuth._run_oauth_flow..handler_factoryTz 127.0.0.1z Listening for callback on port z...z)OAuth flow completed (or server stopped).zW
Go to the following link in your browser, and complete the sign-in prompts at Kaggle:
zF
Once finished, enter the verification code provided in your browser: re )end)r rj r get_oauth_default_redirect_urlrandomrandintr rV r build_start_oauth_urlOAUTH_CLIENT_IDr r#
webbrowseropenprintr r socketserver TCPServerhandle_requestinputr} r )
r rk r~ use_browserr portoauth_start_urlr httpdrO r )ru r@ rk r r _run_oauth_flow sP
zKaggleOAuth._run_oauth_flowc C s
d| _ d S r )r r$ r r r r_ r+ zKaggleOAuth.stop_servercredsc C s |st d| S )NzAuthentication failed.) Exception
introspect)r r r r r _ensure_creds_valid s z KaggleOAuth._ensure_creds_validFc C s2 | ||}| |}| td| d |S )Nz
You are now logged in as [z]
)r r saver )r rk r~ r rq r r r authenticate s
zKaggleOAuth.authenticateN)F)r r; r< r r r r httpserverBaseHTTPRequestHandlerra staticmethodrj r listr} boolr r r_ r r r r r r r s ;
5 r )r5 r2 http.serverr r rh rf r r' r r urllib.parserQ r r r r kagglesdk.kaggle_clientr kagglesdk.kaggle_credsr &kagglesdk.security.types.oauth_servicer r r r r r s"