o )Ti@sdZddlmZmZmZmZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZmZmZmZmZdd lmZdd lmZdd lmZmZmZd gZGd d d e ZdS)zActions to support jsonnet.)AnyDictOptionalUnion)_is_action_value_list)Actionparser_context)ActionJsonSchema)get_loader_exceptions load_value)_get_config_read_modeget_jsonschema_exceptionsimport_jsonnetimport_jsonschemapyyaml_available) find_action)ActionTypeHint)NoneTypePathargument_error ActionJsonnetc seZdZdZ  ddeedeeeefffdd ZddZ d d Z e d d Z e deeee fd eeee feee fffddZ  ddeeefdeeee fded efddZZS)rzGAction to parse a Jsonnet, optionally validating against a JSON Schema.Next_varsschemac sd|vrqtdt|ttfstd||_|durltdd}t|tr]tr)dnd}t|d%zt |}Wnt |yM}ztd ||d}~wwWdn1sXwY| |t |||_dSd|_dS|d |_|d|_tjd i|dS) aaInitializer for ActionJsonnet instance. Args: ext_vars: Key where to find the external variables required to parse the Jsonnet. schema: Schema to validate values against. Raises: ValueError: If a parameter is invalid. jsonschema.exceptions.SchemaError: If the schema is invalid. _validatorrz+ext_vars has to be either None or a string.Nryamljsonload_value_modezProblems parsing schema: _ext_vars)r isinstancestrr ValueErrorrrrr r r check_schemar "_extend_jsonvalidator_with_defaultrpopsuper__init__)selfrrkwargs jsonvalidatormodeex __class__r I/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/_jsonnet.pyr(s0         zActionJsonnet.__init__cOst|dkr9|j|d<|j|d<d|vr2d|dvr2|jdur2ddl}|d|j|jjdd|d<td i|St|d |j|j |d |d d dS) zParses an argument as Jsonnet using ``ext_vars`` if defined. Raises: TypeError: If the argument is not valid. rrrhelpz%sNT) sort_keysr)cfgr ) lenrrrdumpsrrsetattrdest _check_type_)r)argsr*rr r r0__call__As   $zActionJsonnet.__call__c Cst|}i}|r||ji}|s|g}t|D]R\}}zt|tr+|j||dd}n |jdur6|j||||<Wqt t ft t yj}z|sOdndt|d}t d|j d|d||d}~ww|ro|S|d S) NT)r with_metaz element rz Parser key ""z: r)rgetr enumerater!r"parservalidate TypeError RuntimeErrorrr r8) r)valuer4islistrnumvalr-elemr r r0 _check_typeRs&    zActionJsonnet._check_typecCst|trQ|jrSt||j}|std|jdt|to |j}|tthvr2td|jd||j dur:i|_ t|j tsLtd|jd|j d|_ dSdSdS)Nz No argument found for ext_vars=''zType for ext_vars='z ' argument must be dict, given: z Default value for the ext_vars='z(' argument must be dict or None, given: T) r!rrrr#r _typehintdictrdefaultjsonnet_ext_vars)parseractionext_vars_action ext_vars_typer r r0_check_ext_vars_actiones&      z$ActionJsonnet._check_ext_vars_actionreturncsD|duri}ddlfdd|D}dd|D}||fS)zSplits an ``ext_vars`` dict into the ``ext_codes`` and ``ext_vars`` required by Jsonnet. Args: ext_vars: External variables. Values can be strings or any other basic type. Nrcs&i|]\}}t|ts||qSr )r!r"r6.0kvrr r0 s&z0ActionJsonnet.split_ext_vars..cSs i|] \}}t|tr||qSr )r!r"rVr r r0r[s )ritemsr ext_codesr rZr0split_ext_varsys zActionJsonnet.split_ext_varsFjsonnetr<c Cstd}||\}}d}d}|}z t|td}Wn ty#Ynwt|tr.|ddn|}|}z&ttr:dndd t |j ||||d } Wdn1sTwYWnt yq} z t d |d | | d} ~ ww|j dur}|j | |r|dur|| d <|| d<| S)aMethod that can be used to parse Jsonnet independent from an :class:`.ArgumentParser`. Args: jsonnet: Either a path to a Jsonnet file or the Jsonnet content. ext_vars: External variables. Values can be strings or any other basic type. with_meta: Whether to include metadata in config object. Returns: The parsed Jsonnet object. Raises: TypeError: If the input is neither a path to an existent file nor a Jsonnet. rNsnippet)r,F)absoluterrrr]zProblems evaluating Jsonnet 'z': __path____orig__)rr_rr rCr! get_contentr rr evaluate_snippetrDrrrB) r)r`rr<_jsonnetr^fpathfnameravaluesr-r r r0rAs8   zActionJsonnet.parse)NN)NF)__name__ __module__ __qualname____doc__rr"rrMr(r;rJ staticmethodrTrtupler_rboolrA __classcell__r r r.r0rs4' 8 N) rntypingrrrr_actionsr_commonrr _jsonschemar _loaders_dumpersr r _optionalsr rrrr _subcommandsr _typehintsr_utilrrr__all__rr r r r0s