o tBhH @sddlmZmZmZddlmZddlZddlmZddl m Z Gdddej Z e e e fd d Ze fd d Ze fd dZGdddejZeeejeeejeedejeeejddZdS))divisionabsolute_importprint_function) OrderedDictN)ConfigReadError) BASESTRINGcsFeZdZdZddZddZd ddZfd d Zed d Z Z S)LoaderzA customized YAML loader. This loader deviates from the official YAML spec in a few convenient ways: - All strings as are Unicode objects. - All maps are OrderedDicts. - Strings can begin with % without quotation. cCs ||SN)construct_scalar)selfnoderh/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/confuse/yaml_util.py_construct_unicodes zLoader._construct_unicodeccs&t}|V||}||dSr )rconstruct_mappingupdate)r r datavaluerrrconstruct_yaml_maps  zLoader.construct_yaml_mapFc Cst|tjr ||n tjddd|j|jt}|j D]6\}}|j ||d}zt |Wnt yI}z tjd|jd||jd}~ww|j ||d}|||<q|S)Nz%expected a mapping node, but found %s)deepzwhile constructing a mappingzfound unacceptable key (%s)) isinstanceyaml MappingNodeflatten_mapping constructorConstructorErrorid start_markrrconstruct_objecthash TypeError) r r rmappingkey_node value_nodekeyexcrrrrrs.     zLoader.construct_mappingcstt|}|p|dkS)N%)superr check_plainpeek)r plain __class__rrr)9szLoader.check_plaincCs.|dtj|dtj|dtjdS)zModify a PyYAML Loader class to add extra constructors for strings and maps. Call this method on a custom Loader class to make it behave like Confuse's own Loader ztag:yaml.org,2002:strztag:yaml.org,2002:mapztag:yaml.org,2002:omapN)add_constructorr rr)loaderrrradd_constructors=szLoader.add_constructors)F) __name__ __module__ __qualname____doc__rrrr) staticmethodr0 __classcell__rrr,rr s  r c Cslz t|d}tj||dWdWS1swYWdSttjjfy5}zt||d}~ww)aRead a YAML document from a file. If the file cannot be read or parsed, a ConfigReadError is raised. loader is the PyYAML Loader class to use to parse the YAML. By default, this is Confuse's own Loader class, which is like SafeLoader with extra constructors. rbr N)openrloadIOErrorerror YAMLErrorr)filenamer/fr&rrr load_yamlNs  ( r@c Cs8ztj||dWStjjy}zt||d}~ww)aRead a YAML document from a string. If the string cannot be parsed, a ConfigReadError is raised. `yaml_string` is a string to be parsed as a YAML document. `name` is the name to use in error messages. `loader` is the PyYAML Loader class to use to parse the YAML. By default, this is Confuse's own Loader class, which is like SafeLoader with extra constructors. r8N)rr:r<r=r) yaml_stringnamer/r&rrrload_yaml_string\s   rCcCsZt|ts|Sz|d}|tj|d}t||}||WStjjy,|YSw)aParse a value as if it were a YAML scalar to perform type conversion that is consistent with YAML documents. `value` should be a string. Non-string inputs or strings that raise YAML errors will be returned unchanged. `Loader` is the PyYAML Loader class to use for parsing, defaulting to Confuse's own Loader class. Examples with the default Loader: - '1' will return 1 as an integer - '1.0' will return 1 as a float - 'true' will return True - The empty string '' will return None )TF)rrresolver ScalarNoderr<r=)rr/tagr rrrparse_as_scalarks   rHcs:eZdZdZd ddZfddZddZd d ZZS) DumperzaA PyYAML Dumper that represents OrderedDicts as ordinary mappings (in order, of course). Nc Csg}tj|||d}|jdur||j|j<d}t|dr"t|}|D]+\}}||} ||} t| tj r;| j r=d}t| tj rF| j rHd}| | | fq$|durb|j dur_|j |_ |S||_ |S)N) flow_styleFitems)rr alias_keyrepresented_objectshasattrlistrKrepresent_datarrFstyleappenddefault_flow_stylerJ) r rGr"rJrr best_styleitem_key item_valuenode_key node_valuerrrrepresent_mappings2          zDumper.represent_mappingcsHtt||}t|}|jdur|dkrd|_|S|jdur"d|_|S)z~If a list has less than 4 items, represent it in inline style (i.e. comma separated, within square brackets). NTF)r(rIrepresent_listlenrSrJ)r rr lengthr,rrr[s zDumper.represent_listcCs|rd}nd}|d|S)zFRepresent bool as 'yes' or 'no' instead of 'true' or 'false'. yesnoztag:yaml.org,2002:boolrepresent_scalar)r rrrrrrepresent_bools zDumper.represent_boolcCs |ddS)z?Represent a None value with nothing instead of 'none'. ztag:yaml.org,2002:nullrDr`)r rrrrrepresent_nones zDumper.represent_noner ) r1r2r3r4rYr[rbrcr6rrr,rrIs     rIc Cst}t|}|D]5}|sd}n |drd|}nq t|}|r+|ds+n|d|7}q|dd}|||<q t|}d}|D]}|dd}||vr`|||7}|d|7}qK|S)zScan default_data for comments (we include empty lines in our definition of comments) and place them before the same keys in data. Only works with comments that are on one or more own lines, i.e. not next to a yaml mapping.  #z{0} T:rrD)dictiter splitlines startswithformatnextsplitstrip) r default_data comment_map default_lineslinecommentr% out_linesout_datarrrrestore_yaml_commentss0      rv) __future__rrr collectionsrr exceptionsrutilr SafeLoaderr r0r@rCrH SafeDumperrIadd_representerrepresent_dictboolrbtypercrOr[rvrrrrs     @   9