1 files changed, 58 insertions, 0 deletions

diff --git a/email_assistant/plugins/general.py b/email_assistant/plugins/general.py
new file mode 100644
index 0000000..c0517b8
--- /dev/null
+++ b/email_assistant/plugins/general.py
@@ -0,0 +1,58 @@
+
+# An unused sketch of a generalized plugin that uses commonregex to
+# find dates and addresses.
+
+import re
+import logging
+import hashlib
+import datetime
+
+from bs4 import BeautifulSoup
+import commonregex
+import dateutil.parser
+import dateutil.tz
+import inscriptis
+import vobject
+
+def match(msg):
+    if ('reservations@res-marriott.com' in msg['From'] and
+        'Reservation Confirmation' in msg['Subject']):
+        return True
+
+def get_events(msg):
+    events = []
+    log = logging.getLogger('assistant.marriott')
+    for part in msg.walk():
+        if part.get_content_type() == 'text/html':
+            soup = BeautifulSoup(part.get_payload(decode=True).decode('utf8'), 'html.parser')
+
+            start = end = location = None
+            for element in soup.descendants:
+                if not hasattr(element, 'children') and element.string:
+                    text = element.string.strip()
+                    if not text:
+                        continue
+                    found = commonregex.CommonRegex(text)
+                    for date in found.dates:
+                        if not start:
+                            start = dateutil.parser.parse(date).date()
+                            continue
+                        if not end:
+                            end = dateutil.parser.parse(date).date()+datetime.timedelta(days=1)
+                    if found.street_addresses and not location:
+                        location = text
+            if not (start and end and location):
+                continue
+            cal = vobject.iCalendar()
+            event = cal.add('vevent')
+            event.add('dtstart').value = start
+            event.add('dtend').value = end
+            summary = re.match('Reservation Confirmation .*? for (.*)', msg['Subject']).group(1).strip()
+            event.add('summary').value = summary
+            event.add('description').value = inscriptis.get_text(str(soup))
+            event.add('location').value = location
+            uid = hashlib.sha1((str(start) + summary).encode('utf8')).hexdigest()
+            event.add('uid').value = uid
+            events.append(cal)
+            print(repr(cal.serialize()))
+    return events