nij 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 d d ZdS ) N)datetime timedeltatimezone) KaggleClient)KaggleCredentials)ExchangeOAuthTokenRequestc e Zd ZdZdefdZ G d d Z G d dej j Z
e d Z d e
d
ee
d d fd
Zd
ee
dedefdZd ZdefdZdd
ee
dedefdZdS ) KaggleOAuth kagglesdkclientc || _ | | _ d| _ d | _ t j t | _ d S NF) _client http_client _http_client_server_running_credslogging getLogger__name___logger)selfr s S/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/kagglesdk/kaggle_oauth.py__init__zKaggleOAuth.__init__ sB " . . 0 0 $ (22 c D e Zd Zd Zd defdZd dedefdZdedefd Zd
S )
KaggleOAuth.OAuthStatec t t j | _ t j | _ t j | j | _ d S N)
struuiduuid4stater
OAuthState_generate_code_verifier
code_verifier_generate_code_challengecode_challenger s r r z KaggleOAuth.OAuthState.__init__ sQ TZ\\**DJ!,!7!O!O!Q!QD "-"8"Q"QRVRd"e"eD r lengthc * t j | S r )secrets
token_urlsafer+ s r _generate_statez&KaggleOAuth.OAuthState._generate_state# s (00 0r @ returnc f d| cxk rdk sn t d t j | S )N* z;Code verifier length must be between 42 and 128 characters.)
ValueErrorr- r. r/ s r r% z.KaggleOAuth.OAuthState._generate_code_verifier& sA &&&&3&&&& !^___(00 0r r& c | d }t j | }t j | d }|S )Nzutf-8)encodehashlibsha256digestbase64urlsafe_b64encodedecode)r& code_verifier_bytescode_challenge_bytescode_challenge_base64s r r' z/KaggleOAuth.OAuthState._generate_code_challenge+ sZ "/"6"6w"?"? #*>2E#F#F#M#M#O#O $*$<=Q$R$R$Y$YZa$b$b !( (r N)r* )r1 ) r
__module__ __qualname__r intr0 r r% r' r r r$ r s f f f
1 1C 1 1 1 1 1 1C 1 1 1 1 1
)C )C ) ) ) ) ) )r r$ c D e Zd Zddddd d fdZd Zd Zd
Z xZS ) KaggleOAuth.OAuthCallbackHandlerNoauth oauth_state
on_successloggerrI r
rJ r c r || _ || _ || _ || _ t j |i | d S r )_oauth _oauth_state _on_successr superr )r rI rJ rK rL argskwargs __class__s r r z)KaggleOAuth.OAuthCallbackHandler.__init__2 sE DK +D )D !DL EGG d -f - - - - -r c | j dk rd S | | d S # | w xY w)Nz /favicon.ico)path_handle_oauth_callback _stop_serverr) s r do_GETz'KaggleOAuth.OAuthCallbackHandler.do_GETA s[ yN**
$++---!!#####!!####s 9 Ac t j | j }t j |j }d|v rzd|v ru|d d }|d d }| j d | j d| | j d| || j j k rp|
d | dd | | j
d
| | d S | j d | j j d | |
d
| dd | | j
d d S | j d| 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.
)urllibparseurlparserV parse_qsqueryr debugrO r#
send_response send_header end_headerswfilewriterP error)r
parsed_url query_paramsr[ r# s r rW z7KaggleOAuth.OAuthCallbackHandler._handle_oauth_callbackI sJ ..ty99J!<001ABBL %%'\*A*A#F+A.$W-a0 ""#@AAA ""#5t#5#5666 ""#6u#6#6777D - 333&&s+++$$^[AAA$$&&&J$$y $$T*****L&&'o4CTCZ'o'ohm'o'oppp&&s+++$$^[AAA$$&&&J$$%abbbbb ""#T#T#TUUU""3''' === """
s r c 8 | j d S r )rN stop_serverr) s r rX z-KaggleOAuth.OAuthCallbackHandler._stop_serveri s K # # % % % % %r )rI r
rJ r )r rB rC r rY rW rX
__classcell__)rT s @r OAuthCallbackHandlerrG 1 s $(48
.
. !
. 2
.
.
.
.
.
. $ $ $ @ & & & & & & &r rn c t j dv rdS dt j v rt j d dk rdS dS )N)WindowsDarwinTDISPLAY F)platformsystemosenvironrE r r _can_open_browserz KaggleOAuth._can_open_browserl sC ? 5 5 54
" "rz)'<'B'B4ur r[ scopesrJ r c ^ t }||_ |j |_ d|_ | j j j | }t | j |j |j
t j t j t |j z |j | | _ d S )Nauthorization_code)seconds)r
refresh_token access_tokenaccess_token_expirationusernamery )r r[ r&
grant_typer security oauth_clientexchange_oauth_tokenr refreshToken accessTokenr nowr utcr
expires_inr r )r r[ ry rJ requestresponses r _exchange_oauth_tokenz!KaggleOAuth._exchange_oauth_tokenv s +-- + 9 1<(5JJ7SS'<"/!-$,L $>$>S[SfAgAgAg$g &
r no_launch_browserr2 c
| ot } j }|r7t j dd }d| } j d| t j t j
| j j }|rt j | t d t d| d d t f fd
fd }d _ t% j d
|f| 5 } j d| d j r| j j d d d d n # 1 swxY w Y nWt d t d| t dd t+ } | j S )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
r[ c 6 | d S r )r )r[ rJ ry r s r r z9KaggleOAuth._run_oauth_flow..exchange_oauth_token s! **4EEEEEr c : t j | j d|S )NrH )r
rn r )rR rS r rJ r s r handler_factoryz4KaggleOAuth._run_oauth_flow..handler_factory s8 "7 + 3 <
r Tz 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: rs )end)r
rx r get_oauth_default_redirect_urlrandomrandintr rb r$ build_start_oauth_urlOAUTH_CLIENT_IDr# r(
webbrowseropenprintr r socketserver TCPServerhandle_requestinputr r ) r ry r use_browserr portoauth_start_urlr httpdr[ r rJ s `` @@r _run_oauth_flowzKaggleOAuth._run_oauth_flow s ++O 0M0M0O0O (GGII S>$--D5t55L L Q< Q Q R R R!,,.. +AA!1% #&5 B
B OO , , , : ; ; ; ,,,, - - -
F3
F
F
F
F
F
F
F
F
$(D 'd(;_MM
PQV ""#NT#N#N#NOOO* +((*** * + ""#NOOO
P
P
P
P
P
P
P
P
P
P
P
P
P
P
P
m n n n ((( ) ) ) Y
77D & &tV[ A A A{s +AFFFc d| _ d S r )r r) s r rl zKaggleOAuth.stop_server s $ r credsc L |st d | S )NzAuthentication failed.) Exception
introspect)r r s r _ensure_creds_validz KaggleOAuth._ensure_creds_valid s+ 6 455 5 !!!r Fc | || }| | }| t d| d |S )Nz
You are now logged in as [z]
)r r saver )r ry r r r s r authenticatezKaggleOAuth.authenticate sW $$V->??++E22
:X:::;;; r N)F)r rB rC r r r r$ httpserverBaseHTTPRequestHandlerrn staticmethodrx r listr boolr r rl r r rE r r r
r
sb !O3| 3 3 3 3) ) ) ) ) ) ) )(9& 9& 9& 9& 9&t{ A 9& 9& 9&v \
#
tCy
Og
3d3i 3D 3M^ 3 3 3 3j% % %"): " " " "
49 Rc r r
)r< r9 http.serverr r rv rt r r- r r! urllib.parser] r r r r kagglesdk.kaggle_clientr kagglesdk.kaggle_credsr &kagglesdk.security.types.oauth_servicer r
rE r r r s
2 2 2 2 2 2 2 2 2 2 0 0 0 0 0 0 4 4 4 4 4 4 L L L L L Lu u u u u u u u u ur