#!/usr/bin/python

 # 

 # The MIT License

 # 

 # Copyright (c) 2010 Ali Rantakari - http://hasseg.org

 # 

 # Permission is hereby granted, free of charge, to any person obtaining a copy

 # of this software and associated documentation files (the "Software"), to deal

 # in the Software without restriction, including without limitation the rights

 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

 # copies of the Software, and to permit persons to whom the Software is

 # furnished to do so, subject to the following conditions:

 # 

 # The above copyright notice and this permission notice shall be included in

 # all copies or substantial portions of the Software.

 # 

 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

 # THE SOFTWARE.

 import sys

 import datetime

 from CalStoreHelper import *

 from CalendarStore import *

 c = CalStoreHelper()

 # List of calendars to query (use None for all calendars)

 #calendar_names = ['Home','Work']

 calendar_names = None

 # Start and end dates to get events between

 start_date = datetime.datetime.now()

 delta = datetime.timedelta(days=10)

 end_date = start_date + delta

 # Let's create Cocoa date formatters for the NSDate objects

 # we get as task/event property values. We'll use the medium

 # system formats defined by the user in System Preferences.

 df_datetime = NSDateFormatter.new()

 df_datetime.setDateStyle_(NSDateFormatterMediumStyle)

 df_datetime.setTimeStyle_(NSDateFormatterMediumStyle)

 df_onlydate = NSDateFormatter.new()

 df_onlydate.setDateStyle_(NSDateFormatterMediumStyle)

 df_onlydate.setTimeStyle_(NSDateFormatterNoStyle)

 def printHeader(h):

     print '\n%s\n-----------------------------' % h

 def example_getEvents():

     # Example: get events

     printHeader('Events from %s to %s' % (start_date, end_date))

     events = c.getEvents(calendar_names, start_date, end_date)

     for event in events:

         print event.calendar().title(), ':', event.title()

 def example_getEventsByCalendar():

     # Example: get events separated by calendar

     printHeader('Events by Calendar from %s to %s' % (start_date, end_date))

     by_cals_list = c.getEvents(calendar_names, start_date, end_date,

                                by_calendar=True)

     for i in by_cals_list:

         cal = i['calendar']

         events = i['items']

         print cal.title()

         for event in events:

             print '  - ' + event.title()

 def example_getUncompletedTasks():

     # Example: get uncompleted tasks

     printHeader('Uncompleted Tasks')

     tasks = c.getUncompletedTasks(calendar_names)

     for task in tasks:

         print task.calendar().title(), ':', task.title()

 def example_getUncompletedTasksDueBefore():

     # Example: get uncompleted tasks due before a certain date

     db_date = datetime.date.today() + datetime.timedelta(days=10)

     printHeader('Uncompleted Tasks due before ' + str(db_date))

     tasks = c.getUncompletedTasks(calendar_names, due_before_date=db_date)

     for task in tasks:

         print task.calendar().title(), ':', task.title(),\

               '(due: %s)' % df_onlydate.stringFromDate_(task.dueDate())

 def example_getUncompletedTasksByCalendar():

     # Example: get uncompleted tasks separated by calendar

     printHeader('Uncompleted Tasks by Calendar')

     by_cals_list = c.getUncompletedTasks(calendar_names, by_calendar=True)

     for i in by_cals_list:

         cal = i['calendar']

         tasks = i['items']

         print cal.title()

         for task in tasks:

             print '  - ' + task.title()

 def example_XML():

     # Example: get events separated by calendar + XML formatting/markup

     printHeader('Events by Calendar from %s to %s\nWith XML Formatting/Markup'

                 % (start_date, end_date))

     by_cals_list = c.getEvents(calendar_names, start_date, end_date,

                                by_calendar=True)

     import cgi

     def wrapInXMLTag(tag, s):

         return '<%s>%s</%s>' % (tag, cgi.escape(s), tag)

     for i in by_cals_list:

         cal = i['calendar']

         events = i['items']

         print '<calendar>'

         print '  ' + wrapInXMLTag('title', cal.title())

         print '  <events>'

         for event in events:

             print '    <event>'

             print '      ' + wrapInXMLTag('title', event.title())

             print '      ' + wrapInXMLTag('startdate',

                                 df_datetime.stringFromDate_(event.startDate()))

             print '      ' + wrapInXMLTag('enddate',

                                 df_datetime.stringFromDate_(event.endDate()))

             print '    </event>'

         print '  </events>'

         print '</calendar>'

 def example_TeX():

     # Example: get uncompleted tasks separated by calendar + (La)TeX

     #          formatting/markup

     printHeader('Uncompleted Tasks by Calendar \n'

                 'With (La)TeX Formatting/Markup')

     by_cals_list = c.getUncompletedTasks(calendar_names, by_calendar=True)

     def wrapInTexTag(tag, s):

         return '\\%s{ %s }' % (tag, s)

     for i in by_cals_list:

         cal = i['calendar']

         tasks = i['items']

         print '\\subsubsection*{%s}' % cal.title()

         print '\\begin{compactitem}'

         for task in tasks:

             if task.priority() == CalPriorityHigh:

                 print '  \item ' + wrapInTexTag('highpriority', task.title())

             elif task.priority() == CalPriorityMedium:

                 print '  \item ' + wrapInTexTag('mediumpriority', task.title())

             elif task.priority() == CalPriorityLow:

                 print '  \item ' + wrapInTexTag('lowpriority', task.title())

             else:

                 print '  \item ' + task.title()

             if task.dueDate() != None:

                 print '  due: ' + df_onlydate.stringFromDate_(task.dueDate())

         print '\\end{compactitem}'

 def example_CSV():

     # Example: get events + CSV formatting

     printHeader('Events from %s to %s\nWith CSV Formatting'

                 % (start_date, end_date))

     events = c.getEvents(calendar_names, start_date, end_date)

     from UnicodeCSV import UnicodeWriter

     csv_writer = UnicodeWriter(open('/dev/stdout', 'wb'))

     for event in events:

         csv_writer.writerow([

             event.calendar().title(),

             event.title(),

             event.location(),

             event.startDate().descriptionWithCalendarFormat_timeZone_locale_(

                                 None, None, None),

             event.endDate().descriptionWithCalendarFormat_timeZone_locale_(

                                 None, None, None),

             event.notes()

             ])

 examples = [

     {'name':'Get Events',

      'function': example_getEvents},

     {'name':'Get Events by Calendar',

      'function': example_getEventsByCalendar},

     {'name':'Get Uncompleted Tasks',

      'function': example_getUncompletedTasks},

     {'name':'Get Uncompleted Tasks Due Before',

      'function': example_getUncompletedTasksDueBefore},

     {'name':'Get Uncompleted Tasks by Calendar',

      'function': example_getUncompletedTasksByCalendar},

     {'name':'XML Formatting',

      'function': example_XML},

     {'name':'(La)TeX Formatting',

      'function': example_TeX},

     {'name':'CSV Formatting',

      'function': example_CSV}

     ]

 while 1 == 1:

     print '## Input number of example to run, or Q to quit:'

     print

     for i in range(0, len(examples)):

         print '[%i]: %s' % (i, examples[i]['name'])

     print

     while 1 == 1:

         r = raw_input(':')

         if r.lower() == 'q':

             sys.exit(0)

         try:

             s = int(r)

         except:

             continue

         if s < 0 or len(examples) <= s:

             continue

         break

     examples[s]['function']()

     print