Tabs vs Spaces

[frenchie/icalparse.git] / parserrules.py

diff --git a/parserrules.py b/parserrules.py
index 791a860..7db0413 100644 (file)
--- a/parserrules.py
+++ b/parserrules.py
@@ -28,14 +28,20 @@
 
 import vobject
 
+ruleConfig = {}
+ruleConfig["defaultTZ"] = "UTC"
+
 def facebookOrganiser(cal):
        '''Adds organiser details to the body of facebook calendars.'''
 
-       if cal.contents.has_key(u'prodid'):
+       if 'prodid' in cal.contents:
                if not "Facebook" in cal.prodid.value: return cal
 
+       if 'vevent_list' not in dir(cal):
+               return cal
+
        for event in cal.vevent_list:
-               if not event.contents.has_key(u'organizer'): continue
+               if 'organizer' not in event.contents: continue
                try:
                        a = event.organizer.cn_paramlist
                        organizer = "Organised by: " + event.organizer.cn_param + " ("
@@ -49,14 +55,24 @@ def facebookOrganiser(cal):
 def whatPrivacy(cal):
        '''Marks events public so google calendar doesn't have a sad about them.'''
 
+       if 'vevent_list' not in dir(cal):
+               return cal
+
        for event in cal.vevent_list:
-               if event.contents.has_key(u'class'):
-                       getattr(event, 'class').value = "PUBLIC"
+               if 'class' in event.contents:
+                       # Bit of a hack as class is a reserved word in python
+                       del event.contents['class']
+                       event.add('class').value = "PUBLIC"
 
        return cal
 
 def dropAttributes(cal):
        '''Removing unwanted metadata'''
+       if "facebook" in ruleConfig:
+               if ruleConfig["facebook"] == True: return cal
+
+       if 'vevent_list' not in dir(cal):
+               return cal
 
        eventBlacklist = [x.lower() for x in [
                "X-ALT-DESC",
@@ -89,7 +105,7 @@ def dropAttributes(cal):
 
        for event in cal.vevent_list:
                for blacklist in eventBlacklist:
-                       if event.contents.has_key(blacklist): del event.contents[blacklist]
+                       if blacklist in event.contents: del event.contents[blacklist]
 
        for blkl in mainBlacklist:
                while blkl in cal.contents: del cal.contents[blkl]
@@ -98,37 +114,43 @@ def dropAttributes(cal):
 
 def exDate(cal):
        '''Replacing multi-value EXDATES with multiple single-value EXDATES'''
+       if "facebook" in ruleConfig:
+               if ruleConfig["facebook"] == True: return cal
+
+       if 'vevent_list' not in dir(cal):
+               return cal
 
        from datetime import datetime
        from pytz import timezone
 
-       default = timezone('Australia/Perth')
-
+       default = timezone(ruleConfig["defaultTZ"])
 
        for event in cal.vevent_list:
-               if not event.contents.has_key(u'exdate'): continue
+               if 'exdate' not in event.contents: continue
                dates = event.exdate.value
 
-               del event.contents[u'exdate']
+               del event.contents['exdate']
 
                for date in dates:
-                       print date
                        if isinstance(date, datetime):
                                if date.tzinfo is None: date = date.replace(tzinfo = default)
                                date = date.astimezone(vobject.icalendar.utc)
-                       print date
-                       entry = event.add(u'exdate')
+                       entry = event.add('exdate')
                        entry.value = [date]
 
        return cal
 
 def utcise(cal):
-       '''Removing local timezones in favour of UTC'''
+       '''Removing local timezones in favour of UTC. If the remote calendar specifies a timezone
+       then use it, otherwise assume it's in the user-specified or default values'''
+
+       if 'vevent_list' not in dir(cal):
+               return cal
 
        from datetime import datetime
        from pytz import timezone
 
-       default = timezone('Australia/Perth')
+       default = timezone(ruleConfig["defaultTZ"])
 
        for event in cal.vevent_list:
                dtstart = getattr(event, 'dtstart', None)
@@ -152,6 +174,9 @@ def unwantedParams(cal):
                "TZID"
        ]
 
+       if 'vevent_list' not in dir(cal):
+               return cal
+
        for event in cal.vevent_list:
                for attr in event.contents:
                        attr = getattr(event, attr)
@@ -161,4 +186,3 @@ def unwantedParams(cal):
                        except AttributeError: continue
 
        return cal
-