diff options
Diffstat (limited to 'src/FilterPlugin.py')
| -rw-r--r-- | src/FilterPlugin.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/src/FilterPlugin.py b/src/FilterPlugin.py new file mode 100644 index 0000000..6524d20 --- /dev/null +++ b/src/FilterPlugin.py @@ -0,0 +1,22 @@ +from typing import Any +from Item import Item +from PluginInterface import PluginInterface +from utils import get_param + + +class Plugin(PluginInterface): + def __init__(self, id: str, params: dict[str, Any]) -> None: + super().__init__(id, params) + self.filter_expr: str = get_param("filter_expr", params) + print(f"[FilterPlugin#{self.id}] initialized") + + def validate_input_n(self, n: int) -> bool: + return n == 1 + + def process(self, inputs: list[list[Item]]) -> list[Item]: + input_feed: list[Item] = inputs[0] + print(f"[FilterPlugin#{self.id}] process called, n={len(input_feed)}") + expr = f"filter(lambda item: {self.filter_expr}, input_feed)" + ret = list(eval(expr, {"input_feed": input_feed})) + print(f"[FilterPlugin#{self.id}] process returning items, n={len(ret)}") + return ret |
