def __init__(self):
cream.Module.__init__(self, 'org.cream.Chronos')
self.events = {}
self.calendars = ordereddict()
self.colors = find_colors(.57, .72, .79)
self.calendar = cream.ipc.get_object('org.cream.PIM', '/org/cream/PIM/Calendar')
self.calendar.search_for_calendars()
self.calendar.connect_to_signal('calendar_added', self.add_calendar)
self.calendar.connect_to_signal('event_added', lambda u,e: self.add_events([e]))
self.calendar.connect_to_signal('event_removed', lambda u,e: self.remove_events([e]))
self.calendar.connect_to_signal('event_updated', lambda u,e: self.update_events([e]))
self.calendar_ui = CalendarUI()
self.calendar_ui.window.connect('delete_event', lambda *x: self.quit())
self.calendar_ui.connect('calendar-state-changed', self.calendar_state_change_cb)
for calendar in self.calendar.get_calendars():
self.add_calendar(calendar['uid'], calendar)
def add_events():
events = self.calendar.query({})
self.add_events(events)
gobject.timeout_add(1, add_events)
class Chronos(cream.Module):
def __init__(self):
cream.Module.__init__(self, 'org.cream.Chronos')
self.events = {}
self.calendars = ordereddict()
self.colors = find_colors(.57, .72, .79)
self.calendar = cream.ipc.get_object('org.cream.PIM', '/org/cream/PIM/Calendar')
self.calendar.search_for_calendars()
self.calendar.connect_to_signal('calendar_added', self.add_calendar)
self.calendar.connect_to_signal('event_added', lambda u,e: self.add_events([e]))
self.calendar.connect_to_signal('event_removed', lambda u,e: self.remove_events([e]))
self.calendar.connect_to_signal('event_updated', lambda u,e: self.update_events([e]))
self.calendar_ui = CalendarUI()
self.calendar_ui.window.connect('delete_event', lambda *x: self.quit())
self.calendar_ui.connect('calendar-state-changed', self.calendar_state_change_cb)
for calendar in self.calendar.get_calendars():
self.add_calendar(calendar['uid'], calendar)
def add_events():
events = self.calendar.query({})
self.add_events(events)
gobject.timeout_add(1, add_events)
def add_events(self, events):
added_events = []
for event in events:
color = self.calendars[event['calendar_uid']]['color']
event = Event(color=color, **event)
self.events[event.uid] = event
added_events.append(event)
self.calendar_ui.add_events(added_events)
def remove_events(self, events):
removed_events = []
for event in events:
removed_events.append(self.events.pop(event.uid))
self.calendar_ui.remove_events(removed_events)
def update_events(self, events):
updated_events = []
for event in events:
event = Event(**event)
self.events[event.uid] = event
updated_events.append(event)
self.calendar_ui.update_events(updated_events)
def add_calendar(self, uid, calendar):
color = self.colors.next()
calendar.update({'color': color, 'active': True})
self.calendars[calendar['uid']] = calendar
new = ordereddict()
for key in sorted(self.calendars, key=lambda k: self.calendars[k]['name']):
new[key] = self.calendars[key]
self.calendars = new
self.calendar_ui.set_calendars(self.calendars.values())
def calendar_state_change_cb(self, ui, uid, state):
self.calendars[uid]['active'] = state
updated_events = []
for event in self.events.itervalues():
if event.calendar_uid == uid:
event.active = state
updated_events.append(event)
self.calendar_ui.update_events(updated_events)