o xiS( @sdZddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z ddl mZmZdd lmZdddddd defd:ddZd;ddZdddddd d d!dd" dd3d4Zd?d6d7Zd@d8d9ZdS)AzZLog in to Weights & Biases. This authenticates your machine to log data to your account. ) annotationsN)AuthenticationErrorterm) wandb_setup) settings_filewbauth)UNSETDoNotSet) InternalApiFkey str | Nonerelogin bool | Nonehostforcetimeout int | Noneverifyboolreferrer anonymousr returnc Cs|tur tjdddtjdurtddStj}|jr!dS|j r.|j s.tddS|r5| d}t | |dt|||||||pGd d \} } | S) aLog into W&B. You generally don't have to use this because most W&B methods that need authentication can log in implicitly. This is the programmatic counterpart to the `wandb login` CLI. This updates global credentials for the session (affecting all wandb usage in the current Python process after this call) and possibly the .netrc file. If the identity_token_file setting is set, like through the WANDB_IDENTITY_TOKEN_FILE environment variable, then this is a no-op. Otherwise, if an explicit API key is provided, it is used and written to the system .netrc file. If no key is provided, but the session is already authenticated, then the session key is used for verification (if verify is True) and the .netrc file is not updated. If none of the above is true, then this gets the API key from the first of: - The WANDB_API_KEY environment variable - The api_key setting in a system or workspace settings file - The .netrc file (either ~/.netrc, ~/_netrc or the path specified by the NETRC environment variable) - An interactive prompt (if available) Args: key: The API key to use. relogin: If true, get the API key from an interactive prompt, skipping reading .netrc, environment variables, etc. host: The W&B server URL to connect to. force: If true, disallows selecting offline mode in the interactive prompt. timeout: Number of seconds to wait for user input in the interactive prompt. This can be used as a failsafe if an interactive prompt is incorrectly shown in a non-interactive environment. verify: Verify the credentials with the W&B server and raise an AuthenticationError on failure. referrer: The referrer to use in the URL login request for analytics. Returns: bool: If `key` is configured. Raises: AuthenticationError: If `api_key` fails verification with the server. UsageError: If `api_key` cannot be configured and no tty. z^The anonymous parameter to wandb.login() has no effect and will be removed in future versions.FrepeatNz7Calling wandb.login() after wandb.init() has no effect.z'Unable to verify login in offline mode./rmodels)r rrrrrr)rrtermwarnwandbrunr singletonsettings_noop_offlinex_cli_only_moderstrip_update_system_settingsread_system_settings_login) r rrrrrrrglobal_settings logged_in_r-I/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/wandb_login.pylogins<8        r/system_settingssettings_file.SettingsFilesNonec Cs|jddd|r|dkr|jdddn|jd|ddz|WdStjy?}ztt|WYd}~dSd}~ww)z(Update the user's system settings files.rT)globallyzhttps://api.wandb.aibase_urlN)clearsetsaverSaveSettingsErrorrrstr)r0rer-r-r.r'qsr'rT) r rrrrrrupdate_api_key_silent float | Noner9r;r<tuple[bool, str | None]c Cstj} |durtj| j| jd} nt|} |dur| j}|dur&| j}|dur-| j }|dur4| j }t j rEt j sEtddS|rQt|| | ||d} n t| | ||||d} |rlt| tjrlt| j| jjdt|s|t| t| d| durdSt| tjrd | jfSd S) aYLog in to W&B. Arguments are the same as for wandb.login() with the following additions: Args: update_api_key: If true and an explicit API key is given, it will be saved to the .netrc file. _silent: If true, will not print any messages to the console. Returns: A pair (is_successful, key). N)app_urlzQTo use W&B in kaggle you must enable internet in the settings panel on the right.)FN)rr"r;silent)rrrr input_timeout)r r4rT)TN)rr!r"rHostUrlr4r?rr login_timeoutr@rutil _is_kaggle _has_internetr termerror_use_explicit_key_find_or_prompt_for_key isinstance AuthApiKey _verify_loginapi_keyrurlupdate_user_settings_print_logged_in_messager9) r rrrrrrr;r<r"host_urlauthr-r-r.r)sV      r)r"wandb.Settingswbauth.HostUrlr@ wbauth.Authc Cs|jr |s tdtj||d}tj|dd|rCz tj|jj|j dW|Stj yB}zt t |WYd}~|Sd}~ww|S)zDLog in with an explicit key. Same arguments as `_login()`. zIf you're specifying your api key in code, ensure this code is not shared publicly. Consider setting the WANDB_API_KEY environment variable, or running `wandb login` from the command line.)rrM wandb.login())sourceN) _notebookrrrrKuse_explicit_authwrite_netrc_authrrNrMWriteNetrcErrorrr9)r r"rr;r@rRr:r-r-r.rHs& rHrAwbauth.Auth | Nonec Csld}d}ztj|d|||||d}Wn tyd}Ynw|s4|r,tdd|_|Stdd |_|S) zGLog in without an explicit key. Same arguments as `_login()`. FNrV)rrW no_offline no_createrrArTz"W&B disabled due to login timeout.disabledzUsing W&B in offline mode.offline)rauthenticate_session TimeoutErrorrrmodetermlog)r"rrrrrA timed_outrRr-r-r.rIs.    rIr4c Csddlm}t|d|id}z|}Wn%|y)}z td|d|d}~wty9}ztd|d}~ww|sFtd|d d dS) Nr)ConnectionErrorr4)rMdefault_settingszUnable to connect to z to verify API token.z.An error occurred while verifying the API key.z%API key verification failed for host .z! Make sure your API key is valid.)requests.exceptionsrfr validate_api_keyr Exception)r r4rfapiis_api_key_validr:r-r-r.rL"s:    rLc Cst}|}|rG|rdtj|ddnd}|jp|}d}|r8||vr8||kr8dtj|ddd}dtj|dd||}nd }d tjd d d d}tj |d|dddS)z4Print a message telling the user they are logged in.z to green)fgz (yellow)zCurrently logged in as: zW&B API key is configuredzUse z`wandb login --relogin`T)boldz to force reloginz. FrN) rr! _get_usernameclickstyleentity _get_entity _get_teamsrrd) r"rr!usernamehost_strrw entity_strlogin_state_strlogin_info_strr-r-r.rP<s   rP)r r rrrr rrrrrrrr rr rr)r0r1rr rr2)r r rrrr rrrr=rrrr9r;rr<rrr>) r r9r"rSrrTr;rr@rrrU)r"rSrrTrrrrrr9rAr=rr\)r r9r4r9rr2)r"rSrr9rr2)__doc__ __future__rrur wandb.errorsrr wandb.sdkr wandb.sdk.librrwandb.sdk.lib.deprecationrr apisr r/r'r)rHrIrLrPr-r-r-r.sB    ^ P # )