o ni@sddlZddlZddlmZmZddlmZddlZddlm Z eej dZ dZ deded ed efd d Zdeded efd dZdeded ed efddZdeded efddZdS)N)datetime timedelta)Path)AuthenticationErrorz ~/.config/wandb/credentials.jsonz%Y-%m-%d %H:%M:%Sbase_url token_filecredentials_filereturncCs(|s t|||t|||}|dS)aRetrieve an access token from the credentials file. If no access token exists, create a new one by exchanging the identity token from the token file, and save it to the credentials file. Args: base_url (str): The base URL of the server token_file (pathlib.Path): The path to the file containing the identity token credentials_file (pathlib.Path): The path to file used to save temporary access tokens Returns: str: The access token access_token)exists_write_credentials_file_fetch_credentials)rrrdatarW/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/lib/credentials.pyr s  r cCsbt||}d||ii}t|d}tj||ddt|dWddS1s*wYdS)aTObtain an access token from the server and write it to the credentials file. Args: base_url (str): The base URL of the server token_file (pathlib.Path): The path to the file containing the identity token credentials_file (pathlib.Path): The path to file used to save temporary access tokens credentialswindentiN)_create_access_tokenopenjsondumposchmod)rrrrrfilerrrr (s  "r cCsi}t|!}t|}d|vri|d<||dvr |d|}Wdn1s*wYt}d|vr?t|dt}|tkrnt||}t|d}||d|<tj||ddWd|S1siwY|S)aFetch the access token from the credentials file. If the access token has expired, fetch a new one from the server and save it to the credentials file. Args: base_url (str): The base URL of the server token_file (pathlib.Path): The path to the file containing the identity token credentials_file (pathlib.Path): The path to file used to save temporary access tokens Returns: dict: The credentials including the access token. rN expires_atrrr) rrloadrutcnowstrptime_expires_at_fmtrr)rrrcredsrrrrrrr ;s*         r c Cszt|}|}Wdn1swYWn'ty0}ztd||d}~wtyC}ztd||d}~ww|d}d|d}ddi}tj|||d }|jd krktd |jd |j | } t t t| d d} | t| d<| d =| S)aExchange an identity token for an access token from the server. Args: base_url (str): The base URL of the server. token_file (pathlib.Path): The path to the file containing the identity token Returns: dict: The access token and its expiration. Raises: FileNotFoundError: If the token file is not found. OSError: If there is an issue reading the token file. AuthenticationError: If the server fails to provide an access token. NzIdentity token file not found: z-Failed to read the identity token from file: z /oidc/tokenz+urn:ietf:params:oauth:grant-type:jwt-bearer) grant_type assertionz Content-Typez!application/x-www-form-urlencoded)rheadersz!Failed to retrieve access token: z, expires_in)secondsr)rreadstripFileNotFoundErrorOSErrorrequestspost status_codertextrrrrfloatstrftimer!) rrrtokeneurlrr%response resp_jsonrrrrr`s>   r)rrrrpathlibrrequests.utilsr- wandb.errorsrpath expanduserDEFAULT_WANDB_CREDENTIALS_FILEr!strr r dictr rrrrrs   %