o ?i~@sddlZddlZddlmZmZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZddlmZdd lmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/dd l0m1Z1dd l2m3Z3dd l4m5Z5ee6e7e8e e9e:fZ;eGdddZGddde>Z?Gddde?Z@Gddde ZAGddde?ZBdS)N)ABCabstractmethod) defaultdict) dataclassfield)Enum) AnyDictIteratorListOptionalSetTupleTypeUnion)ParserRuleContext) _DEFAULT_MARKER_NoneType ValueKind _get_value_is_interpolation_is_missing_value _is_specialformat_and_raiseget_value_kindis_union_annotationis_valid_value_annotation split_keytype_str) ConfigKeyErrorConfigTypeErrorInterpolationKeyErrorInterpolationResolutionError InterpolationToMissingValueErrorInterpolationValidationErrorMissingMandatoryValueUnsupportedInterpolationTypeValidationError)OmegaConfGrammarParser)parse)GrammarVisitorc@seZdZUeeeefed<eeeefed<eed<eed<dZe e e efed<dZ eed<e d d d Ze e efed <dddZed eeeeffddZdS)Metadataref_type object_typeoptionalkeyNflagsF flags_rootcCsttSN)rdictr5r5K/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/omegaconf/base.pyAszMetadata.)default_factoryresolver_cachereturncCs|jdur i|_dSdSr3)r1selfr5r5r6 __post_init__Cs  zMetadata.__post_init__cCs|jrt|jS|jS)zr5r5r5r6r,,s    r,c@s0eZdZUdZeed<dZeed<dddZdS)ContainerMetadataNkey_type element_typer:cCsp|jdurt|_|jtust|jtsJ|jdur,t|js,tdt|jddd|j dur6i|_ dSdS)NzUnsupported value type: 'T)include_module_name') r-rrH isinstancetyperIrr(rr1r;r5r5r6r=Xs     zContainerMetadata.__post_init__r?)r@rArBrHrrCrIr=r5r5r5r6rGSs  rGc @seZdZUeed<eded<eeeeefed<deddefddZ d eee ffd d Z d eee fd d fddZ dedd d fddZ dVddZd edfddZd edfddZdeeeefdeeeeeefd dfddZded eefddZded eefd d!Zded eefd"d#Z dWd$e d%e d&ed'eed(e d d f d)d*Zed$eeeefd efd+d,ZdXd-d.Z / dYd0ed1eeed edfd2d3Z dZd0ed1eeed edfd4d5Z d[d6d7Z!d efd8d9Z"d efd:d;Z#edZ$edrz Node.__hash__cCr~r3r5r;r5r5r6rBrz Node._valuecCr~r3r5)r<rpr1r5r5r6 _set_valueFrzNode._set_valuecCr~r3r5r;r5r5r6 _is_optionalJrzNode._is_optionalcCr~r3r5r;r5r5r6rNrzNode._is_interpolationcC|jjSr3rOr0r;r5r5r6rRz Node._keycCs ||j_dSr3rrr5r5r6_set_keyU z Node._set_keycCrr3)rOr2r;r5r5r6rxXrzNode._is_flags_rootr2cCs$|jj|kr||j_|dSdSr3)rOr2ra)r<r2r5r5r6_set_flags_root[s  zNode._set_flags_rootcCs |jjtuSr3)rOr-rr;r5r5r6 _has_ref_type`rzNode._has_ref_typer?NN)r:rN)FNr3)r:rf)0r@rArBr,rCr r rErDrXrr]r_rcrarerirr rqrsrwrt Exceptionr}r DictKeyTypeintrrr rrrrrrrrrrrrrrrxrrr5r5r5r6rNfs              &rNcsPeZdZUdZeed<deddeffdd Zdefdd Z d d d Z Z S)rPz Base class for nodes that can contain other nodes. Concrete subclasses include DictConfig, ListConfig, and UnionNode. _contentrSrTcstj||dd|jd<dS)NrSrTr)superrXrVrW __class__r5r6rXlsz Box.__init__r:cCs t|Sr3)rZdeepcopyr;r5r5r6__copy__ps z Box.__copy__NcCsddlm}ddlm}t||r=|jd}t|tr9|jdD]\}}|dur/||t|t r8| q"dSdSt||rj|jd}t|t rf|jdD]}|dur\||t|t re| qQdSdSt|t r|jd}t|t r||t|t r| dSdSdSdS)Nr DictConfig ListConfigr) dictconfigr listconfigrrLrVr4itemsrcrP _re_parentlistrgrN)r<rrcontentrrpitemr5r5r6rtsB                  zBox._re_parentr?) r@rArB__doc__rrCr r,rXrr __classcell__r5r5rr6rPds rPc@seZdZUdZeed<e    dd#e"d)e de eede ed ef d8d9Z#d?d:d;Z$dS)@rfz% Container tagging interface rOTFr0validate_access validate_keythrow_on_missing_valuethrow_on_missing_keyr:cCr~r3r5r<r0rrrrr5r5r6 _get_child zContainer._get_childcCr~r3r5rr5r5r6 _get_noderzContainer._get_nodeNcCr~r3r5rr5r5r6 __delitem__rzContainer.__delitem__rpcCr~r3r5)r<r0rpr5r5r6 __setitem__rzContainer.__setitem__cCr~r3r5r;r5r5r6__iter__rzContainer.__iter__ key_or_indexcCr~r3r5)r<rr5r5r6 __getitem__rzContainer.__getitem__cCs||}|ds ||fS|}|dsJ |dusJ|dd}|ds- ||fS|}|dur=td|dq)N.TrzError resolving key 'rK) startswithrrir )r<r0origrr5r5r6_resolve_key_and_roots     zContainer._resolve_key_and_rootthrow_on_missingrrc Csddlm}|dkr|d|fSt|}|}tt|dD]L}|dur%nE||} ||| ||d\} } t| tr?| j||d} | durgt| tsgd |d|d} ||d} t d |d | d | d | }q|durpd S|d}|||||d\}} |dur||dfS|durt |}||vrt d| |z|j|||||d}W|dur||n |dur||ww|||fS)zA Select a value using dot separated key sequence r) _select_oneN)cr0rthrow_on_type_errorrrrzError trying to access z: node `z.` is not a container and thus cannot contain ``)NNNz Recursive interpolation detected)rSr0rprr) omegaconfrrrangerkrLrNrrfjoinr!idr#add_maybe_resolve_interpolationremove)r<r0rrrrsplitrikrv_ parent_key child_keylast_keyrpvidr5r5r6 _select_implsx          zContainer._select_implrSrNrcCsFz |j||||d}Wn ty|rYdSw|j|||||dS)a Resolve an interpolation. This happens in two steps: 1. The parse tree is visited, which outputs either a `Node` (e.g., for node interpolations "${foo}"), a string (e.g., for string interpolations "hello ${name}", or any other arbitrary value (e.g., or custom interpolations "${foo:bar}"). 2. This output is potentially validated and converted when the node being resolved (`value`) is typed. If an error occurs in one of the above steps, an `InterpolationResolutionError` (or a subclass of it) is raised, *unless* `throw_on_resolution_failure` is set to `False` (in which case the return value is `None`). :param parent: Parent of the node being resolved. :param value: Node being resolved. :param key: The associated key in the parent. :param parse_tree: The parse tree as obtained from `grammar_parser.parse()`. :param throw_on_resolution_failure: If `False`, then exceptions raised during the resolution of the interpolation are silenced, and instead `None` is returned. :return: A `Node` that contains the interpolation result. This may be an existing node in the config (in the case of a node interpolation "${foo}"), or a new node that is created to wrap the interpolated value. It is `None` if and only if `throw_on_resolution_failure` is `False` and an error occurs during resolution. )rr|r0rN)rSrpr0resolvedr)resolve_parse_treer#*_validate_and_convert_interpolation_result)r<rSrpr0rrrrr5r5r6r!s & z0Container._resolve_interpolation_from_parse_treerc Csddlm}m}m}t|t } t||rWt||sWt|} z|| } Wn%tyJ} z|r?|j || | d|d| t dWYd} ~ dSd} ~ wwt | t | krWd} | }| r`||||dSt|tsgJ|S)Nr)AnyNodeInterpolationResultNode ValueNodez#While dereferencing interpolation 'z': )r0rpryrzr{T)rpr0rS) nodesrrrrLrNrvalidate_and_convertr(r}r%rM) r<rSrpr0rrrrr must_wrap res_value conv_valueer5r5r6rXs2 z4Container._validate_and_convert_interpolation_resultr|targetcCs4|}|dur||urtd|}|dusdSdS)Nz%Interpolation to parent node detected)r#re)r<r|rrSr5r5r6%_validate_not_dereferencing_to_parentsz/Container._validate_not_dereferencing_to_parent inter_keyc Csz ||\}}Wnty#}ztd|tdd}~wwz|j|dd|d\}}}WntyL}ztd|tdd}~ww|dusU|dur]td|d|j ||d |S) z0A node interpolation is of the form `${foo.bar}`z.ConfigKeyError while resolving interpolation: NT)rrrz5MissingMandatoryValue while resolving interpolation: zInterpolation key 'z ' not found)r|r) rr r"with_tracebacksysexc_inforr&r$r)r<rr root_nodeexcrSrrpr5r5r6_resolve_node_interpolations:z%Container._resolve_node_interpolation inter_type inter_args.inter_args_strc CsDddlm}||}|dur|}||||||Std|)Nr) OmegaConfzUnsupported interpolation type )rr _get_resolverrr') r<r0r|rrrrresolverrr5r5r6_evaluate_custom_resolvers  z#Container._evaluate_custom_resolvercCsJt|}|tjkr |Stt|}|j||||||dur |dStdS)N)rSrpr0rrr)rr INTERPOLATIONr*rrset)r<rSr0rprr value_kindrr5r5r6rs   z&Container._maybe_resolve_interpolationc sdtdtttdtdffdd }dtdttd fd ttd fdtffd d }t|||d }z||WStyAt y_}ztt |j d| t dd}~ww)z Resolve a given parse tree into its value. We make no assumption here on the type of the tree's root, so that the return value may be of any type. rrr:rNcsj||dS)Nrr)rrr;r5r6node_interpolation_callbackszAContainer.resolve_parse_tree..node_interpolation_callbacknameargs.args_strcsj|||dS)N)r0r|rrr)r)rrrr0r|r<r5r6resolver_interpolation_callbackszEContainer.resolve_parse_tree..resolver_interpolation_callback)rrrz' raised while resolving interpolation: rN)rEr r rrrr+visitr#rrMr@rrr) r<rr|rr0rrvisitorrr5rr6rsB      zContainer.resolve_parse_treecCsddlm}ddlm}|jddurTd|jd<t||r9|jd}t|tr5|jdD]}|q.dSdSt||rV|jd}t|t rX|jdD] }|qMdSdSdSdS)NrrrrRr) rrrrrVrLr4rjrar)r<rrrrprr5r5r6ras(           z!Container._invalidate_flags_cache)TTFFr3rr?)%r@rArBrrGrCrrrDrr rNr rrrrr rrrErrr rrr)ConfigValueContextrrrrrrrrrar5r5r5r6rfs       N  7 +          -rfc@seZdZdZdZdZdS)SCModerrN)r@rArBDICT DICT_CONFIG INSTANTIATEr5r5r5r6r sr c sjeZdZUdZeeed<eede fed<   d*de de de d e d ee d df fd d Z d eeeefd e fddZde d e fddZde d e fddZd efddZd eede ffddZd+de deee e fd dfddZ d+de deee e fd dfddZd e fdd Zd e fd!d"Zd e fd#d$Zd e fd%d&Zd'eee fd dfd(d)ZZS),rga This class handles Union type hints. The `_content` attribute is either a child node that is compatible with the given Union ref_type, or it is a special value (None or MISSING or interpolation). Much of the logic for e.g. value assignment and type validation is delegated to the child node. As such, UnionNode functions as a "pass-through" node. User apps and downstream libraries should not need to know about UnionNode (assuming they only use OmegaConf's public API). rQNrTrr- is_optionalr0rSr:c sz4t|sd|dd}t|t|ttfstdtj|t|d||ddidd| |WdSt yR}zt d||t ||d WYd}~dSd}~ww) Nz"UnionNode got unexpected ref_type z. Please file a bugz3 report at https://github.com/omry/omegaconf/issuesz&Parent type is not omegaconf.ContainerconvertF)r-r.r/r0r1r)r|r0rprzry) rAssertionErrorrLrfrr!rrXr,rrrrE)r<rr-rr0rSrzexrr5r6rX.s0  $zUnionNode.__init__cCs:|}|dur|jjdurdSt|jjS||jjS)Nr)rerOr0rEr)r<r0rSr5r5r6rMs   zUnionNode._get_full_keyrcCsZ|jd}t|tr||}nt|tr||}n||}t|tttfs+J|SNr)rVrLrNrrDrMNotImplemented)r<rrrvr5r5r6rWs      zUnionNode.__eq__cCs||}|tur tS| Sr3)rr)r<rxr5r5r6rbs zUnionNode.__ne__cCt|jdSr)hashrVr;r5r5r6rhrdzUnionNode.__hash__cCs"|jd}t|tttfsJ|Sr)rVrLrNrrE)r<rr5r5r6rks zUnionNode._valuerpr1c CsX|jd}|jd}z |||WdSty+}z ||jd<||jd<|d}~ww)NrrO)rV_set_value_implr)r<rpr1previous_contentprevious_metadatarr5r5r6rps    zUnionNode._set_valuec Csddlm}|jj}|jj}t|}t|r8t|tt fsJ|dur1| s1t dt |d||j d<dSt|trFt dt ||jD]}z|||dd|d|j d<WdSt ydYqIwt d t |d) Nr) _node_wrapz/Value '$VALUE' is incompatible with type hint 'rKrz:Cannot assign container '$VALUE' of type '$VALUE_TYPE' to F)rpr-rr0rSzEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint ')omegaconf.omegaconfrrOr-r>rrrLrErrr(rrVrf__args__)r<rpr1rr-r>candidate_ref_typer5r5r6rzs@      zUnionNode._set_value_implcCs|jdjduS)NrOT)rVr/r;r5r5r6rszUnionNode._is_optionalcCrr)rrVr;r5r5r6rrdzUnionNode._is_interpolationcCrr)rErVr;r5r5r6__str__rdzUnionNode.__str__cCrr)reprrVr;r5r5r6__repr__rdzUnionNode.__repr__rc Cstt|}|jD]\}}|dvrtj||d|j|<q |jd}t|trM|jd}zd|jd<tj||d}||jd<W||jd<n||jd<w|}||jd<|jd|jd<|S)N)rrQ)rrrQ) object__new__rMrVrrZrrLrN)r<rresr0rp src_content old_parent content_copyr5r5r6 __deepcopy__s"      zUnionNode.__deepcopy__)TNNr3)r@rArBrr rfrCrrNrErrDrPrXrrrrrrrr rrrrrr r'rr5r5rr6rgsN    $  '"rg)CrZrabcrr collectionsr dataclassesrrenumrtypingrr r r r r rrrantlr4r_utilsrrrrrrrrrrrrrerrorsr r!r"r#r$r%r&r'r("grammar.gen.OmegaConfGrammarParserr)grammar_parserr*grammar_visitorr+rEbytesrfloatrDrr,rGrNrPrfr rgr5r5r5r6s6  , <,  &0