atom

1 files changed, 80 insertions, 0 deletions

diff --git a/rss_to_gitea/atom.py b/rss_to_gitea/atom.py
new file mode 100644
index 0000000..a19a969
--- /dev/null
+++ b/rss_to_gitea/atom.py
@@ -0,0 +1,80 @@
+import xml.etree.ElementTree as ET
+import requests
+import json
+
+from datetime import datetime
+
+def containts_any_of(_str: str, _list: list[str]):
+    return True in map(lambda a: a in _str, _list)
+
+def date_compare(date1: str, date2: str):
+    d1 = parse_date(date1)
+    d2 = parse_date(date2)
+    return d1 > d2
+
+def parse_date(date: str):
+    '''
+    Parse RFC 3339 Date
+    '''
+    return datetime.strptime(date, "%Y-%m-%dT%H:%M:%SZ")
+
+class FeedException(Exception):
+    pass
+
+class AtomFeed:
+    def __init__(self, _url):
+        self.url = _url
+        xml = self._get_feed()
+        entries = self._parse(xml)
+        self.entries = entries
+        pass
+
+    def _get_feed(self):
+        resp = requests.get(self.url)
+        if resp.status_code >= 400:
+            raise FeedException(f'Failed to get Feed: {resp.status_code}')
+
+        return resp.content
+
+    def get_latest(self, _exclude: list[str] = [], _include: list[str] = []):
+        '''
+        _include is ignored if empty. NOT REGEX!!
+        '''
+        latest ={ }
+
+        for entry in self.entries:
+            if len(latest) <= 0  or date_compare(entry['updated'], latest['updated']):
+                if containts_any_of(entry['title'], _exclude):
+                    continue
+                if not (len(_include) == 0 or containts_any_of(entry['title'], _include)):
+                    continue
+
+                latest = entry
+
+        return latest
+
+
+    def _parse(self, xml):
+        tree = ET.fromstring(xml)
+
+        entries = []
+
+        for entry in tree:
+            if not entry.tag.endswith('entry'):
+                continue
+
+            data = {}
+
+            for att in entry:
+                if att.tag.endswith('updated'):
+                    data['updated'] = att.text
+
+                if att.tag.endswith('title'):
+                    data['title'] = att.text
+
+                if att.tag.endswith('link'):
+                    data['link'] = att.attrib['href']
+
+            entries.append(data)
+
+        return entries