ce\dZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZddlmZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZ dd lm!Z!ddlm"Z"ddlm#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z,ej-rddl m.Z.GddeZ/GddZ0GddeZ1da2dZ3dS)z1Worker threads for using the anaconda-client api.) annotationsN) Unauthorized)QObjectQThreadQTimerSignal)SSLError)CondaAPI)TeamEditionAPI)is_internet_available)CONF AnacondaBrand)anaconda_solvers) constants) notifications) url_utils) version_utils) http_loggerlogger)is_text_stringto_text_string)utils)BinstarcbeZdZdZeeeeZeeeeZfdZdZ dZ xZ S) ClientWorker#Anaconda Client API process worker.ct||_||_||_d|_dS)rFN)super__init__methodargskwargs _is_finished)selfr!r"r# __class__s Alib/python3.11/site-packages/anaconda_navigator/api/client_api.pyr zClientWorker.__init__2s<    !c|jS)z?Return whether or not the worker has finished running the task.)r$r%s r' is_finishedzClientWorker.is_finished:s   r(cTd\}} tjd|j|ji|j}nR#t $rE}t |}|dd}|dd}Yd}~nd}~wwxYw|j |||d|_ dS)zStart the worker process.NNg{Gz?()NT) timesleepr!r"r# Exceptionstrreplace sig_finishedemitr$)r%erroroutputerrs r'startzClientWorker.start>s" v + Jt    T[$);t{;;FF + + +HHEMM#r**EMM#r**EEEEEE + tVU333 s+3 B;A==B) __name__ __module__ __qualname____doc__robjectsig_chain_finishedr6r r+r; __classcell__r&s@r'rr,s--776&&&11L"""""!!! ! ! ! ! ! ! !r(rceZdZdZdS)ArgszDDummy class to pass to anaconda client on token loading and removal.N)r<r=r>r?r(r'rErEMsNNNNr(rEc,eZdZdZdZfdZedZdZdZ dZ e d(d Z e d)d Z d*d ZdZdZdZdZ d(dZ d(dZdZd(dZd)dZdZdZ d+dZdZd,dZd,dZe dZd-d.d"Zd/d%Z d0d&Z!d0d'Z"xZ#S)1 _ClientAPIAnaconda Client API wrapper.ctt|_d|_t j|_g|_g|_ t|_ |j d|j j |jt!j|dS)rINi)rr r _conda_api_anaconda_client_api collectionsdeque_queue_threads_workersr_timer setIntervaltimeoutconnect_cleanr set_logged_data reload_client)r%r&s r'r z_ClientAPI.__init__Vs "**$(!!'))   hh  %%% ##DK000  r(c|jSN)rMr*s r'anaconda_client_apiz_ClientAPI.anaconda_client_apigs ((r(c8|jr8|jD]0}|r|j|1|jr:|jD]0}|r|j|1dS|jdS)z7Check for inactive workers and remove their references.N)rRr+removerQ isFinishedrSstop)r%wts r'rWz_ClientAPI._cleanks = ,] , ,==??,M((+++ = ] , ,<<>>,M((+++ , , K       r(ct|jdkrH|j}||jdSdS)z2Take avalaible worker from the queue and start it.rN)lenrPpopleftr;rS)r%threads r'_startz_ClientAPI._startysY t{  q [((**F LLNNN K        ! r(ct}t|||}|||j|j|j|j|j|j|j ||j ||j |||S)z?Create a worker for this client to be run in a separate thread.) rr moveToThreadr6rVrgquitstartedr;rPappendrQrR)r%r!r"r#rfworkers r'_create_workerz_ClientAPI._create_workersfdF33F#####DK000##FK000v|,,, 6""" V$$$ V$$$  r(Nc j |i}i}i}d|DD]\}}|dd\}}}|d|}|d|}||d Q||dpi ii ddd di d digd x||< d ||d d  d |<d } t |dd } n#t t f$rYnwxYwtj| d| fd dfkrL|ddp d d<|d dp d  d <| d<| d<|dd} | rI| d|<|dd d|<|dd d|<| D]o\}} tj | d d| d <| drC| d r d  d< fd d D d < ||<p||fS)zp Load all the available package information. See load_repadata for full documentation. Nc3zK|]6}dD]1}||iD]}|V27dS))packageszpackages.condaN)getitems).0contentgroupitems r' z,_ClientAPI._load_repodata..s~+ + ;+ + #KKr2288:: + + + + + + + + + + r(-nameversionhomer/rsummary) app_entryapp_typer} latest_buildlatest_versionsizer~typeversionsrr build_numberrrrrrT)unique descriptionc(g|]}|dv |S)rrF)rtr|records r' z-_ClientAPI._load_repodata..s2&&&&.00000r() valuesrsplitrrrlint ValueError TypeErrorrcomparerssortcopy) repodatametadata all_packagesall_appscanonical_name definitionr{r|_ record_build record_typepackagers @r'_load_repodataz_ClientAPI._load_repodatasD  H + + '00+ + + / O/ O &NJ .44S!<< D'1>>&$//D nnY88G!%%dD11F~!dD117R!# ""JJvr22$%&(%zz)R88 " / /  T"V :  % %g . . .&0nnVQ&?&?F6N7 #L ":>>.!#D#DEE  *    %gv6F/GHH,W[\^des^tZuuu!+!;!;!Mvf~v$.NN9b$A$A$VVIEVy!)5~&+2'($..44K O*5vw'/9~~k4/P/P{#G,.8nnZ.N.Nz"7+)//11 ( (MD'"/"4WZ5HQU"V"V"VGJ v ( )$>,29,=F=)&&&&#)*#5&&&z" "(X%%s#D>>EEcH|r|ng}g}i}|D]0}t|dd\}}} d|i||<1i} |D]0}t|dd\}}} d|i| |<1ttt t|t| zt|z} |pi} | D]}||p| |i} | dpd}| dpd}| dpg}| dpd}|| vr5tj}| |dd}tj }n||vrtj }||dd}tj }||vrp|}||d k ot|d k}||d k ot|d k}|r|r tj }n6|r tj }n'|r tj}ntj }tj}tjtjtj|tj|tj|tj|tj|tj|tjd i}|||S) z0Prepare model data for the packages table model.ryrzr|r~r/r}rrrrN)tuplersortedlistsetkeysrrC PIP_PACKAGE INSTALLED CONDA_PACKAGErd MIXGRADABLE UPGRADABLE DOWNGRADABLE NOT_INSTALLED COL_ACTION ACTION_NONECOL_PACKAGE_TYPECOL_NAMECOL_DESCRIPTION capitalize COL_VERSION COL_STATUSCOL_URLCOL_ACTION_VERSIONrl)rqlinkedpiprdatalinked_packagesrr{r|r pip_packagespackages_namespackages_metadatap_datar~urlrtype_statusvers upgradable downgradablerows r'_prepare_model_dataz_ClientAPI._prepare_model_datas ccb$ 9 9N$^%:%:3%B%BCC D'1%.$8OD ! ! ! 6 6N$^%:%:3%B%BCC D'1"+W!5L   T/..0011D9J9J9L9L4M4MMPTU]UbUbUdUdPePeeff i i  %N"+ + D\\$''J+<+@+@r+J+JFjj++1rG**V$$*Czz*--3Hjj!1228bG|## &t,00B??(()$/33IrBBh&&#D%,R%8!8!KSYY!^J'.$q''9#9#Lc$ii1nL!0l0!"#0!"%0!" am"E D!7#5#5#7#7 w f 3$d C KK     r(return-typing.Union['Binstar', TeamEditionAPI, None]ctjddd}tj\}}|tjtjfvr||S|tjkr||S|}tj 5tj d}dddn #1swxYwY ||}||S#tt t"f$r||cYSt$jj$r||cYStj$r_}|jdddks |jd kr;|t2jd d d Yd}~d}~wwxYw)zX Sets the client depending on the settings from the configuration file. mainanaconda_api_urlNrTrrz)izEmail verification failed!z]Your email is not verified. Please verify it on anaconda.org and log into your account again.zEmail not verified)anacondaemail_verification)messagecaptiontags)r rrget_logged_datar ANACONDA_ORGENTERPRISE_EDITION_load_binstar_client TEAM_EDITION_load_team_edition_clientrcatch_and_notifybinstar_clientr get_configuserrr rrequests exceptionsConnectionError BinstarErrorr"rlogoutrNOTIFICATION_QUEUEpush)r%r logged_brandlogged_api_urlrclientr8s r'rYz_ClientAPI.reload_client$s) 26&BTVZ1[1['+';'='=$ n M6 8XY Y Y,,^<< < =5 5 511.AA A   . 0 0 ; ; &1133E:C ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;  22377   (J7 C C C001ABBBBB&6 C C C001ABBBBB!.   JqsOv--5=D`3`3`KKMMM!499%!5? :HHHH s=%CC C *C??,G-(GG%AF?>F??Gci}tj5tj}dddn #1swxYwY||d<|dD]}||d<tj5tj|dddn #1swxYwY|}tj|dd|tj }||_ |S)ar Recreate the binstar client with new updated values. Notes: ------ The Client needs to be restarted because on domain change it will not validate the user since it will check against the old domain, which was used to create the original client. See: https://github.com/ContinuumIO/navigator/issues/1325 Nrsites)tokensiteclsconfig log_level) rrrrrr set_config load_tokenget_server_apiloggingNOTSETrM)r%api_urlrrrbinstars r'rz_ClientAPI._load_binstar_clientYs{   . 0 0 7 7#)4466F 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 u 7O**,, " "D!DKK  . 0 0 4 4  + +F 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4!! &55dVw~6  %,!s#AAA B22B69B6c|jdd}t|||_|jS)zt Sets the '_anaconda_client_api' to user TeamEdition API instead of default Binstar client. ssl_verifyF)rLload_rcrrr rM)r%r verify_ssls r'rz$_ClientAPI._load_team_edition_clientwsB _,,..22<GG $27J$G$G!((r(c|jjS)z6Return the current token registered with authenticate.)rMrr*s r'rz_ClientAPI.tokens(..r(cd} t|jtr|j}n1tj|}n#t$rYnwxYw|S)z!Load current authenticated token.N) isinstancerMr rrr get_api_urlOSError)r%rs r'rz_ClientAPI.load_tokens $3^DD L1<<>>&,778H8H8J8JKK    D  sA%A** A76A7cZt|jtr7|j|||}|j|nY|j||||}t }d|_||_tj |||S)zLogin callback.N) rrMr authenticate store_tokenrErrrr)r%usernamepassword applicationapplication_urlr new_tokenr"s r'_loginz_ClientAPI._logins d/ @ @ >1>>xS]^^I  % 1 1) < < < <1>>xS^`oppI66DDI"DJ  , ,Y = = =r(cD|j}|||||||S)zLogin to anaconda server.)rrn)r%rrrrrr!s r'loginz_ClientAPI.logins*""68X{O]ghhhr(c&d}t}d|_||_t|jt r|jntj ||rR |j n7#tj j $r }|}Yd}~nd}~wt$r }|}Yd}~nd}~wwxYwtjdt!j|S)zr Logout from anaconda.org. This method removes the authentication and removes the token. Nzlogout successful)rErrrrMr rrr remove_tokenremove_authenticationerrorsrr3rinfor rX)r%r8r"es r'rz_ClientAPI.logouts vv ZZ\\ d/ @ @ -  % , , . . . .  - -d 3 3 3zz|| -CCEEEE%,9EEEEEE EEEEEE K+ , , ,  s$B33C'C C'C""C'c<||j||S)a Load all the available packages information for downloaded repodata. For downloaded repodata files (repo.anaconda.com), additional data provided (anaconda cloud), and additional metadata and merge into a single set of packages and apps. :param repodata: Data loaded from the conda cache directories. :param metadata: Metadata info form different sources. For now only from repo.anaconda.com )r)rnr)r%rrs r' load_repodataz_ClientAPI.load_repodatas"""4#68"TTTr(cD|j}||||||S)z=Prepare downloaded package info along with pip pacakges info.)rr)rrn)r%rqrrrr!s r'prepare_model_dataz_ClientAPI.prepare_model_datas6)""    #   r(c.|dS)z'Return current logged user information.Nr) organizationsr*s r'rz_ClientAPI.users!!!---r(c|jjS)zReturn current domain.)rMdomainr*s r'rz_ClientAPI.domains(//r(cP|jj}|||||||S)a]Return all the available packages for a given user. Parameters ---------- type_: Optional[str] Only find packages that have this conda `type`, (i.e. 'app'). access : Optional[str] Only find packages that have this access level (e.g. 'private', 'authenticated', 'public'). )rplatform package_typeraccess)rM user_packagesrn)r%rrrrrr!s r'rqz_ClientAPI.packagess>*8"" % #   r(cd |j|}n#t$ri}YnwxYw|S)z0List all the organizations a user has access to.r)rMrr3)r%rrs r'rz_ClientAPI.organizationssK ,111>>DD   DDD  s  --r4c"t|jtrtjddSi}t j5tj }dddn #1swxYwY|ddS)z*Get the anaconda client url configuration.ranaconda_server_api_urlNrhttps://api.anaconda.org) rrMr r rrrrrrrr% config_datas r'rz_ClientAPI.get_api_urls d/ @ @ ?8F$=>> >"$  . 0 0 < <(.99;;K < < < < < < < < < < < < < < <u&@AAAsA00A47A4cHt|jtr(tjt jdddSi}tj5tj }dddn #1swxYwY|ddS)z/Get the anaconda client info url configuration.rrz api/systemNrr) rrMr rjoinr rrrrrrrr s r'get_api_info_urlz_ClientAPI.get_api_info_urls d/ @ @ ]>$(63L"M"M|\\ \"$  . 0 0 < <(.99;;K < < < < < < < < < < < < < < <u&@AAAsBB Bctj5tj}||d<tj|ddddS#1swxYwYdS)*Set the anaconda client url configuration.rN)rrrrrr)rr!s r' set_api_urlz_ClientAPI.set_api_urls . 0 0 9 9(.99;;K!$K   + +K 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9sAA$$A(+A(T set_conda_sslboolci}tj5tj}dddn #1swxYwY|s*|d|ddS|jddd}|x|d<|d<tj5tj |dddn #1swxYwYtjddd}tjdd g}tj |pd |vrnnt|tr-tjdd |tjdd dn,tjdd dtjdd ||S) z` Get conda ssl configuration and set navigator and anaconda-client accordingly. NrrTrrrssltrusted_serversr/ssl_verificationssl_certificate)rrrrrrrrL config_get communicaterr rnetlocrr)r)r%r(rvaluerr,s r'get_sslz_ClientAPI.get_ssl$s*   . 0 0 7 7#)4466F 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 L::lFJJ|T,J,JKK K**<88DDFFqIMMl[[6;;|vl3  . 0 0 4 4  + +F 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 404x@PRV/W/W,0HUr?rGr propertyr\rWrgrn staticmethodrrrYrrrrrrrrrrrrqrrr$r'r3r7rOrQrBrCs@r'rHrHQs&&O"))X)         "M&M&M&\M&^>>>\>D3333j<)))///   PT(PTiiii: U U U U     ...000 TX    . B B B B B B B B99\9@ F F F F    D\\\\\\\\r(rHc:ttatS)zClient API threaded worker.) CLIENT_APIrHrFr(r' ClientAPIrV|s\\ r()4r? __future__rcollections.abcrNrIrr1typingrrbinstar_client.errorsr qtpy.QtCorerrrrrequests.exceptionsr anaconda_navigator.api.conda_apir 'anaconda_navigator.api.team_edition_apir anaconda_navigator.api.utilsr anaconda_navigator.configr ranaconda_navigator.utilsrrrrrranaconda_navigator.utils.logsrr"anaconda_navigator.utils.py3compatrrr/rrE TYPE_CHECKINGrrrErHrUrVrFr(r'resi87""""""  ......888888888888((((((555555BBBBBB>>>>>>99999999555555333333222222......222222========MMMMMMMM  '&&&&&&!!!!!7!!!BOOOOOOOOe\e\e\e\e\e\e\e\P r(