B YJ3@s"dZddlmZmZy0ddlZddlmZddlm Z ddlm Z WnDe k rddl m ZddlmZddlm Z ddlm Z YnXddlZddlZddlZddlZddlZddlmZddlmZdd lmZeZe Ze Zdd d Zd dZddZ dddZ!ddZ"Gddde#Z$dS)z#YAML file loading and manipulation.)absolute_importprint_functionN) YAMLError) CommentedMap) CommentedSeq)makedirs_ok_if_exists)rename_over_existing) is_stringutf-8cCs|dtt}zBt|d| }||||WdQRXt||Wdyt |Wnt t fk r~YnXXdS)Nz.tmp-w) struuidZuuid4codecsopenwriteflushcloserosremoveIOErrorOSError)pathcontentsencodingZtmpfiler9lib/python3.7/site-packages/anaconda_project/yaml_file.py_atomic_replace-s rcCs8|dkriSttjtjjs$ttj|tjdSdS)N)Loader)strip issubclassryamlRoundTripLoader constructorZSafeConstructorAssertionErrorload)rrrr _load_string<s r'cCstj|tjdS)N)ZDumper)r"dumpZRoundTripDumper)yamlrrr _dump_stringHsr*c Cs|dkrt|}ytj|tjdWnrtk r}zTtdtjdtdt|tjdtdtjdt|tjdt dt|Wdd}~XYnXt j |st j |}t|t||dS)N)rz=ruamel.yaml bug; it failed to parse a file that it generated.)rz the parse error was: zGenerated file was:zFBug in ruamel.yaml library; failed to parse a file that it generated: )r*r"r&r#rprintsysstderrr RuntimeErrorrrisfiledirnamerr)r)filenamerer0rrr _save_fileLs"  r3cCs^t|dr|jt|tr6x:|D] }t|q$Wn$t|trZx|D] }t|qJWdS)Nfa)hasattrr4Zset_block_style isinstancelist_block_style_all_nodesdictvalues)r)elementvaluerrrr8as     r8c@seZdZdZddZddZddZdd Zd d Zd d Z e ddZ e ddZ e ddZ e ddZe ddZe ddZe ddZddZddZed d!Zd"d#Zd$d%Zd&d'Zd(d)Zd/d+d,Ze d-d.Zd*S)0YamlFilea Abstract YAML file, base class for ``ProjectFile`` and ``LocalStateFile``. Be careful with creating your own instance of this class, because you have to think about when other code might load or save in a way that conflicts with your loads and saves. cCs||_d|_d|_|dS)aLoad a YamlFile with the given filename. Raises an exception on an IOError, but if the file is missing this succeeds (then creates the file if and when you call ``save()``). If the file has syntax problems, this sets the ``corrupted`` and ``corrupted_error_message`` properties, and attempts to modify the file will raise an exception. rrN)r1_previous_content _change_countr&)selfr1rrr__init__vs zYamlFile.__init__c Cstd|_d|_d|_d|_|jd|_y>t|jdd}|}WdQRXt ||_ t |j |_ Wnt k r}z|jtjkrd|_ n|Wdd}~XYntk r&}zld|_t||_t|dd}|dk r|jdk r|jdkr|j|_|jdk r|jdkr|j|_d|_ Wdd}~XYnX|j dkrp|jrFt|_ n*||_ t|j |spt |j |_ dS) a)Reload the file from disk, discarding any unsaved changes. If the file has syntax problems, this sets the ``corrupted`` and ``corrupted_error_message`` properties, and attempts to modify the file will raise an exception. Returns: None FNrzutf-8TZ problem_markr) _corrupted_corrupted_error_message_corrupted_maybe_line_corrupted_maybe_columnr?rrr1readr'_yamlr*r>rerrnoZENOENTrr getattrlinecolumnr9_default_contentr8_save_default_content)r@rrr2Zmarkrrrr&s>            z YamlFile.loadcCsdS)Nz yaml filer)r@rrr_default_commentszYamlFile._default_commentcCst}|||S)N)rZyaml_set_start_commentrP)r@rootrrrrNszYamlFile._default_contentcCsdS)zHOverride to change whether we consider a default, unmodified file dirty.Tr)r@rrrrOszYamlFile._save_default_contentcCs|jrtd|j|jfdS)Nz'Cannot modify corrupted YAML file %s %s)rD ValueErrorr1rE)r@rrr_throw_if_corruptedszYamlFile._throw_if_corruptedcCstj|jS)zBasename of the filename.)rrbasenamer1)r@rrrrTszYamlFile.basenamecCs|jS)zGet whether the file is corrupted. A corrupted file has a syntax error so we can't modify and save it. See ``corrupted_error_message`` for what's wrong with it. Returns: True if file is corrupted. )rD)r@rrr corrupteds zYamlFile.corruptedcCs|jS)zGet the error message if file is corrupted, or None if it isn't. Use this to display the problem if the file is corrupted. Returns: Corruption message or None. )rE)r@rrrcorrupted_error_messages z YamlFile.corrupted_error_messagecCs|jS)z}Get the line number for syntax error, or None if unavailable. Returns: Corruption line or None. )rF)r@rrrcorrupted_maybe_lineszYamlFile.corrupted_maybe_linecCs|jS)zzGet the column for syntax error, or None if unavailable. Returns: Corruption column or None. )rG)r@rrrcorrupted_maybe_columnszYamlFile.corrupted_maybe_columncCs|jS)aBGet the number of times we've resynced with the file on disk (reloaded or saved changes). This is used for cache invalidation. If a cached value becomes invalid whenever ``change_count`` increments, then the cached value will be recomputed whenever we save new changes or reload the file. )r?)r@rrr change_countszYamlFile.change_countcCs|jt|jkS)z"Get whether changes are all saved.)r>r*rI)r@rrrhas_unsaved_changesszYamlFile.has_unsaved_changescCs|jd|_dS)zApply any in-memory changes as if we'd saved, but don't actually save. This is used to "try out" a change before we save it. We can load() to undo our changes. rBN)r?)r@rrruse_changes_without_savingsz#YamlFile.use_changes_without_savingcCsB|t|j}||jkr>t|j|j||jd|_||_dS)zWrite the file to disk, only if any changes have been made. Raises ``IOError`` if it fails for some reason. Returns: None rBN)rSr*rIr>r3r1r?)r@rrrrsaves    z YamlFile.savecCsBt|r|fSytdd|DStk r<tdYnXdS)Ncss|] }|VqdS)Nr).0r;rrr 3sz!YamlFile._path..z9YAML file path must be a string or an iterable of strings)r r7 TypeErrorrR)clsrrrr_path-s zYamlFile._pathcCs<|j}x0|D](}||kr0t||tr0||}q dSq W|S)N)rIr6r9)r@piecescurrentprrr_get_dict_or_none7s   zYamlFile._get_dict_or_nonecCsT||j}x@|D]8}||ks.t||tsDt||<t||||}qW|S)N)rSrIr6r9rr8)r@rbrcrdrrr_ensure_dicts_at_path@s    zYamlFile._ensure_dicts_at_pathcCs4|||}||dd}|||d<dS)a6Set a single value at the given path. Overwrites any existing value at the path. This method does not save the file, call ``save()`` to do that. Args: path (str or list of str): single key, or list of nested keys value: any YAML-compatible value type N)rSrarf)r@rr<existingrrr set_valueNs  zYamlFile.set_valuecCsF|||}||dd}|d}|dk rB||krB||=dS)zRemove a single value at the given path. This method does not save the file, call ``save()`` to do that. Args: path (str or list of str): single key, or list of nested keys Nrg)rSrare)r@rrhkeyrrr unset_value_s  zYamlFile.unset_valueNcCs<||}||dd}|dkr(|S||d|SdS)a Get a single value from the YAML file. Args: path (str or list of str): single key, or list of nested keys default: any YAML-compatible value type Returns: the value from the file or the provided default Nrg)rareget)r@rdefaultrhrrr get_valueps zYamlFile.get_valuecCs||jS)z+Get the outermost value from the yaml file.)rSrI)r@rrrrQsz YamlFile.root)N)__name__ __module__ __qualname____doc__rAr&rPrNrOrSpropertyrTrUrVrWrXrYrZr[r\ classmethodrarerfrirkrnrQrrrrr=ms.<     r=)r )N)%rrZ __future__rrZ ruamel_yamlr"Zruamel_yaml.errorrZruamel_yaml.commentsrr ImportErrorZ ruamel.yamlr)Zruamel.yaml.errorZruamel.yaml.commentsrrJrr,r Z"anaconda_project.internal.makedirsrZ anaconda_project.internal.renamerZ$anaconda_project.internal.py2_compatr Z _YAMLErrorZ _CommentedMapZ _CommentedSeqrr'r*r3r8objectr=rrrrs8