U Dv f‡ã@sHdZddlmZmZddlmZdd„Zedœdd„Zedœd d „Z d S) z!XArray support for pandas_flavor.é)Úregister_dataarray_accessorÚregister_dataset_accessor)ÚwrapscsG‡fdd„dƒ}|S)a Makes an XArray-compatible accessor to wrap a method to be added to an xr.DataArray, xr.Dataset, or both. Args: method: A method which takes an XArray object and needed parameters. Returns: The result of calling ``method``. cs,eZdZdZdd„Ze”ƒ‡fdd„ƒZdS)z)make_accessor_wrapper..XRAccessorzXArray accessor for a method.cSs ||_dS)z…Initialize the accessor. Args: xr_obj: The XArray object to which the accessor is attached. N©Z_xr_obj)ÚselfZxr_obj©rú8/tmp/pip-unpacked-wheel-c6c42brj/pandas_flavor/xarray.pyÚ__init__sz2make_accessor_wrapper..XRAccessor.__init__csˆ|jf|ž|ŽS)zþCall the method. Args: *args: Positional arguments to pass to the method. **kwargs: Keyword arguments to pass to the method. Returns: The result of calling ``method``. r)rÚargsÚkwargs©ÚmethodrrÚ__call__s z2make_accessor_wrapper..XRAccessor.__call__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rrrr rrÚ XRAccessorsrr)r rrr rÚmake_accessor_wrappers rr cCst|ƒ}t|jƒ|ƒ|S)z¯Register a method on an XArray DataArray object. Args: method: A method which takes an XArray object and needed parameters. Returns: The method. )rrr©r Zaccessor_wrapperrrrÚ register_xarray_dataarray_method.s rcCst|ƒ}t|jƒ|ƒ|S)z­Register a method on an XArray Dataset object. Args: method: A method which takes an XArray object and needed parameters. Returns: The method. )rrrrrrrÚregister_xarray_dataset_method=s rN) rZxarrayrrÚ functoolsrrÚcallablerrrrrrÚs  (