B W0d!@sdZddlmZddlZddlmZGdddZGdddZdd d d d dZGdddZ edddddZ ee ddddZ ee ddddZ ee ddddZ dS) z accessor.py contains base classes for implementing accessor properties that can be mixed into or pinned onto other pandas classes. ) annotationsN)doccs\eZdZUeZded<eZded<ddddZdddd Z d dfd d Z Z S) DirNamesMixinzset[str] _accessorszfrozenset[str] _hidden_attrs)returncCs |j|jBS)z: Delete unwanted __dir__ for this object. )rr)selfr F/var/www/html/venv/lib/python3.7/site-packages/pandas/core/accessor.py_dir_deletionsszDirNamesMixin._dir_deletionscsfddjDS)z9 Add additional __dir__ for this object. csh|]}t|r|qSr )hasattr).0accessor)rr r sz/DirNamesMixin._dir_additions..)r)rr )rr _dir_additionsszDirNamesMixin._dir_additionsz list[str]cs*tt}|||B}t|S)z Provide method name lookup and completion. Notes ----- Only provide 'public' methods. )setsuper__dir__r rsorted)rrv) __class__r r rszDirNamesMixin.__dir__) __name__ __module__ __qualname__rr__annotations__ frozensetrr rr __classcell__r r )rr rs rc@s>eZdZdZddZddZddZedd d d d d ZdS)PandasDelegatez@ Abstract base class for delegating methods/properties. cOstd|dS)NzYou cannot access the property ) TypeError)rnameargskwargsr r r _delegate_property_get0sz%PandasDelegate._delegate_property_getcOstd|ddS)Nz The property z cannot be set)r)rrvaluer r!r r r _delegate_property_set3sz%PandasDelegate._delegate_property_setcOstd|dS)NzYou cannot call method )r)rrr r!r r r _delegate_method6szPandasDelegate._delegate_methodFstrbool)typ overwritec s^fdd}fdd}x@|D]8}|dkr4||}n||}|sJt||st|||qWdS)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 |S)N)r")r)rr r _getterNsz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._gettercs ||S)N)r$)r new_values)rr r _setterQsz[PandasDelegate._add_delegate_accessors.._create_delegator_property.._setter)fgetfsetr)rpropertygetattr__doc__)rr*r,)delegate)rr _create_delegator_propertyMs   zJPandasDelegate._add_delegate_accessors.._create_delegator_propertycs$fdd}|_tj|_|S)Ncs|jf||S)N)r%)rr r!)rr r f\szSPandasDelegate._add_delegate_accessors.._create_delegator_method..f)rr0r1)rr4)r2)rr _create_delegator_method[s zHPandasDelegate._add_delegate_accessors.._create_delegator_methodr/N)r setattr) clsr2 accessorsr(r)r3r5rr4r )r2r _add_delegate_accessors9s   z&PandasDelegate._add_delegate_accessorsN)F) rrrr1r"r$r% classmethodr9r r r r r+s rFr&r')r(r)csfdd}|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)r))r9)r7)r8r2r)r(r r add_delegate_accessorssz.delegate_names..add_delegate_accessorsr )r2r8r(r)r;r )r8r2r)r(r delegate_namespsr<c@s(eZdZdZdddddZddZd 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&None)rrcCs||_||_dS)N)_name _accessor)rrrr r r __init__szCachedAccessor.__init__cCs,|dkr|jS||}t||j||S)N)r@object __setattr__r?)robjr7Z accessor_objr r r __get__s  zCachedAccessor.__get__N)rrrr1rArEr r r r r=sr=)klassZotherscsfdd}|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\tr:tjdt|dtdtdtddtt|j|S)Nzregistration of accessor z under name z for type z: is overriding a preexisting attribute with the same name.) stacklevel) r warningswarnrepr UserWarningr6r=radd)r)r7rr r decorators " z%_register_accessor..decoratorr )rr7rOr )r7rr _register_accessorsI rP DataFrame)rGcCsddlm}t||S)Nr)rQ)pandasrQrP)rrQr r r register_dataframe_accessors rSSeriescCsddlm}t||S)Nr)rT)rRrTrP)rrTr r r register_series_accessors rUIndexcCsddlm}t||S)Nr)rV)rRrVrP)rrVr r r register_index_accessor%s rW)F)r1 __future__rrJZpandas.util._decoratorsrrrr<r=rPrSrUrWr r r r s  E)%Y