* verbose calendar data import

1 files changed, 79 insertions, 27 deletions

diff --git a/bin/import_cals.py b/bin/import_cals.py
index c21eb0c..8b32691 100644
--- a/bin/import_cals.py
+++ b/bin/import_cals.py
@@ -5,6 +5,8 @@ import time
 import urllib2
 import urlparse
 
+from sqlalchemy import inspect
+
 from rcal.db import Session
 from rcal.model import Calendar, Category
 
@@ -17,9 +19,7 @@ CONFIG = json.load(open(
     'r'))
 
 
-def main():
-    cal_list = None
-
+def get_cal_list():
     cache_path = os.path.join(
         os.path.dirname(os.path.realpath(__file__)),
         CONFIG['cache'])
@@ -35,35 +35,87 @@ def main():
             cache_file.close()
     else:
         cal_list = open(cache_path, 'r').read()
+    return cal_list
 
-    session = Session.create()
 
-    ical_markdown = re.compile(r'^\[iCal\]\((.*)\)$')
+def update_calendar(cal, session):
+    db_cal = Calendar.fetch(cal['uid'], session)
 
-    cells = [row.split('|') for row in cal_list.split('\n')]
+    if inspect(db_cal).pending:
+        print 'Adding calendar %s (%s)' % (cal['name'], cal['uid'])
+        db_cal.name = cal['name']
+        db_cal.url = cal['url']
+        db_cal.website = cal['website']
+        db_cal.category = Category.fetch(cal['category'], session)
+
+    if db_cal.name != cal['name']:
+        print 'Updating calendar name: %s -> %s (%s)' % (
+            db_cal.name, cal['name'], db_cal.uid)
+        db_cal.name = cal['name']
+    if db_cal.url != cal['url']:
+        print 'Updating calendar url: %s -> %s (%s)' % (
+            db_cal.url, cal['url'], db_cal.uid)
+        db_cal.url = cal['url']
+    if db_cal.website != cal['website']:
+        print 'Updating calendar website: %s -> %s (%s)' % (
+            db_cal.website, cal['website'], db_cal.uid)
+        db_cal.website = cal['website']
+    if db_cal.category.name != cal['category']:
+        print 'Updating calendar category: %s -> %s (%s)' % (
+            db_cal.category.name, cal['category'], db_cal.uid)
+        db_cal.category = Category.fetch(cal['category'], session)
+
+
+def get_imported_calendars(cells, ical_markdown):
+    imported_calendars = []
     for row in cells:
-        if len(row) == 7:
-            row = [r for r in row if r]
-            if len(row) == 5:
-                markdown_match = re.match(ical_markdown, row[2])
-                if markdown_match:
-                    ical_url = urlparse.urlparse(markdown_match.group(1))
-                    if ical_url.netloc == 'calendar.google.com':
-                        ical_path = re.sub(
-                            '^/?calendar/ical/', '', ical_url.path).split('/')
-                        if len(ical_path) == 3:
-                            calendar_uid = ical_path[0]
-                            calendar = Calendar.fetch(
-                                calendar_uid,
-                                session,
-                                row[0],
-                                ical_url.geturl())
-                            calendar.website = row[4]
-                            calendar.category = Category.fetch(row[1], session)
-                        else:
-                            print ical_url.geturl()
+        row = [r for r in row if r]
+        if len(row) == 5:
+            markdown_match = re.match(ical_markdown, row[2])
+            if markdown_match:
+                ical_url = urlparse.urlparse(markdown_match.group(1))
+                if ical_url.netloc == 'calendar.google.com':
+                    ical_path = re.sub(
+                        '^/?calendar/ical/', '', ical_url.path).split('/')
+                    if len(ical_path) == 3:
+                        imported_calendars.append({
+                            'uid': ical_path[0],
+                            'url': ical_url.geturl(),
+                            'name': row[0],
+                            'website': row[4],
+                            'category': row[1]
+                        })
                     else:
-                        print ical_url.geturl()
+                        print 'Unknown iCal URL format: %s' % (
+                            ical_url.geturl())
+                else:
+                    print 'Unknown iCal URL format: %s' % (
+                        ical_url.geturl())
+    return imported_calendars
+
+
+def main():
+    session = Session.create()
+
+    cal_list = get_cal_list()
+
+    ical_markdown = re.compile(r'^\[iCal\]\((.*)\)$')
+    cells = [row.split('|') for row in cal_list.split('\n')]
+
+    imported_calendars = get_imported_calendars(cells, ical_markdown)
+    imported_calendar_uids = [c['uid'] for c in imported_calendars]
+
+    db_only_calendars = session.query(Calendar).filter(
+        ~Calendar.uid.in_(imported_calendar_uids)).all()
+
+    if len(db_only_calendars):
+        print 'Local calendars not in remote source:'
+        for cal in db_only_calendars:
+            print '%s (%s)' % (cal.name, cal.uid)
+        print
+
+    for cal in imported_calendars:
+        update_calendar(cal, session)
 
     session.commit()