o i!@sdZddlmZddlZddlmZddlmZGdddZGdd d Z d!d"ddZ GdddZ edddddZ ee ddddZ ee ddddZee dddd ZdS)#z accessor.py contains base classes for implementing accessor properties that can be mixed into or pinned onto other pandas classes. ) annotationsN)doc)find_stack_levelcsPeZdZUeZded<eZded<d ddZd dd Z dfd d Z Z S) DirNamesMixinset[str] _accessorszfrozenset[str] _hidden_attrsreturncCs |j|jBS)z: Delete unwanted __dir__ for this object. )rrselfr O/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/core/accessor.py_dir_deletionss zDirNamesMixin._dir_deletionscsfddjDS)z9 Add additional __dir__ for this object. csh|] }t|r|qSr )hasattr).0accessorr r r sz/DirNamesMixin._dir_additions..)rr r r r _dir_additionsszDirNamesMixin._dir_additions list[str]cs*tt}|||B}t|S)z Provide method name lookup and completion. Notes ----- Only provide 'public' methods. )setsuper__dir__rrsorted)r rv __class__r r rszDirNamesMixin.__dir__)r r)r r) __name__ __module__ __qualname__rr__annotations__ frozensetrrrr __classcell__r r rr rs   rc@s:eZdZdZddZddZddZe ddd dZdS)PandasDelegatez@ Abstract base class for delegating methods/properties. cOtd|)NzYou cannot access the property  TypeErrorr nameargskwargsr r r _delegate_property_get1z%PandasDelegate._delegate_property_getcOstd|d)Nz The property z cannot be setr$)r r'valuer(r)r r r _delegate_property_set4sz%PandasDelegate._delegate_property_setcOr#)NzYou cannot call method r$r&r r r _delegate_method7r+zPandasDelegate._delegate_methodFtypstr overwriteboolc sZfdd}fdd}|D]}|dkr||}n||}|s$t||s*t|||qdS)a Add accessors to cls from the delegate class. Parameters ---------- cls Class to add the methods/properties to. delegate Class to get methods/properties and doc-strings. accessors : list of str List of accessors to add. typ : {'property', 'method'} overwrite : bool, default False Overwrite the method/property in the target class if it exists. cs:fdd}fdd}|_|_t||tjdS)Ncs |SN)r*r r'r r _getterOs z[PandasDelegate._add_delegate_accessors.._create_delegator_property.._gettercs ||Sr3)r-)r new_valuesr4r r _setterRs z[PandasDelegate._add_delegate_accessors.._create_delegator_property.._setter)fgetfsetr)rpropertygetattr__doc__)r'r5r7delegater4r _create_delegator_propertyNs  zJPandasDelegate._add_delegate_accessors.._create_delegator_propertycs$fdd}|_tj|_|S)Ncs|jg|Ri|Sr3)r.)r r(r)r4r r f]szSPandasDelegate._add_delegate_accessors.._create_delegator_method..f)rr;r<)r'r@r=r4r _create_delegator_method\s zHPandasDelegate._add_delegate_accessors.._create_delegator_methodr:N)rsetattr) clsr> accessorsr/r1r?rAr'r@r r=r _add_delegate_accessors:s     z&PandasDelegate._add_delegate_accessorsNFr/r0r1r2) rrrr<r*r-r. classmethodrEr r r r r",sr"Fr/r0r1r2csfdd}|S)a Add delegated names to a class using a class decorator. This provides an alternative usage to directly calling `_add_delegate_accessors` below a class definition. Parameters ---------- delegate : object The class to get methods/properties & doc-strings. accessors : Sequence[str] List of accessor to add. typ : {'property', 'method'} overwrite : bool, default False Overwrite the method/property in the target class if it exists. Returns ------- callable A class decorator. Examples -------- @delegate_names(Categorical, ["categories", "ordered"], "property") class CategoricalAccessor(PandasDelegate): [...] cs|jd|S)N)r1)rE)rCrDr>r1r/r r add_delegate_accessorssz.delegate_names..add_delegate_accessorsr )r>rDr/r1rJr rIr delegate_namesqsrKc@s"eZdZdZd ddZdd Zd S) CachedAccessora Custom property-like object. A descriptor for caching accessors. Parameters ---------- name : str Namespace that will be accessed under, e.g. ``df.foo``. accessor : cls Class with the extension methods. Notes ----- For accessor, The class's __init__ method assumes that one of ``Series``, ``DataFrame`` or ``Index`` as the single argument ``data``. r'r0r NonecCs||_||_dSr3)_name _accessor)r r'rr r r __init__s zCachedAccessor.__init__cCs,|dur|jS||}t||j||Sr3)rOobject __setattr__rN)r objrC accessor_objr r r __get__s  zCachedAccessor.__get__N)r'r0r rM)rrrr<rPrUr r r r rLs  rL)klassotherscsfdd}|S)aG Register a custom accessor on {klass} objects. Parameters ---------- name : str Name under which the accessor should be registered. A warning is issued if this name conflicts with a preexisting attribute. Returns ------- callable A class decorator. See Also -------- register_dataframe_accessor : Register a custom accessor on DataFrame objects. register_series_accessor : Register a custom accessor on Series objects. register_index_accessor : Register a custom accessor on Index objects. Notes ----- When accessed, your accessor will be initialized with the pandas object the user is interacting with. So the signature must be .. code-block:: python def __init__(self, pandas_object): # noqa: E999 ... For consistency with pandas methods, you should raise an ``AttributeError`` if the data passed to your accessor has an incorrect dtype. >>> pd.Series(['a', 'b']).dt Traceback (most recent call last): ... AttributeError: Can only use .dt accessor with datetimelike values Examples -------- In your library code:: import pandas as pd @pd.api.extensions.register_dataframe_accessor("geo") class GeoAccessor: def __init__(self, pandas_obj): self._obj = pandas_obj @property def center(self): # return the geographic center point of this DataFrame lat = self._obj.latitude lon = self._obj.longitude return (float(lon.mean()), float(lat.mean())) def plot(self): # plot this array's data on a map, e.g., using Cartopy pass Back in an interactive IPython session: .. code-block:: ipython In [1]: ds = pd.DataFrame({{"longitude": np.linspace(0, 10), ...: "latitude": np.linspace(0, 20)}}) In [2]: ds.geo.center Out[2]: (5.0, 10.0) In [3]: ds.geo.plot() # plots data on a map cs^trtjdt|dtdtdttdtt|j |S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.) stacklevel) rwarningswarnrepr UserWarningrrBrLradd)rrCr'r r decorators   z%_register_accessor..decoratorr )r'rCr`r r_r _register_accessorsI ra DataFrame)rWcCddlm}t||S)Nr)rb)pandasrbra)r'rbr r r register_dataframe_accessor  reSeriescCrc)Nr)rg)rdrgra)r'rgr r r register_series_accessorrfrhIndexcCrc)Nr)ri)rdrira)r'rir r r register_index_accessor&rfrjrFrG)r< __future__rrZpandas.util._decoratorsrpandas.util._exceptionsrrr"rKrLrarerhrjr r r r s"    E) %  X