Source code for tensortrade.feed.core.accessors

class CachedAccessor:
    """
    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.

    References
    ----------
    .. [1] https://github.com/pandas-dev/pandas/blob/v1.1.0/pandas/core/accessor.py#L285-L289
    """

    def __init__(self, name: str, accessor) -> None:
        self._name = name
        self._accessor = accessor

    def __get__(self, instance, owner):
        if instance is None:
            return self._accessor
        accessor = self._accessor(instance)
        object.__setattr__(instance, self._name, accessor)
        return accessor