>xj#ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZmZmZddlmZejeZd ZGd d eZGd d eZGddeZGddeZGddeZdededdfdZdedefdZdejdefdZ efdejde!de"de"de!f dZ#de d edede!de"de"d!e!defd"Z$dS)#N) BaseModel) RichToolkit)Progress) APIClient)Settings) AuthConfigAuthModewrite_auth_config)FastAPIRichToolkiti,cFeZdZUeed<eed<eed<eed<dZeed<dS)AuthorizationData user_code device_codeverification_uriverification_uri_completeintervalN)__name__ __module__ __qualname__str__annotations__rintQC:\Projects\Historical\venv\Lib\site-packages\fastapi_cloud_cli/commands/_flow.pyr r sONNN""""Hcrr ceZdZUeed<dS) TokenResponse access_tokenN)rrrrrrrrrrsrrc$eZdZUeed<eed<dS) LoginOutput authenticated auth_modeN)rrrboolrr rrrr!r!s*rr!cBeZdZUeed<eed<eed<eed<eed<dS)DeviceAuthorizationOutputrrrrrN)rrrrrrrrrr&r&$sF""""NNNMMMMMrr&ceZdZdS)LoginTimeoutErrorN)rrrrrrr(r(,sDrr(datatoolkitreturnc0|ddS)NuNow you are logged in! 🚀)print)r)r*s rrender_login_outputr.0s MM/00000rauthorization_datac\t|j|j|j|j|jS)N)rrrrr)r&rrrrr)r/s rdevice_authorization_outputr14s9 %+<"4"N$.&2#,    rclientctj}|dd|ji}td|j|t |j S)Nz/login/device/authorization client_idr)z+Device authorization response status code: ) rgetpostr4loggerdebug status_coderaise_for_statusr model_validate_jsontext)r2settingsresponses rstart_device_authorizationr@@s||~~H{{%[(:L,MH LLUx?SUUVVV   0 0 ? ??rrrtimeoutctj}tj}td |d||jdd}td|j|jdvr| |jd kr`| }|d }td ||d kr| |jd krnj|tj|z z } | dkrtt|| } td| dtj | @t|j} td| jS)Nz!Starting to poll for access tokenTz/login/device/tokenz,urn:ietf:params:oauth:grant-type:device_code)rr4 grant_typer5zToken response status code: )rEerrorzToken response error: authorization_pendingrDrz Sleeping for z seconds before retrying...z#Access token received successfully.)rr6time monotonicr8r9r7r4r:r;jsonr(minsleeprr<r=r) r2rrrAr>startr?r)rF remaining sleep_for response_datas rfetch_access_tokenrQOs |~~H N  E LL4555 ;; !*%/L    JH4HJJKKK  z 1 1  % % ' ' '  3 & &==??DHHW%%E LL9%99 : : :///))+++  3 & & t~//%78 >># #),,  KYKKKLLL 9A D"55hmDDM LL6777  %%rprogress cancel_hintc |||5t||||}dddn #1swxYwYnO#t$rd}|d|dYn)t$rd}|d||YnwxYwt t |td d S) N)r*z/Login timed out before authorization completed.rAz(Try again with a longer --timeout value.)hintz/Login cancelled before authorization completed. cancelled)rTuser)r"r#)handle_http_errorsrQr(failKeyboardInterruptr rr!) r2rRr*rrrArSrmessages rcomplete_device_loginr\s]  & &x & A A V V-fk8WUUL V V V V V V V V V V V V V V V    C  ;         C         jl;;;<<< TV < < <rjs $$$$$$******++++++------PPPPPPPPPP::::::  8 $ $ # I)          1k1K1D1111 )      @ L @ @ @ @ @&1 /&/& L/&/&/& /&  /&/&/&/&d= == =  =  =========r