o poi @sBddlZddlmZddlmZddlmZGdddeeZdS)N)Enum)Optional)Literalc @seZdZdZeddededddfddZeddededdedfd d Z edede efd d Z d e de fddZdefddZdS)StrEnumaType of any enumerator with allowed comparison to string invariant to cases. >>> class MySE(StrEnum): ... t1 = "T-1" ... t2 = "T-2" >>> MySE("T-1") == MySE.t1 True >>> MySE.from_str("t-2", source="value") == MySE.t2 True >>> MySE.from_str("t-2", source="value") >>> MySE.from_str("t-3", source="any") Traceback (most recent call last): ... ValueError: Invalid match: expected one of ['t1', 't2', 'T-1', 'T-2'], but got t-3. keyvaluesource)rranyreturncCs||dvr|jD]}||kr||Sq|dvr0|jD]\}}||kr/||Sq!td||d|d)aCreate ``StrEnum`` from a string matching the key or value. Args: value: matching string source: compare with: - ``"key"``: validates only from the enum keys, typical alphanumeric with "_" - ``"value"``: validates only from the values, could be any string - ``"any"``: validates with any key or value, but key has priority Raises: ValueError: if requested string does not match any option based on selected source. )rr )rr zInvalid match: expected one of , but got .) __members__loweritems ValueError_allowed_matches)clsrrenum_keyenum_valrR/home/ubuntu/.local/lib/python3.10/site-packages/lightning_utilities/core/enums.pyfrom_strs   zStrEnum.from_strc CsFz|||WSty"ttd||d|dYdSw)z?Try to create emun and if it does not match any, return `None`.z Invalid string: expected one of r r N)rrwarningswarn UserWarningr)rrrrrr try_from_str:s zStrEnum.try_from_strcCsTgg}}|jD]\}}||||jq |dkr |S|dkr&|S||S)Nrr)r rappendr)rrkeysvalsrrrrrrEs  zStrEnum._allowed_matchesothercCs&t|tr|j}|jt|kS)zCompare two instances.) isinstancerrrstr)selfrrrr__eq__Qs zStrEnum.__eq__cCst|jS)zReturn unique hash.)hashrr)r"rrr__hash__WszStrEnum.__hash__N)r)__name__ __module__ __qualname____doc__ classmethodr!rrrrlistrobjectboolr#intr%rrrrr s"  r) renumrtypingrtyping_extensionsrr!rrrrrs