a ij@sdZddlmZddlZddlZddlZddlmZmZmZddl Z edZ e j dddkrfej fne ejfZGdd d geRZ d d d d Zddd dddZddZddZdddddZdS)z Python utils.) annotationsN)AnyTypeVarUnion_Cls) csneZdZdZddddZedddddZdd d fd d Zd dfdd Ze j rfeddZ Z S)StrEnuma(Like `Enum`, but `enum.auto()` assigns `str` rather than `int`. ```python class MyEnum(epy.StrEnum): SOME_ATTR = enum.auto() OTHER_ATTR = enum.auto() assert MyEnum('some_attr') is MyEnum.SOME_ATTR assert MyEnum.SOME_ATTR == 'some_attr' ``` `StrEnum` is case insensitive. strreturncCs|SN)lower)namestartcount last_valuesrP/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/epy/py_utils.py_generate_next_value_2szStrEnum._generate_next_value_)valuer cCsJt|tr|s||Sdd|D}t|d|jd|dS)NcSsg|] }|jqSr)r).0errr :z%StrEnum._missing_..z is not a valid z. Expected one of ) isinstancer islowerr ValueError __qualname__)clsrZ all_valuesrrr _missing_5s zStrEnum._missing_bool)otherr cst|Sr)super__eq__r)selfr# __class__rrr%@szStrEnum.__eq__intcs tSr)r$__hash__)r&r'rrr*CszStrEnum.__hash__cCsttj|Sr)typeenumEnum__iter__r rrrr.KszStrEnum.__iter__) __name__ __module__r__doc__r classmethodr!r%r*typing TYPE_CHECKINGr. __classcell__rrr'rr !s r r"r cCst|tot|dS)zReturns `True` if the value is instance of `NamedTuple`. This is using some heuristic by checking for a `._field` attribute. Args: x: Object to check Returns: `True` if the object is a `namedtuple` _fields)rtuplehasattr)xrrr is_namedtuplePs r;rz'Union[type[Any], tuple[type[Any], ...]])r typesr cCst|tot||S)zALike `issubclass`, but do not raise error if value is not `type`.)rr+ issubclass)r r<rrr issubclass_^sr>cstfdd}|S)z`__init__` wrapper.csVd|jvr |g|Ri|St|dd|g|Ri|t|dddS)N_epy_is_init_doneFT)__dict__object __setattr__)r&argskwargsinit_fnrrnew_initis  z_wrap_init..new_init functoolswraps)rFrGrrEr _wrap_initfs rKcstfdd}|S)z`__setattr__` wrapper.csLt|dstdt|d|js0|||Std|dt|dS)Nr?zKChild of `@epy.frozen` class should be `@epy.frozen` too. (Error raised by )zCannot assign z in `@epy.frozen` class )r9rr+r?AttributeError)r&rr setattr_fnrr new_setattrzs  z"_wrap_setattr..new_setattrrH)rOrPrrNr _wrap_setattrws rQ)r r cCs6t|tst|jdt|j|_t|j|_|S)aClass decorator which prevent mutating attributes after `__init__`. Example: ```python @epy.frozen class A: def __init__(self): self.x = 123 a = A() a.x = 456 # AttributeError ``` Supports inheritance, child classes should explicitly be marked as `@epy.frozen` if they mutate additional attributes in `__init__`. Args: cls: The class to freeze. Returns: cls: The class object z is not)rr+ TypeErrorr0rK__init__rQrBr/rrrfrozens    rT)r2 __future__rr,rIr4rrrsysr version_infor r r-Z_StrEnumr;r>rKrQrTrrrrs "/