First (working) version

3 files changed, 202 insertions, 0 deletions

diff --git a/afeedprocessor/afeedparser.py b/afeedprocessor/afeedparser.py
new file mode 100644
index 0000000..0c23187
--- /dev/null
+++ b/afeedprocessor/afeedparser.py
@@ -0,0 +1,59 @@
+import feedparser
+import PyRSS2Gen
+import datetime
+
+
+class FeedParser:
+    @staticmethod
+    def date_tuple_to_datetime(date_tuple):
+        return datetime.datetime(*(date_tuple[:5])) if date_tuple else None
+
+    @staticmethod
+    def get_first(lst):
+        return lst[0] if lst and len(lst) > 0 else None
+
+    def get_rss_item_for_entry(self, entry):
+        return PyRSS2Gen.RSSItem(
+            title=entry.get('title'),
+            link=entry.get('link'),
+            description=entry.get('description'),
+            author=entry.get('author'),
+            categories=entry.get('tags'),
+            comments=entry.get('comments'),
+            enclosure=self.get_first(entry.get('enclosures')),
+            guid=entry.get('id'),
+            pubDate=self.date_tuple_to_datetime(entry.get('published_parsed')),
+            source=entry.get('source'),
+        )
+
+    def get_rss2_from_feed(self, feed, entries):
+        return PyRSS2Gen.RSS2(
+            title=feed.get('title'),
+            link=feed.get('link'),
+            description=feed.get('subtitle'),
+
+            language=feed.get('language'),
+            copyright=feed.get('rights'),
+            managingEditor=feed.get('contributors'),
+            webMaster=feed.get('publisher'),
+            pubDate=self.date_tuple_to_datetime(feed.get('published_parsed')),
+            lastBuildDate=self.date_tuple_to_datetime(feed.get('updated_parsed')),
+
+            categories=feed.get('tags'),
+            generator=feed.get('generator'),
+            docs=feed.get('docs'),
+            cloud=feed.get('cloud'),
+            ttl=feed.get('ttl'),
+
+            image=feed.get('image'),
+            rating=None,
+            textInput=feed.get('textinput'),
+            skipHours=None,
+            skipDays=None,
+
+            items=[self.get_rss_item_for_entry(entry) for entry in entries],
+        )
+
+    def parse(self, feed):
+        parsed_feed = feedparser.parse(feed)
+        return self.get_rss2_from_feed(parsed_feed.feed, parsed_feed.entries)
diff --git a/afeedprocessor/afeedprocessor.py b/afeedprocessor/afeedprocessor.py
new file mode 100644
index 0000000..e3663ee
--- /dev/null
+++ b/afeedprocessor/afeedprocessor.py
@@ -0,0 +1,96 @@
+import PyRSS2Gen
+
+from afeedprocessor.anitemprocessor import ItemProcessor
+
+
+class FeedProcessor:
+    def __init__(self, item_processor: ItemProcessor=None):
+        if item_processor is None:
+            self.item_processor = ItemProcessor()
+        else:
+            self.item_processor = item_processor
+
+    def get_title(self, title, feed):
+        return title
+
+    def get_link(self, link, feed):
+        return link
+
+    def get_description(self, description, feed):
+        return description
+
+    def get_language(self, language, feed):
+        return language
+
+    def get_copyright(self, copyright, feed):
+        return copyright
+
+    def get_managing_editor(self, managing_editor, feed):
+        return managing_editor
+
+    def get_web_master(self, web_master, feed):
+        return web_master
+
+    def get_pub_date(self, pub_date, feed):
+        return pub_date
+
+    def get_last_build_date(self, last_build_date, feed):
+        return last_build_date
+
+    def get_categories(self, categories, feed):
+        return categories
+
+    def get_generator(self, generator, feed):
+        return generator
+
+    def get_docs(self, docs, feed):
+        return docs
+
+    def get_cloud(self, cloud, feed):
+        return cloud
+
+    def get_ttl(self, ttl, feed):
+        return ttl
+
+    def get_image(self, image, feed):
+        return image
+
+    def get_rating(self, rating, feed):
+        return rating
+
+    def get_text_input(self, text_input, feed):
+        return text_input
+
+    def get_skip_hours(self, skip_hours, feed):
+        return skip_hours
+
+    def get_skip_days(self, skip_days, feed):
+        return skip_days
+
+    def get_items(self, items, feed):
+        return [self.item_processor.process(item) for item in items]
+
+    def process(self, feed: PyRSS2Gen.RSS2):
+        return PyRSS2Gen.RSS2(
+            title=self.get_title(feed.title, feed),
+            link=self.get_link(feed.link, feed),
+            description=self.get_description(feed.description, feed),
+            language=self.get_language(feed.language, feed),
+            copyright=self.get_copyright(feed.copyright, feed),
+            managingEditor=self.get_managing_editor(feed.managingEditor, feed),
+            webMaster=self.get_web_master(feed.webMaster, feed),
+            pubDate=self.get_pub_date(feed.pubDate, feed),
+            lastBuildDate=self.get_last_build_date(feed.lastBuildDate, feed),
+            categories=self.get_categories(feed.categories, feed),
+            generator=self.get_generator(feed.generator, feed),
+            docs=self.get_docs(feed.docs, feed),
+            cloud=self.get_cloud(feed.cloud, feed),
+            ttl=self.get_ttl(feed.ttl, feed),
+            image=self.get_image(feed.image, feed),
+            rating=self.get_rating(feed.rating, feed),
+            textInput=self.get_text_input(feed.textInput, feed),
+            skipHours=self.get_skip_hours(feed.skipHours, feed),
+            skipDays=self.get_skip_days(feed.skipDays, feed),
+            items=self.get_items(feed.items, feed)
+            # Hurray for code generation!
+        )
diff --git a/afeedprocessor/anitemprocessor.py b/afeedprocessor/anitemprocessor.py
new file mode 100644
index 0000000..3f0eabc
--- /dev/null
+++ b/afeedprocessor/anitemprocessor.py
@@ -0,0 +1,47 @@
+import PyRSS2Gen
+
+
+class ItemProcessor:
+    def get_title(self, title, item):
+        return title
+
+    def get_link(self, link, item):
+        return link
+
+    def get_description(self, description, item):
+        return description
+
+    def get_author(self, author, item):
+        return author
+
+    def get_categories(self, categories, item):
+        return categories
+
+    def get_comments(self, comments, item):
+        return comments
+
+    def get_enclosure(self, enclosure, item):
+        return enclosure
+
+    def get_guid(self, guid, item):
+        return guid
+
+    def get_pub_date(self, pub_date, item):
+        return pub_date
+
+    def get_source(self, source, item):
+        return source
+
+    def process(self, item: PyRSS2Gen.RSSItem):
+        return PyRSS2Gen.RSSItem(
+            title=self.get_title(item.title, item),
+            link=self.get_link(item.link, item),
+            description=self.get_description(item.description, item),
+            author=self.get_author(item.author, item),
+            categories=self.get_categories(item.categories, item),
+            comments=self.get_comments(item.comments, item),
+            enclosure=self.get_enclosure(item.enclosure, item),
+            guid=self.get_guid(item.guid, item),
+            pubDate=self.get_pub_date(item.pubDate, item),
+            source=self.get_source(item.source, item)
+        )