a ijT@sdZddlmZddlZddlZddlZddlmZmZmZm Z e e ee fZ ddddd Z d d d dd d ddddZ dddddZdddddZdd dddZdddddZdS) zAdhoc imports utils.) annotationsN)SequenceNoReturnOptionalUnion StrOrStrListz list[str])restrictreturncs>t|tddDsJtfddtjDS)z5Returns all `sys.modules` matching the restrict name.css|]}d|vVqdS)/N).0moduler r W/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/ecolab/module_utils.py z#get_module_names..csg|]}|r|qSr ) startswith)r mmodulesr r $rz$get_module_names..)normalize_str_to_listalltuplesysr)rr rrget_module_namess rFT)verbose invalidateboolNone)rrrr cCsnt|}|sdS|D]B}|r*td|t||rN|dsNttj|tj|=qtjD] }|q^dS)auClear the `sys.modules` cache. Helpful for interactive development to reload from Jupyter notebook the code we're currently editing (without having to restart the notebook kernel). Usage: ```python ecolab.clear_cached_modules(['visu3d', 'other_module.submodule']) import visu3d import other_module.submodule ``` Args: modules: List of modules to clear (all submodules cleared too) verbose: Whether to display the list of modules cleared. invalidate: If `True` (default), the instances of the module will raise an error when used (to avoid using 2 versions of a module at the same time) Nz Clearing Zetils) rprint_clear_parent_module_attrr_invalidate_modulerrtyping _cleanups)rrrZmodules_to_clear module_namecleanupr r rclear_cached_modules(s  r&str)r$r cCsHd|vr dS|dd\}}tj|}|s0dSt||rDt||dS)z;Remove parent reference (e.g. `path.to.child` -> `path.to`..N)rsplitrrgethasattrdelattr)r$parent_module_nameZ child_name parent_moduler r rr \s  r ztypes.ModuleType)r r cs.|jdddfdd }|j||_dS)zInvalidate the module in-place.r'r)namer cs tdd|dddS)z-All module attribute access will raise error.zCannot access r(z on the old module instance. z was reloaded, so the new module should be used instead. You can pass `invalidate=False` to keep both old and reloaded modules.N)AttributeError)r0r$r r __getattr__ms z'_invalidate_module..__getattr__N)__name____dict__clearr3)r r3r r2rr!hs r!cCs |dS)z(Returns `True` if the module is a proto.Z_pb2)endswithr2r r r _is_protozsr8zOptional[StrOrStrList])xr cCsT|dur gSt|tr*dd|dDSt|ttfsHtd|nt|SdS)NcSsg|] }|qSr )strip)r vr r rrrz)normalize_str_to_list..,zExpected list. Got: ) isinstancer'splitlistr TypeError)r9r r rrs r)__doc__ __future__rrtypesr"rrrrr'rrr&r r!r8rr r r rs 4