parserrules.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (c) 2011 James French <[email protected]>
   4 #
   5 # Permission is hereby granted, free of charge, to any person obtaining a copy
   6 # of this software and associated documentation files (the "Software"), to deal
   7 # in the Software without restriction, including without limitation the rights
   8 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 # copies of the Software, and to permit persons to whom the Software is
  10 # furnished to do so, subject to the following conditions:
  11 #
  12 # The above copyright notice and this permission notice shall be included in
  13 # all copies or substantial portions of the Software.
  14 #
  15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21 # THE SOFTWARE.
  22
  23 # This file describes a series of rules which must handle a vobject object and
  24 # return it to the calling script
  25
  26 # The doc string will be presented to the user when run as verbose, so
  27 # please be polite
  28
  29 import vobject
  30
  31 ruleConfig = {}
  32 ruleConfig["defaultTZ"] = "UTC"
  33
  34 def facebookOrganiser(cal):
  35         '''Adds organiser details to the body of facebook calendars.'''
  36
  37         if cal.contents.has_key(u'prodid'):
  38                 if not "Facebook" in cal.prodid.value: return cal
  39
  40         for event in cal.vevent_list:
  41                 if not event.contents.has_key(u'organizer'): continue
  42                 try:
  43                         a = event.organizer.cn_paramlist
  44                         organizer = "Organised by: " + event.organizer.cn_param + " ("
  45                         organizer += event.organizer.value.lstrip('MAILTO:') + ")\n\n"
  46                         event.description.value = organizer + event.description.value
  47                 except AttributeError:
  48                         organizer = "Organized by: " + event.organizer.value
  49                         event.description.value = organizer + "\n\n" + event.description.value
  50         return cal
  51
  52 def whatPrivacy(cal):
  53         '''Marks events public so google calendar doesn't have a sad about them.'''
  54
  55         for event in cal.vevent_list:
  56                 if event.contents.has_key(u'class'):
  57                         # Bit of a hack as class is a reserved word in python
  58                         del event.contents[u'class']
  59                         event.add('class').value = "PUBLIC"
  60
  61         return cal
  62
  63 def exDate(cal):
  64         '''Replacing multi-value EXDATES with multiple single-value EXDATES'''
  65
  66         from datetime import datetime
  67         from pytz import timezone
  68
  69         default = timezone(ruleConfig["defaultTZ"])
  70
  71         for event in cal.vevent_list:
  72                 if not event.contents.has_key(u'exdate'): continue
  73                 dates = event.exdate.value
  74
  75                 del event.contents[u'exdate']
  76
  77                 for date in dates:
  78                         if isinstance(date, datetime):
  79                                 if date.tzinfo is None: date = date.replace(tzinfo = default)
  80                                 date = date.astimezone(vobject.icalendar.utc)
  81                         entry = event.add(u'exdate')
  82                         entry.value = [date]
  83
  84         return cal
  85
  86 def utcise(cal):
  87         '''Removing local timezones in favour of UTC. If the remote calendar specifies a timezone
  88         then use it, otherwise assume it's in the user-specified or default values'''
  89
  90         from datetime import datetime
  91         from pytz import timezone
  92
  93         default = timezone(ruleConfig["defaultTZ"])
  94
  95         for event in cal.vevent_list:
  96                 dtstart = getattr(event, 'dtstart', None)
  97                 dtend = getattr(event, 'dtend', None)
  98
  99                 for i in (dtstart, dtend):
 100                         if not i: continue
 101                         dt = i.value
 102                         if isinstance(dt, datetime):
 103                                 if dt.tzinfo is None: dt = dt.replace(tzinfo = default)
 104                                 i.value = dt.astimezone(vobject.icalendar.utc)
 105
 106         return cal
 107
 108 def unwantedParams(cal):
 109         '''Removing unwanted parameters'''
 110
 111         blklist = [
 112                 "LANGUAGE",
 113                 "X-VOBJ-ORIGINAL-TZID",
 114                 "TZID"
 115         ]
 116
 117         for event in cal.vevent_list:
 118                 for attr in event.contents:
 119                         attr = getattr(event, attr)
 120                         try:
 121                                 for i in blklist:
 122                                         while i in attr.params: del attr.params[i]
 123                         except AttributeError: continue
 124
 125         return cal