B âÖÓYýOã@sdZddlmZddlZddlZddlZddlZddlmm Z ddl mm Z ddl m Z ddlmZmZmZy ddlZWn ek r ddlmZYnXddd„Zdd „Zd d „Zd d „ZGdd„deƒZdd„Ze d¡Zdd„Z dd„Z!dd„Z"dd„Z#dd„Z$dS)z3Environment class representing a conda environment.é)Úabsolute_importN)Ú is_string)Ú _load_stringÚ _save_fileÚ _YAMLErrorcsVdd„}ˆdkr|‰t‡fdd„|Dƒƒ‰t‡‡fdd„|Dƒƒ}|t|ƒ}t|ƒS)NcSs|S)N©)Úitemrrú8lib/python3.7/site-packages/anaconda_project/env_spec.pyÚ default_key sz4_combine_keeping_last_duplicate..default_keycsg|] }ˆ|ƒ‘qSrr)Ú.0r)Úkey_funcrr ú %sz3_combine_keeping_last_duplicate..csg|]}ˆ|ƒˆkr|‘qSrr)r r)Ú items2_keysr rr r &s)ÚsetÚlistÚtuple)Zitems1Zitems2r r Úcombinedr)rr r Ú_combine_keeping_last_duplicates rcCs t |¡}|dkr|S|jSdS)N)Ú conda_apiÚ parse_specÚname)ÚspecÚparsedrrr Ú_conda_combine_key+s rcCs t |¡}|dkr|S|jSdS)N)Úpip_apirr)rrrrr Ú_pip_combine_key4s rcCst||tdS)N)r )rr)ÚfirstÚsecondrrr Ú_combine_conda_package_lists=src@s6eZdZdZddd„Z edd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZed d!„ƒZed"d#„ƒZed$d%„ƒZed&d'„ƒZd(d)„Zd*d+„Zd,d-„Zed.d/„ƒZed0d1„ƒZd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Z dS)?ÚEnvSpeczdRepresents a set of required conda packages we could potentially instantiate as a Conda environment.rNc Cs\|dk s t‚|dk st‚||_t|ƒ|_t|ƒ|_t|ƒ|_||_d|_d|_d|_ ||_ ||_ | |_ tt  |¡ƒ|_x2tdd„|j DƒƒD]}|dksŒ||j ksŒt‚qŒWtƒ} x*|jD] } t  | ¡} | dk r¸| | | j<q¸W| |_tƒ} x,|jD]"} t  | ¡} | dk rð|  | j¡qðW| |_tƒ}x.|jD]$} t | ¡} | dk r*| || j<q*W||_dS)azConstruct a package set with the given name and packages. Args: name (str): name of the package set conda_packages (list): list of package specs to pass to conda install channels (list): list of channel names pip_packages (list): list of pip package specs to pass to pip description (str or None): one-sentence-ish summary of what this env is inherit_from_name (str or None): name of what we inherit from inherit_from (EnvSpec or None): pull in packages and channels from lock_set (CondaLockSet): locked packages or None NcSsg|] }|j‘qSr)r)r rrrr r psz$EnvSpec.__init__..)ÚAssertionErrorÚ_namerÚ_conda_packagesÚ _channelsÚ _pip_packagesÚ _descriptionÚ _logical_hashÚ _locked_hashÚ _import_hashÚ_inherit_from_namesÚ _inherit_fromÚ _lock_setrZsort_platform_listÚ _platformsÚdictÚconda_packages_for_createrrÚ_conda_specs_for_create_by_namerÚconda_packagesÚaddÚ_conda_logical_specs_name_setÚ pip_packagesrÚ_pip_specs_by_name)Úselfrr0Úchannelsr3Ú descriptionÚinherit_from_namesÚ inherit_fromÚ platformsÚlock_setZconda_specs_by_namerrZname_setZpip_specs_by_namerrr Ú__init__DsD            zEnvSpec.__init__cCs|jS)zGet name of the package set. May be None for the anonymous shared base spec (toplevel packages, channels sections). )r!)r5rrr rŒsz EnvSpec.namecCs|jdkr|jS|jSdS)z'Get the description of the environment.N)r%r!)r5rrr r7•s zEnvSpec.descriptioncCs˜ddl}| ¡}x|D]}| | d¡¡qWx|jD]}| | d¡¡q6Wx|jD]}| | d¡¡qVWx|D]}| | d¡¡qtW| ¡}|S)Nrzutf-8)ÚhashlibZsha1ÚupdateÚencoder3r6Z hexdigest)r5r0r:r=ÚmÚpÚcÚresultrrr Ú _compute_hashs    zEnvSpec._compute_hashcCs"|jdkr| |j|j¡|_|jS)zÒGet a hash of our "logical" requirements. (Changing logical requirements could change the lock set if we recreate it.) Order matters (change in order will count as a change). N)r&rDr0r:)r5rrr Ú logical_hash«s zEnvSpec.logical_hashcCs"|jdkr|j|jdd|_|jS)z×Get a hash of our locked packages (what we'd pass to conda create). This is used to see if we need to reprepare environments. Order matters (change in order will count as a change). Nr)r:)r'rDr.)r5rrr Ú locked_hash¹s zEnvSpec.locked_hashcCs"|jdkr|j|jdd|_|jS)zíGet a hash of parts of the env spec that can appear in environment.yml. This is used to see if we need to re-import the environment.yml, requirements.txt or whatever. Those files don't have platform information. Nr)r:)r(rDr0)r5rrr Ú import_hashÅs zEnvSpec.import_hashcs"d|‰‡fdd„}|j||dS)NÚ_cs t|ˆƒS)N)Úgetattr)r)Ú private_attrrr ÚgetterÓsz&EnvSpec._get_inherited..getter)r )Ú_get_inherited_with_getter)r5Z public_attrr rKr)rJr Ú_get_inheritedÐs zEnvSpec._get_inheritedcsp‡fdd„‰g}ˆ|g|ƒ|d|ks,t‚g}x|D]}| ||ƒ¡q6Wg}x|D]}t|||d}qVW|S)Ncs0x*|D]"}||krˆ|j|ƒ| |¡qWdS)N)r*Úappend)ÚspecsZ accumulatorr)Ú_linearized_ancestorsrr rPÙs  zAEnvSpec._get_inherited_with_getter.._linearized_ancestorséÿÿÿÿ)r )r rNr)r5rKr Z ancestorsZ to_combinerrrr)rPr rLØs    z"EnvSpec._get_inherited_with_gettercCs | dt¡S)zDGet the conda packages to install in the environment as an iterable.r0)rMr)r5rrr r0ëszEnvSpec.conda_packagescCs | d¡S)z0Get the channels to install conda packages from.r6)rM)r5rrr r6ðszEnvSpec.channelscCs | d¡S)z,Get the platforms the environment can be on.r:)rM)r5rrr r:õszEnvSpec.platformscCs | dt¡S)zBGet the pip packages to install in the environment as an iterable.r3)rMr)r5rrr r3úszEnvSpec.pip_packagescCs|jS)z5Conda package names that we require, as a Python set.)r2)r5rrr Úconda_package_names_setÿszEnvSpec.conda_package_names_setcCst|j ¡ƒS)z5Conda package names that we require, as a Python set.)rr/Úkeys)r5rrr Ú"conda_package_names_for_create_setsz*EnvSpec.conda_package_names_for_create_setcCst|j ¡ƒS)z3Pip package names that we require, as a Python set.)rr4rS)r5rrr Úpip_package_names_set szEnvSpec.pip_package_names_setcCs|jS)z'Get ``CondaLockSet`` for this env spec.)r+)r5rrr r;szEnvSpec.lock_setcCs,|jdk r"|jjr"|jjr"|jjS|jSdS)z=Get conda packages (preferring the lock set list if present).N)r+ZenabledZsupports_current_platformZ"package_specs_for_current_platformr0)r5rrr r.sz!EnvSpec.conda_packages_for_createcCs4g}x*|D]"}| |d¡}|dk r | |¡q W|S)N)ÚgetrN)r5ÚnamesÚmappingrOrrrrr Ú_specs_for_package_namess   z EnvSpec._specs_for_package_namescCs| ||j¡S)z>Get the full install specs given an iterable of package names.)rYr/)r5rWrrr Úspecs_for_conda_package_names#sz%EnvSpec.specs_for_conda_package_namescCs| ||j¡S)z>Get the full install specs given an iterable of package names.)rYr4)r5rWrrr Úspecs_for_pip_package_names'sz#EnvSpec.specs_for_pip_package_namescCs|jS)z$Env spec that we inherit stuff from.)r*)r5rrr r9+szEnvSpec.inherit_fromcCs|jS)z*Env spec names that we inherit stuff from.)r))r5rrr r80szEnvSpec.inherit_from_namescCs(tj dtj |d¡¡}tj ||j¡S)zEThe filesystem path to the default conda env containing our packages.ZANACONDA_PROJECT_ENVS_PATHZenvs)ÚosÚenvironrVÚpathÚjoinr)r5Z project_dirZ envs_pathrrr r^5sz EnvSpec.pathcCs†tt |j|j¡ƒ}tt |j|j¡ƒ}tt |j|j¡ƒ}|rXdgttdd„|ƒƒ}|rtdgttdd„|ƒƒ}d |||¡S)zFA string showing the comparison between this env spec and another one.z pip:cSsd|S)Nz r)Úxrrr ÚAsz#EnvSpec.diff_from..z channels:cSsd|S)Nz r)r`rrr raCsÚ )rÚdifflibÚndiffr6r0r3Úmapr_)r5ÚoldÚ channels_diffÚ conda_diffÚpip_diffrrr Ú diff_from;szEnvSpec.diff_fromcCs¦dd„dDƒ}dd„}|t |j|j¡ƒ}x|D]}||kr0| |¡q0Wt|ƒdkrZdS|t |j|j¡ƒ}t|ƒdkr~dS|t |j|j¡ƒ}t|ƒdkr¢dSdS) a%Check whether the diff is exclusively removing 'bokeh' or 'notebook'. This is used for a hack, because we can auto-add 'bokeh' or 'notebook' packages when we anaconda-project init, and that alone shouldn't result in being out of sync with the environment.yml. cSsg|] }d|‘qS)z- r)r Úrrrr r Msz?EnvSpec.diff_only_removes_notebook_or_bokeh..)ZbokehZnotebookcSsttdd„|ƒƒS)NcSs| d¡p| d¡S)Nz- z+ )Ú startswith)Úlinerrr raPszUEnvSpec.diff_only_removes_notebook_or_bokeh..filter_context..)rÚfilter)Úitemsrrr Úfilter_contextOszCEnvSpec.diff_only_removes_notebook_or_bokeh..filter_contextrFT)rcrdr0ÚremoveÚlenr6r3)r5rfZ to_removerprhrkrgrirrr Ú#diff_only_removes_notebook_or_bokehFs    z+EnvSpec.diff_only_removes_notebook_or_bokehcCsæt|jƒ}t|jƒ}|r(| t|d¡t|jƒ}t|jƒ}tjdtj d}|j dk rf|j |dd<n |dd=||dd<||dd<t |ƒd kr ||dd <t |j ƒd krÞt |j ƒd krÈ|j d }n t|j ƒ}||dd <|dS) z6Get JSON for an anaconda-project.yml env spec section.)ÚpipzCsomething: description: null packages: [] channels: [] )ÚLoaderNZ somethingr7Úpackagesr6rr:ér9) rr"r$rNr-r#r,ÚryamlÚloadÚRoundTripLoaderr%rrr8)r5rvr3r6r:Z template_jsonrWrrr Úto_jsonds*              zEnvSpec.to_jsoncCsxt|jƒ}t|jƒ}|r(| t|d¡t|jƒ}tjdtjd}|j dk sPt ‚|j |d<||d<||d<t ||ƒdS)z Save as an environment.yml file.)rtz%name: dependencies: [] channels: [] )ruNrÚ dependenciesr6) rr0r3rNr-r6rxryrzrr r)r5Úfilenamervr3r6Úyamlrrr Úsave_environment_ymlŒs    zEnvSpec.save_environment_yml)rNrrrN)N)N)!Ú__name__Ú __module__Ú __qualname__Ú__doc__r<Úpropertyrr7rDrErFrGrMrLr0r6r:r3rRrTrUr;r.rYrZr[r9r8r^rjrsr{rrrrr rAsD ?                (rc Cspy.t |dd¡}| ¡}WdQRXt|ƒ}Wnttfk rFdSXd}d|kr\|d}|s€d|kr€|dr€tj |d¡}|stj |¡}|  dg¡}t |t ƒsªg}|  dg¡}t |t ƒsÄg}g}g}xh|D]`} t | ƒrê|  | ¡qÒt | tƒrÒd| krÒt | dt ƒrÒx&| dD]} t | ƒr|  | ¡qWqÒWg} x"|D]} t | ƒr@|   | ¡q@Wt||| |d d S) z=Load an environment.yml as an EnvSpec, or None if not loaded.rkzutf-8NrÚprefixr|r6rtr)rr0r6r3r:)ÚcodecsÚopenÚreadrÚIOErrorrr\r^ÚbasenamerVÚ isinstancerrrNr-r) r}ÚfileÚcontentsr~rZraw_dependenciesZ raw_channelsr0r3ZdepZpip_depr6Zchannelrrr Ú_load_environment_ymlŸsD           rŽz^-([-a-zA-Z0-9]+)\s(.*)c Csþy&t |dd¡}| ¡}WdQRXWnttfk r>dSXg}x¨|D] }| ¡}| d¡sJ|dkrjqJt |¡}|dk rà|  d¡}|  d¡}|dkr¤|  |¡qê|dkrêt j   t j  |¡|¡}t|ƒ} | dk rê| | j¡qJ|  |¡qJWtd d d |d S) z=Load a requirements.txt as an EnvSpec, or None if not loaded.rkzutf-8Nú#ÚrwéÚeÚdefaultr)rr0r6r3)r†r‡Ú readlinesr‰rÚstriprlÚ_requirement_option_reÚsearchÚgrouprNr\r^r_ÚdirnameÚ_load_requirements_txtÚextendr3r) r}rŒÚlinesrvrmr@ZoptionÚpackager^Z child_specrrr rš×s.      ršcCs| d¡rt|ƒSt|ƒSdS)Nz.txt)ÚendswithršrŽ)r}rrr Ú_load_importables rŸcCs<d}x2|D]*}tj ||¡}t|ƒ}|dk r ||fSq WdS)N)zenvironment.ymlzenvironment.yamlzrequirements.txt)NN)r\r^r_rŸ)Údirectory_pathÚ filenamesr}Zfullrrrr Ú_find_importable_spec s  r¢cCsJt|ƒ\}}|dkrdSx(|D] }|j|jkr|j|jkrdSqW||fS)N)NN)r¢rrG)Z project_specsr rr}Úexistingrrr Ú!_find_out_of_sync_importable_specs    r¤c Cs0|dkrd}n|f}tddggt ¡dd|dS)Nrr“Zanacondaz-Default environment spec for running commands)rr0r6r:r7r8r9)rrZdefault_platforms_with_current)Zshared_base_specr9rrr Ú_anaconda_default_env_spec"sr¥)N)%rƒZ __future__rr†rcr\ÚreZ#anaconda_project.internal.conda_apiZinternalrZ!anaconda_project.internal.pip_apirZ$anaconda_project.internal.py2_compatrZanaconda_project.yaml_filerrrZ ruamel_yamlrxÚ ImportErrorZ ruamel.yamlr~rrrrÚobjectrrŽÚcompiler–ršrŸr¢r¤r¥rrrr Ús6      `5 +