diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/AggroConfig.py | 2 | ||||
| -rw-r--r-- | app/FeedSinkPlugin.py | 52 | ||||
| -rw-r--r-- | app/database.py | 12 | ||||
| -rw-r--r-- | app/server.py | 31 |
4 files changed, 60 insertions, 37 deletions
diff --git a/app/AggroConfig.py b/app/AggroConfig.py index 6ff6801..7e6a5f3 100644 --- a/app/AggroConfig.py +++ b/app/AggroConfig.py @@ -3,6 +3,8 @@ from dataclasses import dataclass @dataclass class AggroConfig: + server_host: str + server_port: int db_path: str plugins: dict[str, dict[str, str]] graph: dict[str, list[str]] diff --git a/app/FeedSinkPlugin.py b/app/FeedSinkPlugin.py index 6b0125d..2c85167 100644 --- a/app/FeedSinkPlugin.py +++ b/app/FeedSinkPlugin.py @@ -1,47 +1,34 @@ from typing import Any -from app.Item import Item -from app.PluginInterface import PluginInterface -from app.utils import get_param import xml.etree.ElementTree as ET +from tinydb import Query -# <?xml version="1.0" encoding="UTF-8" ?> -# <rss version="2.0"> - -# <channel> -# <title>W3Schools Home Page</title> -# <link>https://www.w3schools.com</link> -# <description>Free web building tutorials</description> -# <item> -# <title>RSS Tutorial</title> -# <link>https://www.w3schools.com/xml/xml_rss.asp</link> -# <description>New RSS tutorial on W3Schools</description> -# </item> -# <item> -# <title>XML Tutorial</title> -# <link>https://www.w3schools.com/xml</link> -# <description>New XML tutorial on W3Schools</description> -# </item> -# </channel> - -# </rss> +from app.Item import Item +from app.PluginInterface import PluginInterface +from app.utils import get_param +from app.database import database_manager class Plugin(PluginInterface): def __init__(self, id: str, params: dict[str, Any]) -> None: super().__init__(id, params) - self.feed_name: str = get_param("feed_name", params) + self.feed_id: str = get_param("feed_id", params) + self.feed_title: str = get_param("feed_title", params) + self.feed_link: str | None = params.get("feed_link", None) + self.feed_description: str = get_param("feed_description", params) print(f"[FeedSinkPlugin#{self.id}] initialized") + print(f"[FeedSinkPlugin#{self.id}] feed will be served at path /{self.feed_id}") def build_xml(self, items: list[Item]): rss = ET.Element("rss", {"version": "2.0"}) channel = ET.SubElement(rss, "channel") title = ET.SubElement(channel, "title") - title.text = "channel title" - link = ET.SubElement(channel, "link") - link.text = "channel link" + title.text = self.feed_title + if self.feed_link is not None: + link = ET.SubElement(channel, "link") + link.text = self.feed_link description = ET.SubElement(channel, "description") - description.text = "channel description" + description.text = self.feed_description for item in items: item_elem = ET.SubElement(channel, "item") @@ -56,10 +43,15 @@ class Plugin(PluginInterface): def process(self, source_id: str | None, items: list[Item]) -> list[Item]: print(f"[FeedSinkPlugin#{self.id}] process called, n={len(items)}") + if source_id is None: raise Exception(f"FeedSinkPlugin#{self.id} can not be scheduled") + if database_manager.db is None: + raise Exception("Database is not initialized") ret = self.build_xml(items) - print(f"[FeedSinkPlugin#{self.id}] process returning XML:") - print(ret) + + Q = Query() + database_manager.db.upsert({"feed_id": self.feed_id, "feed_xml": ret}, Q.feed_id == self.feed_id) # type: ignore + print(f"[FeedSinkPlugin#{self.id}] processed") return [] diff --git a/app/database.py b/app/database.py index fdeac1a..b796dc8 100644 --- a/app/database.py +++ b/app/database.py @@ -3,14 +3,12 @@ from tinydb import TinyDB from app.AggroConfig import AggroConfig -def setup_db(config: AggroConfig): - global database_manager - database_manager = DatabaseManager(config) - - class DatabaseManager: - def __init__(self, config: AggroConfig): + def __init__(self): + self.db: TinyDB | None = None + + def setup(self, config: AggroConfig): self.db = TinyDB(config.db_path) -database_manager: DatabaseManager +database_manager: DatabaseManager = DatabaseManager() diff --git a/app/server.py b/app/server.py new file mode 100644 index 0000000..785e30f --- /dev/null +++ b/app/server.py @@ -0,0 +1,31 @@ +from typing import Any +import bottle as _bottle # type: ignore +from tinydb import Query + +from app.database import database_manager + +bottle: Any = _bottle + + +@bottle.route("/<feed_id>") +def index(feed_id: str): + if database_manager.db is None: + raise Exception("Database is not initialized") + + Q = Query() + res = database_manager.db.search(Q.feed_id == feed_id) + if len(res) == 0: + bottle.abort(400, f"No feed found with id {feed_id}") + + if len(res) > 1: + bottle.abort(400, f"Weird number of feeds found with id {feed_id}: {len(res)}") + + bottle.response.set_header("content-type", "application/xml") + + feed: Any = res[0] # type: ignore + feed_xml: str = feed["feed_xml"] + return feed_xml + + +def run_web_server(host: str, port: int): + bottle.run(host=host, port=port) |