"""Search the org files for the calendar events.

Scans the passed directories for the .org files and saves the events

found there into a multilevel dict of lists: events[year][month][day]

The returned dict uses defaultdict so *do not* rely on the

KeyError exception etc.! Check if the element exists with

.get(key) before accessing it!

"""

def year_dict():

return defaultdict(month_dict)

def month_dict():

return defaultdict(day_dict)

def day_dict():

return defaultdict(event_list)

def event_list():

return list()

events = year_dict()

for org_directory in org_directories:

for filename in glob.iglob(os.path.join(org_directory, ORG_GLOB)):

if filename.endswith(ORG_ARCHIVE_SUFFIX):

continue

with open(filename, "r") as filehandle:

last_heading = None

for line in filehandle:

heading_match = re.search(r'^\*+\s+(.*)', line)

if heading_match:

last_heading = heading_match.group(1)

# strip the tags

last_heading = re.sub(r'\s*\S*$', last_heading, '')

match = re.search(r'<(\d{4})-(\d{2})-(\d{2}).*?>', line)

if match:

year, month, day = [ int(field) for field in match.group(1,2,3) ]

month -= 1 # months are indexed from 0 in Gtk.Calendar

events[year][month][day].append(last_heading)

def __init__(self, org_directories):

self.window = Gtk.Window()

self.window.set_wmclass("traycalendar", "TrayCalendar")

self.window.set_resizable(False)

self.window.set_decorated(False)

self.window.set_gravity(Gdk.Gravity.STATIC)

window_width = 300

# Set the window geometry.

geometry = Gdk.Geometry()

geometry.min_width = window_width

geometry.max_width = window_width

geometry.base_width = window_width

self.window.set_geometry_hints(

None, geometry,

Gdk.WindowHints.MIN_SIZE |

Gdk.WindowHints.MAX_SIZE |

Gdk.WindowHints.BASE_SIZE)

# Create the listview for the calendar events.

list_model = Gtk.ListStore(str)

list_view = Gtk.TreeView(list_model)

list_column = Gtk.TreeViewColumn("Events", Gtk.CellRendererText(), text=0)

list_column.set_fixed_width(window_width)

list_view.append_column(list_column)

# Create the calendar widget.

calendar = Gtk.Calendar()

self.calendar_events = scan_org_for_events(org_directories)

calendar.connect('month-changed', self.mark_calendar_events)

calendar.connect('day-selected', self.display_event_list, list_model)

self.mark_calendar_events(calendar)

self.display_event_list(calendar, list_model)

close_button = Gtk.Button("Close")

close_button.connect('clicked', lambda event: self.window.destroy())

vbox = Gtk.VBox()

vbox.add(close_button)

vbox.add(calendar)

vbox.add(list_view)

self.window.add(vbox)

rootwin = self.window.get_screen().get_root_window()

# get_pointer is deprecated but using Gdk.Device.get_position

# is not viable here: we have no access to the pointing device.

screen, x, y, mask = rootwin.get_pointer()

x -= window_width

# Show the window right beside the cursor.

self.window.move(x,y)

self.window.show_all()

def mark_calendar_events(self, calendar):

"""Update the days with calendar events list for the selected month."""

year, month, day = calendar.get_date()

calendar.freeze_notify()

calendar.clear_marks()

for day in self.calendar_events[year][month]:

calendar.mark_day(day)

calendar.thaw_notify()

def display_event_list(self, calendar, event_list):

"""Update the calendar event list for the selected day."""

year, month, day = calendar.get_date()

event_list.clear()

# get(day) used instead of [day] because we use defaultdict

# and it would create a new element.

events = self.calendar_events[year][month].get(day)

if events:

for event in events:

def on_left_click(event):

window = CalendarWindow(org_directories)

def on_right_click(button, time, data):

Gtk.main_quit()

statusicon = Gtk.StatusIcon()

statusicon.set_from_icon_name('x-office-calendar')

statusicon.connect('activate', on_left_click)

statusicon.connect('popup-menu', on_right_click)

import argparse

parser = argparse.ArgumentParser()

parser.add_argument(

"--no-tray",

help="Show the calendar windows immediately and quit after it's closed.",

action='store_true',

)

parser.add_argument(

"--org-directory", "-d",

help="Directories to search for *.org; default: ~/org/.",

action='append',

dest='org_directories',

)

args = parser.parse_args()

if not args.org_directories:

args.org_directories = [DEFAULT_ORG_DIRECTORY]

if args.no_tray:

window_mode(args.org_directories)

else: