tni dZddlZddlmZddlZejdZd de de d dfd Z dde fd Z dd Z dS)z runpod | cli | config.py A collection of functions to set and validate configurations. Configurations are TOML files located under ~/.runpod/ N)Pathz~/.runpod/config.tomldefaultFapi_keyprofilereturnc<tjtjtdt td|sSttd5}|tj |vrtd dddn #1swxYwYttdd5}| d |zd z| d |zd zddddS#1swxYwYdS) aG Sets the user's credentials in ~/.runpod/config.toml If profile already exists user must use `update_credentials` instead. Args: api_key (str): The user's API key. profile (str): The profile to set the credentials for. --- File Structure --- [default] api_key = "RUNPOD_API_KEY" T)exist_okrbz9Profile already exists. Use `update_credentials` instead.NwzUTF-8)encoding[z] z api_key = "z" ) osmakedirspathdirnameCREDENTIAL_FILErtouchopentomlload ValueErrorwrite)rr overwrite cred_files _/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/runpod/cli/groups/config/functions.pyset_credentialsrsK004@@@@...  /4 ( ( I$)I.... O/                osW 5 5 59g -... /%7888999999999999999999s$8&B++B/2B/ 7DDDcDtjtsdS t td5}t j|}dddn #1swxYwY||vrdd|dfSd||vrdd|dfSn#ttf$rYd SwxYwd S) z= Checks if the credentials file exists and is valid. )Fz%~/.runpod/config.toml does not exist.r NFz!~/.runpod/config.toml is missing z profile.rz-~/.runpod/config.toml is missing api_key for )Fz/~/.runpod/config.toml is not a valid TOML file.)TN) rrexistsrrrr TypeErrorr)rrconfigs rcheck_credentialsr!-s  7>>/ * *>==H /4 ( ( *IYy))F * * * * * * * * * * * * * * * & PgPPPP P F7O + +RRRR  , z "HHHGGGH :s:BA BA""B%A"&B5BBBctjtsdSt td5}t j|}dddn #1swxYwY||vrdS||S)zV Returns the credentials for the specified profile from ~/.runpod/config.toml Nr )rrrrrrr)rr credentialss rget_credentialsr$Hs 7>>/ * *t ot $ $+ i ** +++++++++++++++k!!t w sAA!$A!)rF)r) __doc__rpathlibrtomlirr expanduserrstrrr!r$rr,s '$$%<==99S939PT9999:s6       r+