blob: 6524d20a910b05c2ec22e1d9255c2432fa468a96 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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
|