o -wiY+@sdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z mZddlmZddlZddlmZmZddlZddlmZdd lmZdd lmZdd lmZdd lmZm Z m!Z!e rpdd l"m#Z#dZ$dZ%dZ&dZ'dZ(e$e%e&e'gZ)ej*ddGdddZ+Gddde,Z-edZ.e ej/dddZ0dEddZ1dFd!d"Z2dGdHd&d'Z3    ( ( ( dIdJd5d6Z4dKd8d9Z5dLd=d>Z6 dGdMd@dAZ7dGdNdCdDZ8dS)Oz apikey util.) annotationsN)partial) TYPE_CHECKINGCallableLiteral)urlparse) NETRC_FILESget_netrc_auth) InternalApi)term) url_registry) wandb_setup)_is_databricksisattyprompt_choices)Settingsz*Private W&B dashboard, no account requiredzCreate a W&B accountzUse an existing W&B accountzDon't visualize my results UnconfiguredT)frozenc@s&eZdZUded<ded<ded<dS)_NetrcPermissionsboolexists read_access write_accessN)__name__ __module__ __qualname____annotations__rrQ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/lib/apikey.pyr+s  rc@seZdZdZdS)WriteNetrcErrorz.Raised when we cannot write to the netrc file.N)rrr__doc__rrrrr2sr)allowmustneverfalsetrue) hide_inputerrdefault Mode | NonereturncCs|pd}ddd}|||S)Nr#r!)r%r$)get)r( anon_modemappingrrr_fixup_anon_mode<s  r.strcCstjd}|rtj|StD]}tjd}tjtj||r,tj||Sqt dkr5dnd}tjtjd|S)z"Return the path to the netrc file.NETRC~Windowsz.netrc_netrc) osenvironr+path expanduserrrjoinplatformsystem) netrc_filehome_dirrrrget_netrc_file_pathDs   r=app_urlreferrer str | NonecCs"d}|r d|}d|d|S)aGenerate a prompt string for API key authorization. Creates a URL string that directs users to the authorization page where they can find their API key. Args: app_url: The base URL of the W&B application. referrer: Optional referrer parameter to include in the URL. Returns: A formatted string with instructions and the authorization URL. z?ref=z0You can find your API key in your browser here: z /authorizer)r>r?refrrr_api_key_prompt_strWs  rCFsettingsrapiInternalApi | Noneinput_callbackCallable | Nonebrowser_callback no_offliner no_createlocalstr | bool | NonecCsl|pt}tj}|p t|}t|j} |jpd} |j} ddtD} | dkr*| t | r/|j r1|r6| t | r;|j r=|rB| t | rXdtjvrXtj}tj| } | durX| S| dkr_t }n*| skttjrnttjrntrqt}n|rvt}nt| dkr| d }nt| |j | d }d} | r|d n|d }|t kr|} n|t kr|r|d dnd} | s|rd|nd}td| d|||} nc|tkr|r|nd} | s|js|s| }dD]}| |r| t|d}qtd|dt!"ddtt#| |||} n|tkrdS|t kr dS| r)|r)|nd\} }| s4t$d| S)zPrompt for api key. Returns: str - if key is configured None - if dryrun is selected False - if unconfigured (notty) FcSsg|]}|qSrr).0choicerrr sz"prompt_api_key..r#z google.colabNr"r) input_timeoutjupyterz2: Paste an API key from your profile and hit enterzK: Paste an API key from your profile and hit enter, or press ctrl+c to quitT)signupz&ref=rAzCreate an account here: z/authorize?signup=true)zhttp://zhttps://z Logging into z-. (Learn how to deploy a W&B server locally: z wandb-server))NFNo API key specified.)%getpassr LOG_STRINGr r. anonymous_jupyterr> LOGIN_CHOICESremoveLOGIN_CHOICE_ANON login_timeoutLOGIN_CHOICE_DRYRUNLOGIN_CHOICE_NEWsysmodulesLOG_STRING_NOCOLORwandbrSattempt_colab_loginrstdoutstdinrLOGIN_CHOICE_NOTTYLOGIN_CHOICE_EXISTSlenrcreate_anonymous_api_keytermlogstripis_local startswithr urlrC ValueError)rDrErGrIrJrKrLr? log_stringr,rSr>choiceskeyresultapi_askrBhostprefixrYrrrprompt_api_keyjs                 ry netrc_pathc Csd}d}d}zt|}d}t|jtj@}t|jtj@}Wn)ty,d}d}YntyH}zt d|d|WYd}~nd}~wwt |||dS)z1Check if we can read and write to the netrc file.FTzUnable to read permissions for z, N)rrr) r4statrst_modeS_IWUSRS_IRUSRFileNotFoundErrorOSErrorrd termerrorr)rz file_existsrrsterrrcheck_netrc_accesss(  "rrwentityrtc Csd|vr |ddnd|f\}}t|dkr#td|dt|dt|j}t}t|}|jr5|js=t d|d d |}d } zt |} | d } Wd n1s]wYWn!t yqtd Ynty} zt d|| d } ~ wwz^t |dD} | rd} | D]} | d ks|| vrd} q| r| d8} q| | d qtd|d|| tdj|||dWd n1swYt|tjtjBWd Sty} zt d|| d } ~ ww)zAdd our host and key to .netrc.-rQrA(z,API-key must be exactly 40 characters long: z (z chars)Cannot access z. In order to persist your API key, grant read and write permissions for your user to the file or specify a different file with the environment variable "NETRC=".zmachine N z"No netrc file found, creating one.zUnable to read wrzAppending key for z to your netrc file: z machine {host} login {entity} password {key} )rwrrtzUnable to write )splitrjrqrnetlocr=rrrropenreadrmrrdrlrwritetextwrapdedentformatr4chmodr{r~r})rwrrt_ key_suffixnormalized_hostrz netrc_access machine_line orig_linesfrskiplinerrr write_netrcsf          rNonecCsf|std|p t}d|vr|ddnd|f\}}t|dkr*tdt|t|jd|dS)NrVrrQrArz.API key must be 40 characters long, yours was user)rqr rrjrbase_url)rDrtrErsuffixrrr write_key)s   rSettings | NonecCsl|dur tj}|jr|jStt}|jr&|js&t dtddS|jr4t |j }|r4|dSdS)Nr.) r singletonrDapi_keyrr=rrrdtermwarnr r)rDrauthrrrr>s    r)r(r)r*r))r*r/)N)r>r/r?r@r*r/)NNNFFFN)rDrrErFrGrHrIrHrJrrKrrLrr?r@r*rM)rzr/r*r)rwr/rr/rtr/)rDrrtr@rErFr*r)rDrr*r@)9r __future__r dataclassesr4r9r{rar functoolsrtypingrrr urllib.parserclickrequests.utilsrr rd wandb.apisr wandb.errorsr wandb.errors.linksr wandb.sdkr wandb.utilrrrwandb.sdk.wandb_settingsrr]r`rir_rhr[ dataclassr ExceptionrModepromptrWr.r=rCryrrrrrrrrsh             h @