1from __future__
import annotations
3__all__ = (
"DataFrameAction",)
6from typing
import Iterable, Any, Mapping
8from ..configurableActions
import ConfigurableAction
12 _actionCache: Mapping[int, Any]
14 cache = Field(doc=
"Controls if the results of this action should be cached,"
15 " only works on frozen actions",
16 dtype=bool, default=
False)
17 cacheArgs = ListField(doc=
"If cache is True, this is a list of argument keys that will be used to "
18 "compute the cache key in addition to the DataFrameId",
19 dtype=str, optional=
True)
24 def call_wrapper(function):
25 def inner_wrapper(self, dataFrame, **kwargs):
29 if name
not in kwargs:
30 raise ValueError(f
"{name} is not part of call signature and cant be used for "
32 extra.append(kwargs[name])
35 if self.
cache and self._frozen:
39 result = function(self, dataFrame, **kwargs)
40 if self.
cache and self._frozen:
47 def __call__(self, dataFrame, **kwargs) -> Iterable[Any]:
48 """This method should return the result of an action performed on a
51 raise NotImplementedError(
"This method should be overloaded in a subclass")
55 """This property should return an iterable of columns needed by this action
57 raise NotImplementedError(
"This method should be overloaded in a subclass")
Any __call__(self, *args, **kwargs)
Iterable[str] columns(self)
Iterable[Any] __call__(self, dataFrame, **kwargs)
None __init_subclass__(cls, **kwargs)