aboutsummaryrefslogtreecommitdiffstats
path: root/app/FeedSinkPlugin.py
diff options
context:
space:
mode:
authorJan Tuomi <jans.tuomi@gmail.com>2023-09-03 19:07:30 +0300
committerJan Tuomi <jans.tuomi@gmail.com>2023-09-10 19:01:00 +0300
commit7f2e5ee5bf59e5814d62990401714e01cd69ba07 (patch)
tree0f3f9fcba77ccf45d81f6090df0195c7f34e98d3 /app/FeedSinkPlugin.py
parent7c90b13a7eaf520126a40180f932125240b46364 (diff)
Refactor plugins to own dir
Diffstat (limited to 'app/FeedSinkPlugin.py')
-rw-r--r--app/FeedSinkPlugin.py86
1 files changed, 0 insertions, 86 deletions
diff --git a/app/FeedSinkPlugin.py b/app/FeedSinkPlugin.py
deleted file mode 100644
index c0fc4c9..0000000
--- a/app/FeedSinkPlugin.py
+++ /dev/null
@@ -1,86 +0,0 @@
-import time
-from typing import Any
-import xml.etree.ElementTree as ET
-
-from tinydb import Query
-
-from app.Item import Item
-from app.PluginInterface import PluginInterface
-from app.utils import get_param
-from app.DatabaseManager import database_manager
-
-
-class Plugin(PluginInterface):
- def __init__(self, id: str, params: dict[str, Any]) -> None:
- super().__init__(id, 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 = 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 = self.feed_description
-
- for item in items:
- item_elem = ET.SubElement(channel, "item")
- if item.title is not None:
- item_title = ET.SubElement(item_elem, "title")
- item_title.text = item.title
- if item.link is not None:
- item_link = ET.SubElement(item_elem, "link")
- item_link.text = item.link
- if item.description is not None:
- item_description = ET.SubElement(item_elem, "description")
- item_description.text = item.description
- if item.category is not None:
- item_category = ET.SubElement(item_elem, "category")
- item_category.text = item.category
- if item.comments is not None:
- item_comments = ET.SubElement(item_elem, "comments")
- item_comments.text = item.comments
- if item.pub_date is not None:
- item_pub_date = ET.SubElement(item_elem, "pubDate")
- item_pub_date.text = time.strftime(
- "%a, %d %b %Y %H:%M:%S +0000", item.pub_date
- )
- if item.author is not None:
- item_author = ET.SubElement(item_elem, "author")
- item_author.text = item.author
-
- for enc in item.enclosures:
- ET.SubElement(
- item_elem,
- "enclosure",
- {
- "length": enc.length,
- "type": enc.type,
- "url": enc.url,
- },
- )
-
- return ET.tostring(rss, "unicode")
-
- 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)
-
- Q = Query()
- database_manager.feeds.upsert({"feed_id": self.feed_id, "feed_xml": ret}, Q.feed_id == self.feed_id) # type: ignore
- print(f"[FeedSinkPlugin#{self.id}] processed")
- return []