Move rss code to a package

1 files changed, 4 insertions, 174 deletions

diff --git a/pyblog b/pyblog
index a9abd8da..e44d2719 100755
--- a/pyblog
+++ b/pyblog
@@ -42,6 +42,8 @@ import requests
 
 import toml
 
+from rss import *
+
 ROOTDIR = os.path.dirname(os.path.realpath(__file__))
 
 
@@ -121,178 +123,6 @@ def current_datetime():
     return datetime.datetime.fromtimestamp(round(time.time()),
                                            dateutil.tz.tzlocal())
 
-
-class AtomFeed(object):
-    """Class for storing atom:feed data and metadata.
-
-    https://tools.ietf.org/html/rfc4287.
-
-    """
-
-    # pylint: disable=invalid-name,too-many-instance-attributes
-
-    def __init__(self):
-        """Define available attributes."""
-        self.author = None  # atom:author
-        self.generator = None  # atom:generator, optional
-        self.icon = None  # atom:icon, optional
-        self.logo = None  # atom:logo, optional
-        self.id_text = None  # atom:id, just use URI
-        self.id = None  # atom:id
-        self.links = []  # list of atom:link
-        self.title_text = None  # the text of atom:title
-        self.title = None  # atom:title
-        self.subtitle_text = None  # the text of atom:subtitle
-        self.subtitle = None  # atom:subtitle
-        self.updated_datetime = None  # update time as a datetime object
-        self.updated = None  # atom:updated
-        self.entries = []  # list of atom:entry, in reverse time order
-        self.feed = None  # atom:feed, assembled
-
-    def assemble_feed(self):
-        """Assemble atom:feed."""
-        # pylint: disable=multiple-statements
-        self.feed = ET.Element("feed", xmlns="http://www.w3.org/2005/Atom")
-        self.feed.append(self.title)
-        if self.subtitle is not None: self.feed.append(self.subtitle)
-        for link in self.links:
-            self.feed.append(link)
-        self.feed.append(self.updated)
-        self.feed.append(self.id)
-        self.feed.append(self.author)
-        if self.icon is not None: self.feed.append(self.icon)
-        if self.logo is not None: self.feed.append(self.icon)
-        if self.generator is not None: self.feed.append(self.generator)
-        # include at most FEED_MAX_ENTRIES entries in the feed
-        for entry in self.entries[:FEED_MAX_ENTRIES]:
-            self.feed.append(entry.entry)
-
-    def dump_feed(self):
-        """Dump atom:feed XML."""
-        if self.feed is None:
-            self.assemble_feed()
-        return ET.tostring(self.feed).decode("utf-8")
-
-
-class AtomEntry(object):
-    """Class for storing atom:entry data and metadata."""
-
-    # pylint: disable=invalid-name,too-many-instance-attributes
-
-    def __init__(self):
-        """Define available attributes."""
-        self.author = None  # atom:author
-        self.id_text = None  # atom:id, just use URI
-        self.id = None  # atom:id
-        self.relpath = None  # HTML page path relative to home
-        self.link = None  # atom:link
-        self.title_text = None  # plain text title
-        self.title = None  # atom:title
-        self.updated_datetime = None  # update time as a datetime object
-        self.updated = None  # atom:updated
-        self.content_html = None  # content as HTML markup
-        self.content = None  # atom:content
-        self.entry = None  # atom:entry, assembled
-
-    def assemble_entry(self):
-        """Assemble atom:entry."""
-        self.entry = ET.Element("entry")
-        self.entry.append(self.title)
-        self.entry.append(self.link)
-        self.entry.append(self.updated)
-        self.entry.append(self.id)
-        self.entry.append(self.author)
-        self.entry.append(self.content)
-
-    def dump_entry(self):
-        """Dump atom:entry XML."""
-        if self.entry is None:
-            self.assemble_entry()
-        return ET.tostring(self.entry).decode("utf-8")
-
-
-class RssFeed(object):
-    """Class for storing an RSS 2.0 feed.
-
-    https://validator.w3.org/feed/docs/rss2.html.
-
-    """
-
-    # pylint: disable=too-many-instance-attributes
-
-    REQUIRED_ELEMENTS = ["title", "link", "description"]
-    OPTIONAL_ELEMENTS = ["language", "copyright", "managingEditor", "webMaster",
-                         "pubDate", "lastBuildDate", "category", "generator",
-                         "docs", "cloud", "ttl", "image", "textInput",
-                         "skipHours", "skipDays"]
-
-    def __init__(self):
-        """Define available attributes."""
-        self.rssurl = None  # the URL of the rss feed
-        self.atomlink = None
-        for element in self.REQUIRED_ELEMENTS:
-            setattr(self, element, None)
-        for element in self.OPTIONAL_ELEMENTS:
-            setattr(self, element, None)
-        self.docs = ET.Element("docs")
-        self.docs.text = "https://validator.w3.org/feed/docs/rss2.html"
-        self.author_text = None
-        self.update_timestamp = None
-        self.items = []
-        self.rss = None
-        self.channel = None
-
-    def assemble_rss(self):
-        """Assemble RSS 2.0 feed."""
-        self.rss = ET.Element("rss", version="2.0", nsmap={"atom": "http://www.w3.org/2005/Atom"})
-        self.channel = ET.SubElement(self.rss, "channel")
-        # https://validator.w3.org/feed/docs/warning/MissingAtomSelfLink.html
-        self.atomlink = ET.SubElement(self.channel, "{http://www.w3.org/2005/Atom}link",
-                                      href=self.rssurl, rel="self", type="application/rss+xml")
-        for element in self.REQUIRED_ELEMENTS:
-            self.channel.append(getattr(self, element))
-        for element in self.OPTIONAL_ELEMENTS:
-            attr = getattr(self, element)
-            if attr is not None:
-                self.channel.append(attr)
-        # include at most FEED_MAX_ENTRIES items in the RSS feed
-        for item in self.items[:FEED_MAX_ENTRIES]:
-            self.channel.append(item.item)
-
-    def dump_rss(self):
-        """Dump RSS feed XML."""
-        if self.rss is None:
-            self.assemble_rss()
-        return ET.tostring(self.rss).decode("utf-8")
-
-
-class RssItem(object):
-    """Class for storing an RSS 2.0 item."""
-
-    ELEMENTS = ["title", "link", "description", "author", "category", "comments",
-                "enclosure", "guid", "pubDate", "source"]
-
-    def __init__(self):
-        """Define available attributes."""
-        for element in self.ELEMENTS:
-            setattr(self, element, None)
-        self.timestamp = None
-        self.item = None
-
-    def assemble_item(self):
-        """Assemble an RSS 2.0 item."""
-        self.item = ET.Element("item")
-        for element in self.ELEMENTS:
-            attr = getattr(self, element)
-            if attr is not None:
-                self.item.append(attr)
-
-    def dump_item(self):
-        """Dump RSS item XML."""
-        if self.item is None:
-            self.assemble_item()
-        return ET.tostring(self.item).decode("utf-8")
-
 def generate_menu():
     """Generate menu."""
 
@@ -1002,11 +832,11 @@ def generate_index_and_feed():
     rss.lastBuildDate.text = email.utils.formatdate(rss.update_timestamp, usegmt=True)
 
     with open(ATOM, "w", encoding="utf-8") as atom:
-        atom.write("%s\n" % feed.dump_feed())
+        atom.write("%s\n" % feed.dump_feed(FEED_MAX_ENTRIES))
         sys.stderr.write("wrote atom.xml\n")
 
     with open(RSS, "w", encoding="utf-8") as rssxml:
-        rssxml.write("%s\n" % rss.dump_rss())
+        rssxml.write("%s\n" % rss.dump_rss(FEED_MAX_ENTRIES))
         sys.stderr.write("wrote rss.xml\n")
 
     generate_sitemap(feed)