X-Git-Url: https://git.ucc.asn.au/?p=frenchie%2Ficalparse.git;a=blobdiff_plain;f=parserrules.py;h=4ba71e8a3544d20e9a5502c22d6d059f9bb2a96a;hp=cf60f984f6695d3ef398b7b4154bd37306c786e0;hb=HEAD;hpb=4cd352c4a6906d90e171e36d3027e803009e4af6

diff --git a/parserrules.py b/parserrules.py
index cf60f98..7db0413 100644
--- 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,16 +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'):
+		if 'class' in event.contents:
 			# Bit of a hack as class is a reserved word in python
-			del event.contents[u'class']
+			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",
@@ -91,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]
@@ -100,35 +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:
 			if isinstance(date, datetime):
 				if date.tzinfo is None: date = date.replace(tzinfo = default)
 				date = date.astimezone(vobject.icalendar.utc)
-			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,21 +186,3 @@ def unwantedParams(cal):
 			except AttributeError: continue
 
 	return cal
-
-def exDate(cal):
-	'''Changes multi-EXDATE into singles (apple can't obey even simple specs)'''
-
-	for event in cal.vevent_list:
-		if not event.contents.has_key(u'exdate'): continue
-		dates = event.exdate.value
-		try: tzid = event.exdate.tzid_param
-		except AttributeError: tzid = ''
-
-		del event.contents[u'exdate']
-
-		for date in dates:
-			entry = event.add(u'exdate')
-			entry.value = [date]
-			if tzid: entry.tzid_param = tzid
-
-	return cal