Owg<',dZddlmZddlmZmZddlZddlmZddl m Z GddZ Gd d Z d d d f ddZ GddZeddddZeedddZeedddZeedddZy)z accessor.py contains base classes for implementing accessor properties that can be mixed into or pinned onto other pandas classes. ) annotations)CallablefinalN)doc)find_stack_levelcleZdZUeZded<eZded<eddZ ddZ d fd Z xZ S) DirNamesMixinset[str] _accessorszfrozenset[str] _hidden_attrsc4|j|jzS)z: Delete unwanted __dir__ for this object. )r r )selfs K/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/core/accessor.py_dir_deletionszDirNamesMixin._dir_deletionss !3!333cZ|jDchc]}t||s|c}Scc}w)z9 Add additional __dir__ for this object. )r hasattr)raccessors r_dir_additionszDirNamesMixin._dir_additionss%*.TXGD(/ 4 4 U  rr cZeZdZdZd dZd dZd dZedddf d dZy ) PandasDelegatez@ Abstract base class for delegating methods/properties. ctd|)NzYou cannot access the property  TypeErrorrnameargskwargss r_delegate_property_getz%PandasDelegate._delegate_property_get6s9$@AArc td|d)Nz The property z cannot be setr()rr+valuer,r-s r_delegate_property_setz%PandasDelegate._delegate_property_set9s-v^<==rctd|)NzYou cannot call method r(r*s r_delegate_methodzPandasDelegate._delegate_method<s1$899rFc|SNxs rzPandasDelegate.Fs1rTcdfd }dfd }|D]J} |st| d|dk(r || } n|| } |s t|| r>t|| | Ly)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. accessor_mapping: Callable, default lambda x: x Callable to map the delegate's function to the cls' function. raise_on_missing: bool, default True Raise if an accessor does not exist on delegate. False skips the missing accessor. c fd}fd}|_|_t||tjS)Nc&|jSr5)r.)rr+s r_getterz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._getter_s22488rc(|j|Sr5)r1)r new_valuesr+s r_setterz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._setterbs224DDr)fgetfsetr)rpropertygetattr__doc__)r+r=r@accessor_mappingdelegates` r_create_delegator_propertyzJPandasDelegate._add_delegate_accessors.._create_delegator_property^sI 9 E $G #G H&6t&<=EE rcdfd}|_tj|_|S)Nc0|jg|i|Sr5)r3)rr,r-r+s rfzSPandasDelegate._add_delegate_accessors.._create_delegator_method..fos ,t,,TCDCFCCr)rrDrE)r+rKrFrGs` r_create_delegator_methodzHPandasDelegate._add_delegate_accessors.._create_delegator_methodns1 DAJ*:4*@AIIAIHrNrCr+str)rDrsetattr) clsrG accessorstyp overwriterFraise_on_missingrHrLr+rKs ` ` r_add_delegate_accessorsz&PandasDelegate._add_delegate_accessors?so>    &D$H&6t&:  1.add_delegate_accessorss/ ##   -- $  rr6)rGrQrRrSrFrTr]s`````` rdelegate_namesr^sP   "!rceZdZdZddZdZy)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``. c ||_||_yr5)_name _accessor)rr+rs r__init__zCachedAccessor.__init__s !rc| |jS|j|}tj||j||Sr5)rcobject __setattr__rb)robjrP accessor_objs r__get__zCachedAccessor.__get__s= ;>> !~~c* 3 L9rN)r+rNrrX)rr r!rErdrjr6rrr`r`s&" rr`)klassotherscfd}|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 c trItjdt|dtdtdtt t t|jj|S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.) stacklevel) rwarningswarnrepr UserWarningrrOr`r add)rrPr+s r decoratorz%_register_accessor..decorator2s{ 3  MM+DN+;<:,jc 401+-   T>$9: 4 rr6)r+rPrvs`` r_register_accessorrwsR  r DataFrame)rlc&ddlm}t||S)Nr)rx)pandasrxrw)r+rxs rregister_dataframe_accessorr{Bs dI ..rSeriesc&ddlm}t||S)Nr)r|)rzr|rw)r+r|s rregister_series_accessorr~Is dF ++rIndexc&ddlm}t||S)Nr)r)rzrrw)r+rs rregister_index_accessorrPs dE **r) rQrrRrNrSrVrFrWrTrVrM)rE __future__rtypingrrrqpandas.util._decoratorsrpandas.util._exceptionsrr r&r^r`rwr{r~rr6rrrs #'4