a dGD@sHddlZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z mZddlmZmZmZmZmZmZddlmZmZmZmZddlmZmZmZGd d d eZGd d d eZ Gd ddeZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%GdddeZ&GdddeZ'GdddeZ(dS)N)abstractmethod)Enum)Path)AnyDictOptionalTypeUnion) ValueKind_is_interpolation get_type_ofget_value_kindis_primitive_containertype_str)Box DictKeyTypeMetadataNode)ReadonlyConfigErrorUnsupportedValueTypeValidationErrorcs2eZdZUeed<eeeedfdd ZedddZ d(eee e e fdd d d Z edd d dZeed ddZeeed ddZe dddZe dddZee dddZee dddZedddZee eefdddd Ze dd!d"Ze dd#d$Zeeeefe d%d&d'ZZS)) ValueNode_valparentvaluemetadatacsRddlm}tj||d||||Wdn1sD0YdS)Nr) read_write)rr) omegaconfrsuper__init__ _set_value)selfrrrr __class__X/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/omegaconf/nodes.pyr s  zValueNode.__init__returncCs|jSN)rr"r%r%r&_valueszValueNode._valueNrflagsr(cCsL|drtdt|tr|jj}t|tr:t||ur:t|jj}td|ddSNzEValue '$VALUE' of type '$VALUE_TYPE' is incompatible with type hint '') _metadataref_typer1typer type_hintrr"rr9r;r%r%r&_strict_validate_type0s   zValueNode._strict_validate_typecCs`|dur6|rdSt|jj}td|d|d|ddurR|||S||SdS)z Validates input and converts to canonical form :param value: input value :return: converted value ("100" may be converted to 100 for example) NzIncompatible value 'z' for field of type 'r7convertF) _is_optionalrr8r9rr0r=_validate_and_convert_impl)r"rZ ref_type_strr%r%r&r38s  zValueNode.validate_and_convertcCsdSr)r%r"rr%r%r&r@Nsz$ValueNode._validate_and_convert_implcCs t|jSr))r2rr*r%r%r&__str__RszValueNode.__str__cCst|drt|jSdS)NrZ __INVALID__)hasattrreprrr*r%r%r&__repr__UszValueNode.__repr__otherr(cCs$t|tr|j|jkS|j|kSdSr))r1AnyNoder)r"rGr%r%r&__eq__Xs  zValueNode.__eq__cCs||}|tusJ| Sr))rINotImplemented)r"rGxr%r%r&__ne__^s  zValueNode.__ne__cCs t|jSr)hashrr*r%r%r&__hash__cszValueNode.__hash__)resmemor(cCs2tj|j|d|jd<|j|jd<|j|jd<dS)N)rQr8r_parent)copydeepcopyr8__dict__rrR)r"rPrQr%r%r&_deepcopy_implfs zValueNode._deepcopy_implcCs|jjSr))r8optionalr*r%r%r&r?nszValueNode._is_optionalcCs t|Sr))r r+r*r%r%r&r qszValueNode._is_interpolation)keyr(cCs@|}|dur.|jjdur dSt|jjSn||jjSdS)N)Z _get_parentr8rXr2 _get_full_key)r"rXrr%r%r&rZts  zValueNode._get_full_key)N)__name__ __module__ __qualname__r__annotations__rrrr r+rr2boolr!r=r3rr@rBrErIrLintrOrVr?r r rrZ __classcell__r%r%r#r&rs"  rcsbeZdZd eeeeeeeefdfdd Z eedddZ ee efddd d Z Z S) rHNrrXrr-c s$tj||ttd|d|dddS)NTr9 object_typerXrWr-r)rr rrr"rrXrr-r#r%r&r s zAnyNode.__init__r5cCs@ddlm}|ddur<||stdYn0|S)NzFValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Integer)r:r2r` ValueErrorr)r"rvalr%r%r&r@ s    z&IntegerNode._validate_and_convert_implrjcCst}||||Sr))rwrVrkr%r%r&rls zIntegerNode.__deepcopy__)NNNTN)r[r\r]rrrr_rr2r r`r@rlrar%r%r#r&rws rwc sdeZdZd eeeeeeeeefdfdd Z ee dddZ ee efdd d d Z ZS) BytesNodeNTrnc s$tj||t||tt|dddSrp)rr rrtrrr#r%r&r szBytesNode.__init__r5cCst|tstd|S)Nz;Value '$VALUE' of type '$VALUE_TYPE' is not of type 'bytes')r1rtrrAr%r%r&r@0s  z$BytesNode._validate_and_convert_implrjcCst}||||Sr))rzrVrkr%r%r&rl7s zBytesNode.__deepcopy__)NNNTN)r[r\r]rrrr_rr2r rtr@r`rlrar%r%r#r&rzsrzc seZdZdeeeeeeeeefdfdd Z ee dddZ eed d d Z e d d dZee efddddZZS) FloatNodeNTrnc s$tj||t||tt|dddSrp)rr rfloatrrr#r%r&r >szFloatNode.__init__r5cCsFz&t|tttfvrt|WStWnty@tdYn0dS)NzDValue '$VALUE' of type '$VALUE_TYPE' could not be converted to Float)r:r|r2r`rxrrAr%r%r&r@Rs   z$FloatNode._validate_and_convert_implrFcCst|tr|j}n|}|jdur,|dur,dS|jdurB|durBdS|jdurX|durXdSt|jtrpt|jnd}t|trt|nd}|j|kp|o|S)NTF)r1rrr|mathisnan)r"rGZ other_valZnan1Znan2r%r%r&rI]s zFloatNode.__eq__r'cCs t|jSr)rMr*r%r%r&rOlszFloatNode.__hash__rjcCst}||||Sr))r{rVrkr%r%r&rlos zFloatNode.__deepcopy__)NNNTN)r[r\r]rrrr_rr2r r|r@rIr`rOrlrar%r%r#r&r{=s  r{c sdeZdZd eeeeeeeeefdfdd Z eedddZ ee efdd d d Z Z S) BooleanNodeNTrnc s$tj||t||tt|dddSrp)rr rr_rrr#r%r&r vszBooleanNode.__init__r5c Cst|tr|St|tr |dkSt|trz|t|WSty}zX|dvrdWYd}~dS|dvr~WYd}~dStdt d|WYd}~qd}~00ntddS)Nr)yesyontrueT)nonofffalseFz5Value '$VALUE' is not a valid bool (type $VALUE_TYPE)) r1r_r`r2r@rxlowerrwith_tracebacksysexc_info)r"rer%r%r&r@s,      z&BooleanNode._validate_and_convert_implrjcCst}||||Sr))rrVrkr%r%r&rls zBooleanNode.__deepcopy__)NNNTNrur%r%r#r&rusrc seZdZdZdeeeeeefe ee e ee ee fdfdd Z e dddd Ze edd d Zeeee ed d dZe ee fddddZZS)EnumNodeaQ NOTE: EnumNode is serialized to yaml as a string ("Color.BLUE"), not as a fully qualified yaml type. this means serialization to YAML of a typed config (with EnumNode) will not retain the type of the Enum when loaded. This is intentional, Please open an issue against OmegaConf if you wish to discuss this decision. NT) enum_typerrXrror-c stt|trt|ts$td|di|_||_|jD]\}}|j |j|<q:t j ||t |||||dddS)Nz.EnumNode can only operate on Enum subclasses ()rqr) r1r: issubclassrrfieldsr __members__itemsrrr r) r"rrrXrror-nameZconstantr#r%r&r s&  zEnumNode.__init__r5cCs2|jj}t||s.t|jj}td|ddSr6)r8r9r1rr;rr<r%r%r&r=s    zEnumNode._strict_validate_typecCs|j|j|dS)N)rr)validate_and_convert_to_enumrrAr%r%r&r@sz#EnumNode._validate_and_convert_impl)rrr(c Cs t|ttfs&t||s&td|t||r4|Szlt|ttfrHtt|tr\||WSt|tr|jd}||r|t |d}||WSdsJWndtt fy}zFd dd|j D}td|dtd |WYd}~n d}~00dS) Nz4Value $VALUE ($VALUE_TYPE) is not a valid input for .Fz, cSsg|]}|qSr%r%).0rKr%r%r& z9EnumNode.validate_and_convert_to_enum..z)Invalid value '$VALUE', expected one of []r)r1r2r`rr|r_rxr[ startswithlenKeyErrorjoinrkeysrrr)rrprefixrZvalidr%r%r&rs4          z%EnumNode.validate_and_convert_to_enumrjcCst|jd}||||S)N)r)rrrVrkr%r%r&rls  zEnumNode.__deepcopy__)NNNTN)r[r\r]__doc__rrrr r2rrr_rr r=r@ staticmethodrr`rlrar%r%r#r&rs& rcseZdZdZdeeeeeeee fdfdd Z deeeee fddddZ eed d d Z ee efdd d dZe dddZZS)InterpolationResultNodez@ Special node type, used to wrap interpolation results. Nrbc s@tj||ttd|d|dd|dus0d|vr<|dddS)NTrcrr.)rr rrZ _set_flagrer#r%r&r s z InterpolationResultNode.__init__r,cCs"|drtd|||_dS)Nr.r/)r0rr3rr4r%r%r&r!s z"InterpolationResultNode._set_valuer5cCs|Sr)r%rAr%r%r&r@sz2InterpolationResultNode._validate_and_convert_implrjcCstdSr))NotImplementedError)r"rQr%r%r&rlsz$InterpolationResultNode.__deepcopy__r'cCsdS)NFr%r*r%r%r&r sz)InterpolationResultNode._is_interpolation)NNN)N)r[r\r]rrrrrr2r_r r!r@r`rlr rar%r%r#r&rs r))rSr}rabcrenumrpathlibrtypingrrrrr Zomegaconf._utilsr r r r rrZomegaconf.baserrrrZomegaconf.errorsrrrrrHrmrvrwrzr{rrrr%r%r%r&s&    j&&)'"81U