o mi: @s~ddlZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl Z e dkZejddZdZdZded Zd ZeeZGd d d eZGd ddeZGdddeZd=de ede efddZde efddZdefddZ d=de edefddZ!dede eeffddZ"dedefdd Z#dede eeffd!d"Z$d#ed$ede efd%d&Z%Gd'd(d(e&Z'  d>d#ed$e ed)e ede eeeffd*d+Z(d,edefd-d.Z)d/e ee*fde eeffd0d1Z+ 2d?d3ed4e,deeeeefffd5d6Z-   d@de ed7e eeefd)e ede'fd8d9Z.d:edeeeee eeeefffffd;d<Z/dS)AN)AnyDictMappingOptionalTupleUnionWindowsz.dockerz config.jsonz .dockercfgz docker.iozhttps://index.z/v1/zc@seZdZdZdS) DockerErrorzBase class from which all other exceptions inherit. If you want to catch all errors that the Docker SDK might raise, catch this base exception. N)__name__ __module__ __qualname____doc__rrO/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/docker/auth.pyr sr c@ eZdZdS)InvalidConfigFileErrorNr r r rrrrrrc@r)InvalidRepositoryErrorNrrrrrr!rr config_pathreturnc Csttd|ttjtttjttg}t dt ||D]}tj |r8t d||Sq$t ddS)NzTrying paths: zFound file at path: zNo config file found) listfilterconfig_path_from_environmentospathjoinhome_dirDOCKER_CONFIG_FILENAMELEGACY_DOCKER_CONFIG_FILENAMElogdebugreprexists)rpathsrrrrfind_config_file%s$   r%cCs*tjd}|s dStj|tjtS)N DOCKER_CONFIG)renvirongetrrbasenamer) config_dirrrrr>s rcCstr tjddStjdS)zGet the user's home directory. Uses the same logic as the Docker Engine client - use %USERPROFILE% on Windows, $HOME/getuid on POSIX. USERPROFILE~)IS_WINDOWS_PLATFORMrr'r(r expanduserrrrrrEs rc Cst|}|siSzt|}t|}|WdWS1s wYWnttfy?}z t|WYd}~nd}~wwtdiS)N4All parsing attempts failed - returning empty config)r%openjsonloadOSError ValueErrorr r!)r config_filefconferrrload_general_configQs  & r: repo_namecCsXd|vr td|dt|\}}|ddks|ddkr&td|dt||fS) Nz://z)Repository name cannot contain a scheme ()r-zInvalid index name (z%). Cannot begin or end with a hyphen.)rsplit_repo_nameresolve_index_name)r; index_name remote_namerrrresolve_repository_nameds    rCrAcCst|}|dtkr t}|S)Nzindex.)convert_to_hostname INDEX_NAME)rArrrr@rs r@cCsT|dd}t|dksd|dvr"d|dvr"|ddkr"t|fS|d|dfS)N/.r: localhost)splitlenrE)r;partsrrrr?ys  $r? authconfigregistrycCst|ts t|}||SN) isinstance AuthConfigget_credential_store)rNrOrrrrSs  rSc seZdZd*dedeeddffdd Ze d+deeeee ffd e deeeee fffd d Z e d*d eed eeee fdeeddfddZ e deeeee fffddZe deefddZe defddZe de fddZ d*deedeeee ffddZdeededeeee ffddZdeddfd d!Zdeedeefd"d#Zdeeeee fffd$d%Zd&ed'eee fddfd(d)ZZS),rRNdct credstore_envrcs8t|d|vri|d<||||_t|_dSNauths)super__init__update_credstore_envdict_stores)selfrTrU __class__rrrYs   zAuthConfig.__init__Fentriesraise_on_errorcCsi}|D]j\}}t|ts%td|d|r!td|iSd|vr:td|d|di||<qd|vrLtd|d i||<qt|d\}}td t|d t|d |||d |d||<q|S)aParse authentication entries. Args: entries: Dict of authentication entries. raise_on_error: If set to true, an invalid format will raise InvalidConfigFileError Returns: Authentication registry. zConfig entry for key z is not auth configz#Invalid configuration for registry identitytokenz*Found an IdentityToken entry for registry IdentityTokenauthzAuth data for z> is absent. Client might be using a credentials store instead.zFound entry (registry=z , username=r<emailusernamepasswordrf serveraddress) itemsrQr\r r!r decode_authr"r()clsrarbr8rOentryrhrirrr parse_auths:   zAuthConfig.parse_authr config_dictc CsT|sJt|}|s |i|Szt| }t|}Wdn1s"wYWn!tttfyI}zt||t ||WYd}~Sd}~wwi}t |t sSJ| drltd| d|j|dddi| drtd| d|di| drtd | d|di|r|||Std |d||i|S) aOLoad authentication data from a Docker configuration file. If the config_path is not passed in it looks for a configuration file in the root directory. Lookup priority: explicit config_path parameter > DOCKER_CONFIG environment variable > ~/.docker/config.json > ~/.dockercfg. NrWzFound 'auths' sectionT)rb credsStorezFound 'credsStore' section credHelperszFound 'credHelpers' sectionzNCouldn't find auth-related section ; attempting to interpret as auth-only file)r%r1r2r3r4KeyErrorr5r r!_load_legacy_configrQrr(rZropop)rmrrprUr6r7r9resrrr load_configsB           zAuthConfig.load_configcC |diSrVr(r^rrrrW zAuthConfig.authscCs |ddS)Nrqryrzrrr creds_storer{zAuthConfig.creds_storecCrx)Nrrryrzrrr cred_helpersr{zAuthConfig.cred_helperscCs|j o |j o |j SrP)rWr|r}rzrrris_empty szAuthConfig.is_emptyrOcCs|js|jr(||}|dur(td||||}|dur#|Std|r.t|nt}tdt|||j vrNtdt||j |S|j D]\}}t||krktdt||SqStddS)aReturn the authentication data for a specific registry. As with the Docker client, legacy entries in the config with full URLs are stripped down to hostnames before checking for a match. Returns None if no match was found. NzUsing credentials store z/No entry in credstore - fetching from auth dictLooking for auth entry for zFound No entry found) r|r}rSr r!_resolve_authconfig_credstorer@rEr"rWrk)r^rO store_namecfgkeyr8rrrresolve_authconfigs(       zAuthConfig.resolve_authconfigcredstore_namec Cs|r|tkrt}tdt|||}z'||}d|i}|dtkr0|d|d<|WS||d|dd|WSt j t fyPtdYdSt j ye}z t dt|d}~ww) Nr ServerAddressUsernameSecretrd)rPasswordrzCredentials store error: )rE INDEX_URLr r!r"_get_store_instancer(TOKEN_USERNAMErZ dockerpycredsCredentialsNotFoundr5 StoreErrorr )r^rOrstoredatarvr9rrrr2s(      z(AuthConfig._resolve_authconfig_credstorenamezdockerpycreds.StorecCs*||jvrtj||jd|j|<|j|S)N) environment)r]rStorer[)r^rrrrrKs  zAuthConfig._get_store_instancecCs"|r|tkrt}|j|p|jSrP)rErr}r(r|)r^rOrrrrSRs zAuthConfig.get_credential_storecCsh|j}|jr ||j}|D] }|||j||<q|jD] \}}|||||<q%|SrP) rWcopyr|rrkeysrr}rk)r^ auth_datarkregrrrrget_all_credentialsXs  zAuthConfig.get_all_credentialsrrcCs||d|<dSrVr)r^rrrrradd_authfszAuthConfig.add_authrPF)r r r rrrrY classmethodstrrboolrorwpropertyrWr|r}r~rrrrSrr __classcell__rrr_rrRs^  8 4  " &rRrUcCst|ts t||}||SrP)rQrRr)rNrOrUrrrrjs   rurlcCs |dddddddS)Nzhttp://r,zhttps://rFrGr)replacerK)rrrrrDts rDrecCsBt|tr |d}t|}|dd\}}|d|dfS)Nascii:rGutf8)rQrencodebase64 b64decoderKdecode)resloginpwdrrrrlxs   rlFrarbcCs t||S)zParse authentication entries. Args: entries: Dict of authentication entries. raise_on_error: If set to true, an invalid format will raise InvalidConfigFileError Returns: Authentication registry. )rRro)rarbrrrros rorpcCst|||SrP)rRrw)rrprUrrrrwsrwr6c CstdzIg}t|%}|D]}||ddqt|dkr*tdWdn1s4wYt |d\}}dt |||dt diiWSt ye}z t|WYd}~nd}~wwtd iS) Nz+Attempting to parse legacy auth file formatz = rGz$Invalid or empty configuration file!rrWrgr0) r r!r1 readlinesappendstriprKrLrrlrEr Exception)r6rr7linerhrir9rrrrts4      rtrP)NNr)NNN)0rr2loggingrplatformtypingrrrrrrrsystemr.rrrrrErr getLoggerr r rr rrrr%rrr:rCr@r?rSr\rRrrDbytesrlrrorwrtrrrrs      d  "   "