From bead581a92c51ab1e7f18161972d73c6ffa3eb3f Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Sun, 10 Sep 2023 11:19:57 +0300 Subject: Add dotenv and env var in config support --- plugins/DigestPlugin.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'plugins/DigestPlugin.py') diff --git a/plugins/DigestPlugin.py b/plugins/DigestPlugin.py index d6e8eb4..b56af0d 100644 --- a/plugins/DigestPlugin.py +++ b/plugins/DigestPlugin.py @@ -7,7 +7,7 @@ from tinydb import Query from app.Item import Item from app.PluginInterface import Params, PluginInterface from app.DatabaseManager import database_manager -from app.utils import dict_to_item, get_param, item_to_dict +from app.utils import dict_to_item, get_config_or_default, item_to_dict, get_config def add_days(start: datetime, days: int): @@ -99,14 +99,16 @@ def parse_interval(interval: str) -> tuple[int, str]: class Plugin(PluginInterface): def __init__(self, id: str, params: Params) -> None: super().__init__(id, params) - self.digest_title_prefix = get_param("digest_title_prefix", params) - self.digest_description = params.get("digest_description", None) - self.digest_link = params.get("digest_link", None) - from_datatime_str = get_param("from_datetime", params) + self.digest_title_prefix = get_config(params, "digest_title_prefix") + self.digest_description = get_config_or_default( + params, "digest_description", None + ) + self.digest_link = get_config_or_default(params, "digest_link", None) + from_datatime_str = get_config(params, "from_datetime") self.from_datetime = datetime.fromisoformat(from_datatime_str) - interval = get_param("interval", params) + interval = get_config(params, "interval") self.interval_pair = parse_interval(interval) - self.max_length = int(params.get("max_length", "1000")) + self.max_length = int(get_config_or_default(params, "max_length", "1000")) print(f"[DigestPlugin#{self.id}] initialized") def process(self, source_id: str | None, items: list[Item]) -> list[Item]: -- cgit v1.3