Axj` dZddlZddlZddlmZddlZddlZej dZ d de de d dfd Z dde fd Z dd ZdS)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_keyprofilereturnctjt}tj|dt tdttdd5} |}| rtj |ntj }n,#tj j$r}td|d}~wwxYw dddn #1swxYwY|s||vrtdd |i||<t!j|d \}} tj|d d5} tj|| dddn #1swxYwYtj| tdS#t*$rtj| wxYw) a: Sets the user's credentials in ~/.runpod/config.toml If profile already exists user must pass overwrite=True. 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_okrzUTF-8)encoding/~/.runpod/config.toml is not a valid TOML file.NzFProfile already exists. Use set_credentials(overwrite=True) to update.rz.toml)dirsuffixw)ospathdirnameCREDENTIAL_FILEmakedirsrtouchopenreadstriptomlkitparsedocument exceptions ParseError ValueErrortempfilemkstempfdopendumpreplace BaseExceptionunlink) rr overwritecred_dir cred_filecontentconfigexcfdtmp_pathtmp_files SC:\Projects\Historical\venv\Lib\site-packages\runpod/cli/groups/config/functions.pyset_credentialsr0sfw//HK4((((... osW 5 5 5 Y Ynn&&G==??( g&&&%'' F !, Y Y YNOOUX X Y  Y Y Y Y Y Y Y Y Y Y Y Y Y Y Y  f  X !'*F7O#AAALB Yr3 1 1 1 +X L * * * + + + + + + + + + + + + + + + 8_-----  ( sg:D>/ * *>==H /4 ( ( *IYy))F * * * * * * * * * * * * * * * & PgPPPP P F7O + +RRRR  , z "HHHGGGH :s:BA BA""B%A"&B5BBBc tjtsdS t td5}t j|}dddn #1swxYwYn#ttf$rYdSwxYw||vrdS||S)z Returns the credentials for the specified profile from ~/.runpod/config.toml Returns None if the file does not exist, is not valid TOML, or does not contain the requested profile. Nr2r3)rr( credentialss r/get_credentialsr;\s 7>>/ * *t /4 ( ( /I)I..K / / / / / / / / / / / / / / / z "ttk!!t w s4A*A A*A""A*%A"&A**A?>A?)rF)r)__doc__rrpathlibrtomlir5rr expanduserrstrr0r8r;r/rCs '$$%<==,,S,3,PT,,,,^s6      rB