E g<'^dZddlmZddlmZmZddlZddlmZddl m Z GddZ Gd d Z d d d fd%dZ GddZeddd&dZeedd&d Zeed!d&d"Zeed#d&d$ZdS)'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_levelc~eZdZUeZded<eZded<ed dZ d dZ d fd Z xZ S) DirNamesMixinset[str] _accessorszfrozenset[str] _hidden_attrsreturnc |j|jzS)z: Delete unwanted __dir__ for this object. )r r selfs I/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/core/accessor.py_dir_deletionszDirNamesMixin._dir_deletionss !333c*fdjDS)z9 Add additional __dir__ for this object. c4h|]}t||S)hasattr).0accessorrs r z/DirNamesMixin._dir_additions.."s(TTTXGD(zPandasDelegate.Fs1rT accessorsrtyp overwriteboolaccessor_mappingCallable[[str], str]raise_on_missingr Nonecdfd }dfd }|D]_} |st| d|dkr || } n || } |st|| st|| | `dS)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. r,r-c fd}fd}|_|_t||tjS)Nc.|Sr;)r4)rr,s r_getterz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._getter_s224888rc0||Sr;)r7)r new_valuesr,s r_setterz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._setterbs224DDDr)fgetfsetr)r$propertygetattr__doc__)r,rJrMrCdelegates` r_create_delegator_propertyzJPandasDelegate._add_delegate_accessors.._create_delegator_property^s 9 9 9 9 9 E E E E E $G #G H&6&6t&<&<==E rchfd}|_tj|_|S)Nc&|jg|Ri|Sr;)r9)rr2r3r,s rfzSPandasDelegate._add_delegate_accessors.._create_delegator_method..fos&,t,TCDCCCFCCCr)r$rQrR)r,rWrCrSs` r_create_delegator_methodzHPandasDelegate._add_delegate_accessors.._create_delegator_methodnsP D D D D DAJ*:*:4*@*@AAIAIHrNrPr,r-)rQrsetattr) clsrSr?r@rArCrErTrXr,rWs ` ` r_add_delegate_accessorsz&PandasDelegate._add_delegate_accessors?s>                & &D$ H&6&6t&<&>>>::::  1<!%E&E&E&E&[E&E&E&rr+Fc|Sr;rr<s rr>r>sqrTr?rr@r-rArBrCrDrEc&fd}|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. 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. Returns ------- callable A class decorator. Examples -------- @delegate_names(Categorical, ["categories", "ordered"], "property") class CategoricalAccessor(PandasDelegate): [...] c>||S)N)rArCrE)r\)r[rCr?rSrArEr@s radd_delegate_accessorsz.delegate_names..add_delegate_accessorss9 ##   -- $    rr)rSr?r@rArCrEras`````` rdelegate_namesrbsCP           "!rc eZdZdZd dZdZdS) 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,r-r rFc"||_||_dSr;)_name _accessor)rr,rs r__init__zCachedAccessor.__init__s !rc||jS||}t||j||Sr;)rgobject __setattr__rf)robjr[ accessor_objs r__get__zCachedAccessor.__get__sB ;> !~~c** 3 L999rN)r,r-r rF)r$r%r&rRrhrnrrrrdrdsA&""""     rrd)klassothersr,cfd}|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 LtrYtjdt|dtdtdtt t t|j |S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.) stacklevel) rwarningswarnrepr UserWarningrrZrdr add)rr[r,s r decoratorz%_register_accessor..decorator2s 3    M1DNN11::11)-c111+--      T>$99::: 4   rr)r,r[rzs`` r_register_accessorr{s+R       r DataFrame)rpc.ddlm}t||S)Nr)r|)pandasr|r{)r,r|s rregister_dataframe_accessorrBs$      dI . ..rSeriesc.ddlm}t||S)Nr)r)r~rr{)r,rs rregister_series_accessorrIs$ dF + ++rIndexc.ddlm}t||S)Nr)r)r~rr{)r,rs rregister_index_accessorrPs$ dE * **r) r?rr@r-rArBrCrDrErBrY)rR __future__rtypingrrrupandas.util._decoratorsrpandas.util._exceptionsrr r+rbrdr{rrrrrrrs #"""""''''''444444