o )Ti- @sUdZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZdd lmZgd ZeZdad d Zd dZddZddZddZd`ddZ eeedZ!e"e#e fe$d<iZ%e"e#e&e'e(dffe$d<ddddZ)e"e#e*fe$d<iZ+e"e#e,e#fe$d<d e#fd!d"Z-dad#e e#d e&e'e(dffd$d%Z.d&d'Z/d(d)Z0e.erd*nd+Z1d,e#fd-d.Z2dbd,e#d/e*fd0d1Z3dddd2Z4ddd3Z5d4d5Z6d6d7Z7d8d9Z8d:d;Z9dZ;e"e#e fe$d?<er e7e;d@<dAdAdBdAdCZ M dcd#e#dNe e#gefdOe&e'e(dfdPe*fdQdRZ?d#e#fdSdTZ@dUe#dVe ege#ffdWdXZAdddZd[ZBe?d\e e.d\d]e'ed e'efd^d_ZCdS)ez$Code related to loading and dumping.N) HelpFormatter)suppress)AnyCallableOptional)load_value_mode parent_parser)import_jsonnetimport_toml_dumpsimport_toml_loadsomegaconf_supportpyyaml_availableruamel_support)ArgumentParser) get_loader set_loader set_dumpercCs|}|dkr dS|dkrdS|dkrdSz<|s(|dr-|ddr-t|WS|dd dd d ddd d rMd |vsHd|vrPt|WSWtSWtSty\YtSw) NtrueTfalseFnull-r.e)stripisdigit startswithintreplacefloat ValueError not_loadedvaluer&Q/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/_loaders_dumpers.py load_basics*" &  r(cCsjtrtSddl}Gdddt|d|j}dd}||d||d|dtd tjtd |atS) Nrc@s eZdZdS)z.get_yaml_default_loader..DefaultLoaderN)__name__ __module__ __qualname__r&r&r&r' DefaultLoader:sr, CSafeLoadercsFd|jvr |j|_|jD]\}}fdd|D|j|<qdS)Nyaml_implicit_resolverscs g|] \}}|kr||fqSr&r&).0tagregexp tag_to_remover&r' CszMget_yaml_default_loader..remove_implicit_resolver..)__dict__r.copyitems)clsr3 first_lettermappingsr&r2r'remove_implicit_resolver>s   z9get_yaml_default_loader..remove_implicit_resolverztag:yaml.org,2002:timestampztag:yaml.org,2002:floata ^(?: [-+]?(?:[0-9][0-9_]*)\.[0-9_]*(?:[eE][-+]?[0-9]+)? |[-+]?(?:[0-9][0-9_]*)(?:[eE][-+]?[0-9]+) |\.[0-9_]+(?:[eE][-+][0-9]+)? |[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+\.[0-9_]* |[-+]?\.(?:inf|Inf|INF) |\.(?:nan|NaN|NAN))$z -+0123456789.) yaml_default_loaderyamlgetattr SafeLoaderadd_implicit_resolverrecompileXlist)r=r,r;r&r&r'get_yaml_default_loader3s"  rEcCsddl}|j|td}t|trP|rPtdd|DrPt|dkr4|t t |dkr4|}|St |d dd  d }t|dkrP|t |krP|}|S) Nr)Loadercss|]}|duVqdSNr&)r/vr&r&r' azyaml_load..r:z {} r,)r=loadrE isinstancedictallvalueslenrnextitersetr split)streamr=r%keysr&r&r' yaml_load]s$$rZcCsddl}||SNr)jsonloads)r%r\r&r&r' json_loadks r^cCstd\}}||S)N toml_load)r )r% toml_loads_r&r&r'r_qs r_rc Csddlm}||\}}td}z|j||||d}Wt|StyAzt|WYSty@}ztt ||d}~www)Nr) ActionJsonnet jsonnet_load)ext_vars ext_codes) _jsonnetrbsplit_ext_varsr evaluate_snippet RuntimeErrorjson_or_yaml_loadjson_or_yaml_loader_exceptionsr"str)rXpathrdrbrerfvalexr&r&r'rcvs  rc)r=r\tomlloaders.loader_exceptionsTFloader_json_superset loader_paramsreturncCs.t}|durt}|dusJ|j}|SrG)rgetr parser_mode)modeparserr&r&r'get_load_value_modes  rzrxcCs|durt}|tvrK|dkrtdjft|<t|S|dkr+tdjft|<t|S|dkr omegaconf+ omegaconfcsi|] \}}|vr||qSr&r&)r/krHparamsr&r' szload_value..)rr(rzr#rsrvrrqset_omegaconf_loaderrtr7rOrr!boolrl)r%rkwargsrrxloader load_kwargsr&rr' load_values$  r)default_flow_style allow_unicode sort_keys) ensure_asciircCsddl}|j|fitSr[)r= safe_dumpdump_yaml_kwargs)datar=r&r&r' yaml_dumpsrcCs*td|}t|j}||j}||S)Nr=)dumpers#create_help_formatter_with_commentsformatter_classprogadd_yaml_comments)rrydumpr formatterr&r&r'yaml_comments_dumps    rcCsddl}|j|fdditS)Nr separators)rMrKr\dumpsdump_json_kwargsrr\r&r&r'json_compact_dumpsrcCs"ddl}|j|fdditdS)Nrindentr rrr&r&r'json_indented_dumpsrcCstd}||S)N toml_dump)r )r toml_dumpsr&r&r'rsr)r=r\ json_compact json_indentedrpr|r yaml_commentsz# z// )r=rr|rpcomment_prefix dump_formatcCs(|dhttvrtd|ddS)NrwzUnknown output format "z".)unionrVrr")rr&r&r'check_valid_dump_formatsrryr with_commentscs|dkr|jtvr |jnd}|r,|dtvr'|dkrtdtd|d|d}|dr5||fn|f}t||}|jr^t|r^t|dfdd |jD}|d|}|S) Nrwr= _commentsz7ruamel.yaml is required for dumping YAML with comments.z3Dumping with comments is not supported for format 'z'.rc3s|]}|VqdSrGr&)r/lineprefixr&r'rI0rJz$dump_using_format..)rwrr"r dump_headerrrvjoin)ryrrrargsrheaderr&rr'dump_using_format#s  rr& loader_fn exceptions json_supersetcCsH|t|<|t|<|t|<ttt|jdd}|r"|t|<dSdS)aSets the value loader function to be used when parsing with a certain mode. The ``loader_fn`` function must accept as input a single str type parameter and return any of the basic types {str, bool, int, float, list, dict, None}. If this function is not based on PyYAML for things to work correctly the exceptions types that can be raised when parsing a value fails should be provided. Args: mode: The parser mode for which to set its loader function. Example: "yaml". loader_fn: The loader function to set. Example: ``yaml.safe_load``. exceptions: Exceptions that the loader can raise when load fails. Example: (yaml.YAMLError,). json_superset: Whether the loader can load JSON data. rN) rqrrrsrVrDinspect signature parametersrt)rxrrrrr&r&r'r5s rcCst|S)z5Returns the current loader function for a given mode.)rqrr&r&r'rRsr format_name dumper_fncCs |t|<dS)zSets the dumping function for a given format name. Args: format_name: Name to use for dumping with this function. Example: ``yaml_custom``. dumper_fn: The dumper function to set. Example: ``yaml.safe_dump``. N)r)rrr&r&r'rWs rrcCs<tr|tvrddlm}||}t||tddSdSdS)Nr)get_omegaconf_loaderr=)r rq _optionalsrrr{)rxrrr&r&r'ras  rr|rcs$ddlmGfddd|}|S)aCreates a dynamic class that combines a formatter with YAML comment functionality. Args: formatter_class: The base formatter class to extend. Returns: A new class that inherits from both the formatter and YAMLCommentFormatter. rYAMLCommentFormattercs0eZdZfddZdedefddZZS)zAcreate_help_formatter_with_comments..DynamicHelpFormattercs tj|i|||_dSrG)super__init___yaml_formatter)selfrr)r __class__r&r'rxszJcreate_help_formatter_with_comments..DynamicHelpFormatter.__init__cfgrucSs |j|S)z Adds help text as yaml comments.)rr)rrr&r&r'r|s zScreate_help_formatter_with_comments..DynamicHelpFormatter.add_yaml_comments)r)r*r+rrlr __classcell__r&r)rr'DynamicHelpFormatterwsr) _formattersr)rrr&rr'rls  r)rNrG)F)r&T)r)D__doc__rrAargparser contextlibrtypingrrr_commonrr rr r r r rr_type_checkingr__all__objectr#r<r(rErZr^r_rcrqrPrl__annotations__rrtupletype ExceptionrsrrtrVrzr{rjrrkrrrrrrrrrrrrrrrrrrr&r&r&r's    *  $