From b7f129b29b8b8103336bfae265b3d5a0bc5ea03a Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Fri, 1 Sep 2023 19:17:17 +0300 Subject: Initial commit --- src/FilterPlugin.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 src/FilterPlugin.py (limited to 'src/FilterPlugin.py') 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 -- cgit v1.3