aboutsummaryrefslogtreecommitdiffstats
path: root/app/FilterPlugin.py
blob: fd3984ae9a74a63dcb52f789fb71264ab53520f8 (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 app.Item import Item
from app.PluginInterface import PluginInterface
from app.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