U c@sdZddlmZddlmZmZmZddlZddl m Z m Z ddl m Z mZddlmZddlmZerdd lmZdd lmZed d d Zeddd ZddddddddZdddddddddZGdddeZGdd d eZdS) z? Shared methods for Index subclasses backed by ExtensionArray. ) annotations) TYPE_CHECKINGCallableTypeVarN) ArrayLikenpt)cache_readonlydoc) ABCDataFrame)Index) IntervalArray)NDArrayBackedExtensionArray_TNDArrayBackedExtensionIndex)bound_ExtensionIndexTExtensionIndexFstrtypebool)namedelegatecachewrapcst|tts"tjdkr~|rJfdd}|_j|_t|}qfdd}fdd}|_j|_t||}n*ts}nfdd }|_j|_|S) a Make an alias for a method of the underlying ExtensionArray. Parameters ---------- name : str Name of an attribute the class should inherit from its EA parent. delegate : class cache : bool, default False Whether to convert wrapped properties into cache_readonly wrap : bool, default False Whether to wrap the inherited result in an Index. Returns ------- attribute, method, property, or cache_readonly Zgetset_descriptorcs t|jSN)getattr_dataselfrA/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/indexes/extension.pycached=sz"_inherit_from_data..cachedcsZt|j}rVt|t|jr4t|j||jdSt|trH||St||jdS|S)Nr) rr isinstancer _simple_newrr set_indexr rresult)rrr r!fgetFs   z _inherit_from_data..fgetcst|j|dSr)setattrrrvaluerr r!fsetPsz _inherit_from_data..fsetcs|d|krtdt|j|jf||}rxt|t|jrVt|j||jdSt|trj||St ||jdS|S)NZinplacezcannot use inplace with r) ValueErrorr__name__rr#r$rr r%r )rargskwargsr')attrrr r!method_s  z"_inherit_from_data..method)rr#propertyrr.__doc__rcallable)rrrrr"r2r(r,r )r1rrr!_inherit_from_data#s$      r6z list[str]z:Callable[[type[_ExtensionIndexT]], type[_ExtensionIndexT]])namesrrrreturncsdddfdd }|S)a Class decorator to pin attributes from an ExtensionArray to a Index subclass. Parameters ---------- names : List[str] delegate : class cache : bool, default False wrap : bool, default False Whether to wrap the inherited result in an Index. ztype[_ExtensionIndexT])clsr8cs*D] }t|d}t|||q|S)N)rr)r6r))r9rmethrrr7rr r!wrapperszinherit_names..wrapperr )r7rrrr<r r;r! inherit_namesqsr=c@sHeZdZUdZded<ddZeejd ddZe d d d d Z dS)rz> Index subclass for indexes backed by ExtensionArray. z+IntervalArray | NDArrayBackedExtensionArrayrcCs |j|S)zE Convert value to be insertable to underlying array. )rZ_validate_setitem_valuer*r r r!_validate_fill_valuesz#ExtensionIndex._validate_fill_valueNcCs^z4||}t|tjrt|}t|ts0td|WStk rX|t|YSXdS)Nz,The map function must return an Index object) r#npZndarrayr TypeError ExceptionZastypeobjectmap)rZmapperZ na_actionr'r r r!rCs  zExtensionIndex.mapznpt.NDArray[np.bool_]r8cCs |jSr)rZisnarr r r!_isnanszExtensionIndex._isnan)N) r. __module__ __qualname__r4__annotations__r>r r rCrrEr r r r!rs  c@s8eZdZUdZded<ddddZddd d d Zd S) rzK Index subclass for indexes backed by NDArrayBackedExtensionArray. r rz np.ndarrayrDcCs|jjSr)r_ndarrayrr r r!_get_engine_targetsz.NDArrayBackedExtensionIndex._get_engine_targetr)r'r8cCs |j|jjjkst|j|Sr)ZdtyperrIAssertionErrorZ_from_backing_datar&r r r!_from_join_targetsz-NDArrayBackedExtensionIndex._from_join_targetN)r.rFrGr4rHrJrLr r r r!rs )FF)FF)r4 __future__rtypingrrrZnumpyr?Zpandas._typingrrZpandas.util._decoratorsrr Zpandas.core.dtypes.genericr Zpandas.core.indexes.baser Zpandas.core.arraysr Zpandas.core.arrays._mixinsr rrr6r=rrr r r r!s(       O*