def setUp(self):
c = vobject.base.readComponents(open("test.ics","r")).next()
self.ical = ICal(c)
c = vobject.base.readComponents(open("test2.ics","r")).next()
self.ical2 = ICal(c)
c = vobject.base.readComponents(open("onlytodo.ics","r")).next()
self.ical3 = ICal(c)
def GetEvents(self, since, until, force_reload):
wc = WebCal(url, username, passwd)
uids = wc.get_calendar_uids()
isoformat = "%Y-%m-%dT%H:%M:%S.%f"
events = []
for uid in uids:
cal = ICal(wc.get_calendar(uid))
feed = cal.get_events()
for event in feed:
start_date = event.get_start_datetime()
end_date = event.get_end_datetime()
if ((end_date - start_date) == timedelta(1)):
allday = True
else:
allday = False
start = int(mktime(start_date.timetuple()))
end = int(mktime(end_date.timetuple()))
events.append((
'',
str(event.get_summary()),
'',
allday,
start,
end,
{}))
return events
def setUp(self):
c = vobject.base.readComponents(open("test.ics", "r")).next()
self.ical = ICal(c)
c = vobject.base.readComponents(open("test2.ics", "r")).next()
self.ical2 = ICal(c)
c = vobject.base.readComponents(open("onlytodo.ics", "r")).next()
self.ical3 = ICal(c)
class ICalTest(unittest.TestCase):
def setUp(self):
c = vobject.base.readComponents(open("test.ics","r")).next()
self.ical = ICal(c)
c = vobject.base.readComponents(open("test2.ics","r")).next()
self.ical2 = ICal(c)
c = vobject.base.readComponents(open("onlytodo.ics","r")).next()
self.ical3 = ICal(c)
def test_get_event_ids(self):
ids = self.ical.get_event_ids()
self.assertEqual(32, len(ids))
ids = self.ical2.get_event_ids()
self.assertEqual(1, len(ids))
ids = self.ical3.get_event_ids()
self.assertEqual(0, len(ids))
def test_get_events(self):
r = self.ical.get_events()
self.assertEqual(32, len(r))
def test_datetime(self):
ids = self.ical.get_events()
#test with date
dts1 = date(2010, 8, 13)
dtsret1 = ids[0].get_start_datetime()
self.assertEqual(dts1, dtsret1)
dte1 = date(2010, 8, 14)
dteret1 = ids[0].get_end_datetime()
self.assertEqual(dte1, dteret1)
dts1 = datetime(2010, 8, 13, 0, 0, 0, 0)
dtsret1 = ids[0].get_start_datetime()
# comparing tz-aware and unaware dates
self.assertRaises(TypeError, lambda(x,y): x == y, dts1, dtsret1)
ids[0].set_end_datetime(dteret1 + timedelta(days=1))
ids = self.ical.get_events()
dte1new = date(2010, 8, 15)
dteret1new = ids[0].get_end_datetime()
self.assertEqual(dte1new, dteret1new)
# test with datetime
dts1 = datetime(2010, 7, 24, 0, 0, 0, 0, UTC())
dtsret1 = ids[1].get_start_datetime()
self.assertEqual(dts1, dtsret1)
dte1 = datetime(2010, 7, 24, 23, 59, 0, 0, UTC())
dteret1 = ids[1].get_end_datetime()
self.assertEqual(dte1, dteret1)
dts1 = datetime(2010, 7, 25, 0, 0, 0, 0)
dtsret1 = ids[1].get_start_datetime()
# comparing tz-aware and unaware dates
self.assertRaises(TypeError, lambda(x,y): x == y, dts1, dtsret1)
ids[1].set_end_datetime(dteret1 + timedelta(hours=1))
ids = self.ical.get_events()
dte1new = datetime(2010, 7, 25, 0, 59, 0, 0, UTC())
dteret1new = ids[1].get_end_datetime()
self.assertEqual(dte1new, dteret1new)
def test_summary(self):
ids = self.ical.get_events()
sum_first = 'Grape Festival 2010'
sum_last = 'Blind Guardian + Enforcer + more at Gasometer'
self.assertEqual(sum_first, ids[0].get_summary())
self.assertEqual(sum_last, ids[-1].get_summary())
sum_test = 'Testing sum'
ids[0].set_summary(sum_test)
ids = self.ical.get_events()
self.assertEqual(sum_test, ids[0].get_summary())
self.assertEqual(sum_last, ids[-1].get_summary())
def test_location(self):
ids = self.ical.get_events()
loc_first = 'Letisko, Slovakia'
loc_last = 'Gasometer, Austria'
self.assertEqual(loc_first, ids[0].get_location())
self.assertEqual(loc_last, ids[-1].get_location())
def test_events_dates(self):
before = self.ical.events_before(datetime(2010, 7, 10, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(before))
before = self.ical.events_before(datetime(2010, 12, 7, 0, 0, 0, 0, UTC()))
self.assertEqual(32, len(before))
before = self.ical.events_before(datetime(2010, 10, 3, 0, 0, 0, 0, UTC()))
self.assertEqual(12, len(before))
between = self.ical.events_between(
datetime(2010, 8, 20, 0, 0, 0, 0, UTC()),
datetime(2010, 8, 24, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(between))
between = self.ical.events_between(
datetime(2010, 8, 12, 0, 0, 0, 0, UTC()),
datetime(2010, 8, 24, 0, 0, 0, 0, UTC()))
self.assertEqual(2, len(between))
after = self.ical.events_after(datetime(2010, 7, 10, 0, 0, 0, 0, UTC()))
self.assertEqual(32, len(after))
after = self.ical.events_after(datetime(2010, 8, 20, 0, 0, 0, 0, UTC()))
self.assertEqual(26, len(after))
after = self.ical.events_after(datetime(2010, 12, 7, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(after))
after = self.ical3.events_after(datetime(2011, 3, 3, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(after))
def test_url(self):
ids = self.ical.get_events()
url = ids[0].get_url()
self.assertEqual(url, "http://www.last.fm/festival/1416224+Grape+Festival+2010")
url = ids[-1].get_url()
self.assertEqual(url, "http://www.last.fm/event/1328092+Blind+Guardian+at+Gasometer+on+16+October+2010")
def test_attendees(self):
ids = self.ical2.get_events()
at = ids[0].get_attendees()
self.assertEquals(2, len(at))
self.assertEquals(u"Milgrim", at[0].name)
self.assertEquals("mailto:[email protected]", at[0].address)
self.assertEquals("REQ-PARTICIPANT", at[0].role)
self.assertEquals("TRUE", at[0].rsvp_request)
self.assertEquals("NEEDS-ACTION", at[0].rsvp_status)
self.assertEquals(str(at[0]), "ATTENDEE;RSVP=TRUE;ROLE=REQ-PARTICIPANT;CN=Milgrim;PARTSTAT=NEEDS-ACTION:m\r\n ailto:[email protected]\r\n")
self.assertEquals("Idoru", at[1].name)
self.assertEquals("mailto:[email protected]", at[1].address)
self.assertEquals("REQ-PARTICIPANT", at[1].role)
self.assertEquals("TRUE", at[1].rsvp_request)
self.assertEquals("DECLINED", at[1].rsvp_status)
self.assertEquals(str(at[1]), "ATTENDEE;RSVP=TRUE;ROLE=REQ-PARTICIPANT;CN=Idoru;PARTSTAT=DECLINED:mailto:\r\n [email protected]\r\n")
#================== CONFIG ================
url = 'CALENDAR URL'
username = '******'
passwd = 'YOUR PASSWORD'
#================ END CONFIG ==============
from pywebcal import WebCal, ICal
from datetime import datetime, timedelta
from dateutil.tz import tzical, gettz
wc = WebCal(url, username, passwd)
uids = wc.get_calendar_uids()
n = datetime.now(gettz())
u = n + timedelta(days=7)
events = []
for uid in uids:
cal = ICal(wc.get_calendar(uid))
es = cal.events_between(n, u)
for dt, e in es:
events.append((e.get_summary(), e.get_start_datetime()))
print "Events in next week: %s\n" % events
class ICalTest(unittest.TestCase):
def setUp(self):
c = vobject.base.readComponents(open("test.ics", "r")).next()
self.ical = ICal(c)
c = vobject.base.readComponents(open("test2.ics", "r")).next()
self.ical2 = ICal(c)
c = vobject.base.readComponents(open("onlytodo.ics", "r")).next()
self.ical3 = ICal(c)
def test_get_event_ids(self):
ids = self.ical.get_event_ids()
self.assertEqual(32, len(ids))
ids = self.ical2.get_event_ids()
self.assertEqual(1, len(ids))
ids = self.ical3.get_event_ids()
self.assertEqual(0, len(ids))
def test_get_events(self):
r = self.ical.get_events()
self.assertEqual(32, len(r))
def test_datetime(self):
ids = self.ical.get_events()
#test with date
dts1 = date(2010, 8, 13)
dtsret1 = ids[0].get_start_datetime()
self.assertEqual(dts1, dtsret1)
dte1 = date(2010, 8, 14)
dteret1 = ids[0].get_end_datetime()
self.assertEqual(dte1, dteret1)
dts1 = datetime(2010, 8, 13, 0, 0, 0, 0)
dtsret1 = ids[0].get_start_datetime()
# comparing tz-aware and unaware dates
self.assertRaises(TypeError, lambda (x, y): x == y, dts1, dtsret1)
ids[0].set_end_datetime(dteret1 + timedelta(days=1))
ids = self.ical.get_events()
dte1new = date(2010, 8, 15)
dteret1new = ids[0].get_end_datetime()
self.assertEqual(dte1new, dteret1new)
# test with datetime
dts1 = datetime(2010, 7, 24, 0, 0, 0, 0, UTC())
dtsret1 = ids[1].get_start_datetime()
self.assertEqual(dts1, dtsret1)
dte1 = datetime(2010, 7, 24, 23, 59, 0, 0, UTC())
dteret1 = ids[1].get_end_datetime()
self.assertEqual(dte1, dteret1)
dts1 = datetime(2010, 7, 25, 0, 0, 0, 0)
dtsret1 = ids[1].get_start_datetime()
# comparing tz-aware and unaware dates
self.assertRaises(TypeError, lambda (x, y): x == y, dts1, dtsret1)
ids[1].set_end_datetime(dteret1 + timedelta(hours=1))
ids = self.ical.get_events()
dte1new = datetime(2010, 7, 25, 0, 59, 0, 0, UTC())
dteret1new = ids[1].get_end_datetime()
self.assertEqual(dte1new, dteret1new)
def test_summary(self):
ids = self.ical.get_events()
sum_first = 'Grape Festival 2010'
sum_last = 'Blind Guardian + Enforcer + more at Gasometer'
self.assertEqual(sum_first, ids[0].get_summary())
self.assertEqual(sum_last, ids[-1].get_summary())
sum_test = 'Testing sum'
ids[0].set_summary(sum_test)
ids = self.ical.get_events()
self.assertEqual(sum_test, ids[0].get_summary())
self.assertEqual(sum_last, ids[-1].get_summary())
def test_location(self):
ids = self.ical.get_events()
loc_first = 'Letisko, Slovakia'
loc_last = 'Gasometer, Austria'
self.assertEqual(loc_first, ids[0].get_location())
self.assertEqual(loc_last, ids[-1].get_location())
def test_events_dates(self):
before = self.ical.events_before(
datetime(2010, 7, 10, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(before))
before = self.ical.events_before(
datetime(2010, 12, 7, 0, 0, 0, 0, UTC()))
self.assertEqual(32, len(before))
before = self.ical.events_before(
datetime(2010, 10, 3, 0, 0, 0, 0, UTC()))
self.assertEqual(12, len(before))
between = self.ical.events_between(
datetime(2010, 8, 20, 0, 0, 0, 0, UTC()),
datetime(2010, 8, 24, 0, 0, 0, 0, UTC()))
self.assertEqual(0, len(between))
between = self.ical.events_between(
datetime(2010, 8, 12, 0, 0, 0, 0, UTC()),
datetime(2010, 8, 24, 0, 0, 0, 0, UTC()))
self.assertEqual(2, len(between))
after = self.ical.events_after(datetime(2010, 7, 10, 0, 0, 0, 0,
UTC()))
self.assertEqual(32, len(after))
after = self.ical.events_after(datetime(2010, 8, 20, 0, 0, 0, 0,
UTC()))
self.assertEqual(26, len(after))
after = self.ical.events_after(datetime(2010, 12, 7, 0, 0, 0, 0,
UTC()))
self.assertEqual(0, len(after))
after = self.ical3.events_after(datetime(2011, 3, 3, 0, 0, 0, 0,
UTC()))
self.assertEqual(0, len(after))
def test_url(self):
ids = self.ical.get_events()
url = ids[0].get_url()
self.assertEqual(
url, "http://www.last.fm/festival/1416224+Grape+Festival+2010")
url = ids[-1].get_url()
self.assertEqual(
url,
"http://www.last.fm/event/1328092+Blind+Guardian+at+Gasometer+on+16+October+2010"
)
def test_attendees(self):
ids = self.ical2.get_events()
at = ids[0].get_attendees()
self.assertEquals(2, len(at))
self.assertEquals(u"Milgrim", at[0].name)
self.assertEquals("mailto:[email protected]", at[0].address)
self.assertEquals("REQ-PARTICIPANT", at[0].role)
self.assertEquals("TRUE", at[0].rsvp_request)
self.assertEquals("NEEDS-ACTION", at[0].rsvp_status)
self.assertEquals(
str(at[0]),
"ATTENDEE;RSVP=TRUE;ROLE=REQ-PARTICIPANT;CN=Milgrim;PARTSTAT=NEEDS-ACTION:m\r\n ailto:[email protected]\r\n"
)
self.assertEquals("Idoru", at[1].name)
self.assertEquals("mailto:[email protected]", at[1].address)
self.assertEquals("REQ-PARTICIPANT", at[1].role)
self.assertEquals("TRUE", at[1].rsvp_request)
self.assertEquals("DECLINED", at[1].rsvp_status)
self.assertEquals(
str(at[1]),
"ATTENDEE;RSVP=TRUE;ROLE=REQ-PARTICIPANT;CN=Idoru;PARTSTAT=DECLINED:mailto:\r\n [email protected]\r\n"
)
