a d~@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 <dd ddZeeeeefd ddZdS)Metadataref_type object_typeoptionalkeyNflagsF flags_rootcCsttSN)rdictr5r5W/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/omegaconf/base.pyAzMetadata.)default_factoryresolver_cachereturncCs|jduri|_dSr3)r1selfr5r5r6 __post_init__Cs zMetadata.__post_init__cCs|jrt|jS|jSdS)zd?Z'edUe eeeefd d@dAdBZ(eeddCdDZ)eeddEdFZ*e ddGdHZ+e d d&dIdJZ,eddKdLZ-ed dMdNdOZ.eddPdQZ/d S)VNode _metadataBox_parent _flags_cacheparentmetadatacCs"||jd<||jd<d|jd<dS)NrOrQrR__dict__r>rTrUr5r5r6__init__ls  z Node.__init__r;cCst|j}|d=|SNrR)copyrWr> state_dictr5r5r6 __getstate__qs zNode.__getstate__N)r]r<cCs|j|d|jd<dSrZ)rWupdater\r5r5r6 __setstate__ws zNode.__setstate__)rTr<cCs,|dust|tsJ||jd<|dSNrQ)rLrPrW_invalidate_flags_cacher>rTr5r5r6 _set_parent{s zNode._set_parentcCsd|jd<dSrZrVr=r5r5r6rbszNode._invalidate_flags_cachecCs$|jd}|dus t|ts J|Sra)rWrLrPrcr5r5r6 _get_parents zNode._get_parent ContainercCsl|jd}|dus t|ts Jt|trN|jd}|dusJt|tsJJ|S|dusdt|tsdJ|SdS)z{ Like _get_parent, but returns the grandparent in the case where `self` is wrapped by a UnionNode. rQN)rWrLrP UnionNoderf)r>rTZ grandparentr5r5r6_get_parent_containers   zNode._get_parent_container)r1valuesr<cCst|tr|g}|dus"t|tr(|g}t|dkr@t||}t|t|krXtdt|D]\\}}||}|dur|jjdusJ||jjvr|jj|=q`|jjdusJ||jj|<q`||S)Nrz3Inconsistent lengths of input flag names and values) rLrFrElen ValueError enumeraterOr1rb)r>r1riidxflagvaluer5r5r6 _set_flags$     zNode._set_flag)rnr<cCs|jjdusJ|jj|S)zc :param flag: flag to inspect :return: the state of the flag on this node. N)rOr1get)r>rnr5r5r6_get_node_flagszNode._get_node_flagcCs`|jd}|dur i}|jd<||t}|turF||}|||<|dus\t|ts\J|SrZ)rWrqr_get_flag_no_cacherLrE)r>rncacheretr5r5r6 _get_flags   zNode._get_flagcCs^|jj}|dusJ||vr0||dur0||S|rrnr1rTr5r5r6rss zNode._get_flag_no_cache)r0rocausemsg type_overrider<cCs0t||||durt|n|||dds,JdS)N)noder0roryrxrzF)rrF)r>r0rorxryrzr5r5r6_format_and_raiseszNode._format_and_raiser0r<cCsdSr3r5r>r0r5r5r6 _get_full_keyszNode._get_full_keycCs|jdd}|dusJ|S)NT)throw_on_resolution_failure_dereference_node_impl)r>r{r5r5r6_dereference_nodes  zNode._dereference_nodeF)rmemor<cCs|j||dS)Nrrr)r>rrr5r5r6_maybe_dereference_nodeszNode._maybe_dereference_nodecCs^|s |S|}|dur,|r(tddS|dus8J|}|j|||tt|||dS)Nz8Cannot resolve interpolation for a node without a parent)rTr0ro parse_treerr)rrhr#_key&_resolve_interpolation_from_parse_treer*r)r>rrrTr0r5r5r6rs&  zNode._dereference_node_implcCs~|}|dur"t|tsJ|S|dur4t|ts8J|durd|}|dur^t|ts8Jq8|durvt|tszJ|Sr3)rerLrfrP)r>rootr5r5r6 _get_roots zNode._get_rootcCst|S)zY Check if the node's value is `???` (does *not* resolve interpolations). )rr=r5r5r6 _is_missing*szNode._is_missingcCs |duS)zZ Check if the node's value is `None` (does *not* resolve interpolations). N)_valuer=r5r5r6_is_none0sz Node._is_noneotherr<cCsdSr3r5r>rr5r5r6__eq__6sz Node.__eq__cCsdSr3r5rr5r5r6__ne__:sz Node.__ne__cCsdSr3r5r=r5r5r6__hash__>sz Node.__hash__cCsdSr3r5r=r5r5r6rBsz Node._valueror1r<cCsdSr3r5)r>ror1r5r5r6 _set_valueFszNode._set_valuecCsdSr3r5r=r5r5r6 _is_optionalJszNode._is_optionalcCsdSr3r5r=r5r5r6rNszNode._is_interpolationcCs|jjSr3rOr0r=r5r5r6rRsz Node._keycCs ||j_dSr3rr~r5r5r6_set_keyUsz Node._set_keycCs|jjSr3)rOr2r=r5r5r6rwXszNode._is_flags_root)r2r<cCs |jj|kr||j_|dSr3)rOr2rb)r>r2r5r5r6_set_flags_root[s zNode._set_flags_rootcCs |jjtuSr3)rOr-rr=r5r5r6 _has_ref_type`szNode._has_ref_type)NN)FN)N)N)0rArBrCr,rDr r rFrErYrr^r`rdrbrerhrr rprrrvrs Exceptionr|r DictKeyTypeintrrr rrrrrrrrrrrrrrrwrrr5r5r5r6rNfs          "rNcsReZdZUdZeed<ededfdd ZedddZ d dd d Z Z S) rPz Base class for nodes that can contain other nodes. Concrete subclasses include DictConfig, ListConfig, and UnionNode. _contentrScstj||dd|jd<dS)NrSr)superrYrWrX __class__r5r6rYlsz Box.__init__r;cCs t|Sr3)r[deepcopyr=r5r5r6__copy__psz Box.__copy__NcCs ddlm}ddlm}t||rt|jd}t|trr|jdD],\}}|dur^||t|t rD| qDnt||r|jd}t|t r|jdD](}|dur||t|t r| qn@t|t r|jd}t|t r||t|t r| dS)Nr DictConfig ListConfigr) dictconfigr listconfigrrLrWr4itemsrdrP _re_parentlistrgrN)r>rrcontentrroitemr5r5r6rts0                   zBox._re_parent) rArBrC__doc__rrDr r,rYrr __classcell__r5r5rr6rPds rPc @seZdZUdZeed<ed5eeeeee e e e e e fdddZ ed6eeeeee e e e e e fddd Zeed d d d Zeeed dddZeeedddZeeedddZeedefd ddZd7eeee eeee de ee e fdddZe ddeejee eee ddddZe ddeeee dd d!d"Ze e d d#d$d%Zee eedd&d'd(Zee eeed)feed)fed*d+d,Z d8e dee ee eee e d-d.d/Z!d9e"e e eee eed0d1d2Z#d dd3d4Z$d S):rfz% Container tagging interface rOTF)r0validate_access validate_keythrow_on_missing_valuethrow_on_missing_keyr<cCsdSr3r5r>r0rrrrr5r5r6 _get_childs zContainer._get_childcCsdSr3r5rr5r5r6 _get_nodes zContainer._get_nodeNr}cCsdSr3r5r~r5r5r6 __delitem__szContainer.__delitem__)r0ror<cCsdSr3r5)r>r0ror5r5r6 __setitem__szContainer.__setitem__r;cCsdSr3r5r=r5r5r6__iter__szContainer.__iter__) key_or_indexr<cCsdSr3r5)r>rr5r5r6 __getitem__szContainer.__getitem__cCs~|}|ds||fS|}|ds,J|dus8J|dd}|dsPqr|}|dur,td|dq,||fSdS)N.rzError resolving key 'rK) startswithrrhr )r>r0origrr5r5r6_resolve_key_and_roots     zContainer._resolve_key_and_root)r0throw_on_missingrrr<c Csddlm}|dkr|d|fSt|}|}tt|dD]}|durJq||} ||| ||d\} } t| tr~| j||d} | durt| tsd |d|d} ||d} t d |d | d | d | }q:|durd S|d}|||||d\}} |dur||dfS|dur>t |}||vr4t d| |z,|j|||||d}W|dur||n|dur||0|||fS)zA Select a value using dot separated key sequence r) _select_oneN)cr0rZthrow_on_type_errorrrrzError trying to access z: node `z.` is not a container and thus cannot contain ``)NNNz Recursive interpolation detected)rTr0rorr) omegaconfrrrangerjrLrNrrfjoinr!idr#add_maybe_resolve_interpolationremove)r>r0rrrrsplitrikru_Z parent_keyZ child_keylast_keyroZvidr5r5r6 _select_implst              zContainer._select_implrN)rTror0rrrr<cCsFz|j||||d}Wnty0|r*YdS0|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)rTror0resolvedr)resolve_parse_treer#*_validate_and_convert_interpolation_result)r>rTror0rrrrr5r5r6r!s&  z0Container._resolve_interpolation_from_parse_tree)rTror0rrr<c Csddlm}m}m}t|t } t||rt||st|} z|| } WnJty} z2|r~|j || | d|d| t dWYd} ~ dSd} ~ 00t | t | krd} | }| r||||dSt|tsJ|SdS)Nr)AnyNodeInterpolationResultNode ValueNodez#While dereferencing interpolation 'z': )r0rorxryrzT)ror0rT) nodesrrrrLrNrZvalidate_and_convertr(r|r%rM) r>rTror0rrrrrZ must_wrapZ res_valueZ conv_valueer5r5r6rXs. z4Container._validate_and_convert_interpolation_result)r{targetr<cCs*|}|dur&||urtd|}qdS)Nz%Interpolation to parent node detected)r#re)r>r{rrTr5r5r6%_validate_not_dereferencing_to_parentsz/Container._validate_not_dereferencing_to_parent inter_keyrr<c Csz||\}}Wn@tyR}z(td|tdWYd}~n d}~00z|j|dd|d\}}}Wn@ty}z(td|tdWYd}~n d}~00|dus|durtd|dn|j ||d |SdS) 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_nodeexcrTrror5r5r6_resolve_node_interpolations2  z%Container._resolve_node_interpolation.)r0r{ inter_type inter_argsinter_args_strr<c CsHddlm}||}|dur6|}||||||Std|dS)Nr) OmegaConfzUnsupported interpolation type )rrZ _get_resolverrr') r>r0r{rrrrresolverrr5r5r6_evaluate_custom_resolvers  z#Container._evaluate_custom_resolver)rTr0rorrr<cCsFt|}|tjkr|Stt|}|j||||||dur<|ntdS)N)rTror0rrr)rrZ INTERPOLATIONr*rrset)r>rTr0rorrZ value_kindrr5r5r6rs  z&Container._maybe_resolve_interpolation)rr{rr0r<c stttttddfdd }tttdfttdftdfdd }t|||d }z ||WSty|YnJt y}z2tt |j d | t d WYd }~n d }~00d S) 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. rNrcsj||dS)Nrr)rrr=r5r6node_interpolation_callbackszAContainer.resolve_parse_tree..node_interpolation_callback.)nameargsargs_strr<csj|||dS)N)r0r{rrr)r)rrrr0r{r>r5r6resolver_interpolation_callbackszEContainer.resolve_parse_tree..resolver_interpolation_callback)rrrz' raised while resolving interpolation: rN)rFr r rrrr+visitr#rrMrArrr) r>rr{rr0rrvisitorrr5rr6rs*     zContainer.resolve_parse_treecCsddlm}ddlm}|jddurd|jd<t||rl|jd}t|tr|jdD] }|q\n6t||r|jd}t|t r|jdD] }|qdS)NrrrrRr) rrrrrWrLr4rirbr)r>rrrrorr5r5r6rbs          z!Container._invalidate_flags_cache)TTFF)TTFF)N)N)NN)%rArBrCrrHrDrrrErr rNr rrrrr rrrFrrr rrr)ZConfigValueContextrrrrrrrrrbr5r5r5r6rfs     P  9 +         -rfc@seZdZdZdZdZdS)SCModerrN)rArBrCDICTZ DICT_CONFIGZ INSTANTIATEr5r5r5r6rsrcsHeZdZUdZeeed<eede fed<d&e e e e ee ddfdd Z eeeefe d d d Ze e d d dZe e d ddZedddZeede fdddZd'e eee e fddddZd(e eee e fddddZe dddZe dddZe ddd Ze dd!d"Zeee fdd#d$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). rQNrT)rr- is_optionalr0rTr<c szft|s"d|dd}t|t|ttfs8tdtj|t|d||ddidd| |Wn:t y}z"t d||t ||d WYd}~n d}~00dS) 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/r0r1rS)r{r0roryrx) rAssertionErrorrLrfrr!rrYr,rrrrF)r>rr-rr0rTryexrr5r6rY.s,  zUnionNode.__init__r}cCs@|}|dur.|jjdur dSt|jjSn||jjSdS)Nr)rerOr0rFr)r>r0rTr5r5r6rMs  zUnionNode._get_full_keyrcCsZ|jd}t|tr ||}n t|tr6||}n ||}t|tttfsVJ|SNr)rWrLrNrrErMNotImplemented)r>rrrur5r5r6rWs      zUnionNode.__eq__cCs||}|turtS| Sr3)rr)r>rxr5r5r6rbs zUnionNode.__ne__r;cCst|jdSr)hashrWr=r5r5r6rhszUnionNode.__hash__cCs"|jd}t|tttfsJ|Sr)rWrLrNrrF)r>rr5r5r6rks zUnionNode._valuerc Csf|jd}|jd}z|||Wn<ty`}z$||jd<||jd<|WYd}~n d}~00dS)NrrO)rW_set_value_implr)r>ror1Zprevious_contentZprevious_metadatarr5r5r6rps    zUnionNode._set_valuec Csddlm}|jj}|jj}t|}t|rnt|tt fs>J|durb| sbt dt |d||j d<nxt|trt dt |nZ|jD]>}z"|||dd|d|j d<WqWqt yYqYq0qt d t |ddS) Nr) _node_wrapz/Value '$VALUE' is incompatible with type hint 'rKrz:Cannot assign container '$VALUE' of type '$VALUE_TYPE' to F)ror-rr0rTzEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint ')Zomegaconf.omegaconfrrOr-r@rrrLrFrrr(rrWrf__args__)r>ror1rr-r@Zcandidate_ref_typer5r5r6rzs>         zUnionNode._set_value_implcCs|jdjduS)NrOT)rWr/r=r5r5r6rszUnionNode._is_optionalcCst|jdSr)rrWr=r5r5r6rszUnionNode._is_interpolationcCst|jdSr)rFrWr=r5r5r6__str__szUnionNode.__str__cCst|jdSr)reprrWr=r5r5r6__repr__szUnionNode.__repr__)rr<c Cstt|}|jD]$\}}|dvrtj||d|j|<q|jd}t|tr|jd}z0d|jd<tj||d}||jd<W||jd<q||jd<0n|}||jd<|jd|jd<|S)N)rrQ)rrrQ) object__new__rMrWrr[rrLrN)r>rresr0roZ src_contentZ old_parentZ content_copyr5r5r6 __deepcopy__s       zUnionNode.__deepcopy__)TNN)N)N)rArBrCrr rfrDrrNrFrrErPrYrrrrrrrr rrrrrrr rr5r5rr6rgs<      'rg)Cr[rabcrr collectionsrZ dataclassesrrenumrtypingrr r r r r rrrZantlr4r_utilsrrrrrrrrrrrrrerrorsr r!r"r#r$r%r&r'r(Z"grammar.gen.OmegaConfGrammarParserr)Zgrammar_parserr*Zgrammar_visitorr+rFbytesrfloatrErr,rHrNrPrfrrgr5r5r5r6s6  , <,   &0