wxPython/demo/CalendarCtrl.py

   1
   2 import  wx
   3 import  wx.calendar
   4
   5 #----------------------------------------------------------------------
   6
   7 class TestPanel(wx.Panel):
   8     def __init__(self, parent, ID, log):
   9         wx.Panel.__init__(self, parent, ID)
  10         self.log = log
  11
  12         cal = wx.calendar.CalendarCtrl(self, -1, wx.DateTime_Now(), pos = (25,50),
  13                              style = wx.calendar.CAL_SHOW_HOLIDAYS
  14                              | wx.calendar.CAL_SUNDAY_FIRST
  15                              | wx.calendar.CAL_SEQUENTIAL_MONTH_SELECTION
  16                              )
  17         self.cal = cal
  18         self.Bind(wx.calendar.EVT_CALENDAR, self.OnCalSelected, id=cal.GetId())
  19
  20         # Set up control to display a set of holidays:
  21         self.Bind(wx.calendar.EVT_CALENDAR_MONTH, self.OnChangeMonth, cal)
  22         self.holidays = [(1,1), (10,31), (12,25) ]    # (these don't move around)
  23         self.OnChangeMonth()
  24
  25         cal2 = wx.calendar.CalendarCtrl(self, -1, wx.DateTime_Now(), pos = (325,50))
  26         self.Bind(wx.calendar.EVT_CALENDAR_SEL_CHANGED,
  27                   self.OnCalSelChanged, cal2)
  28
  29     def OnCalSelected(self, evt):
  30         self.log.write('OnCalSelected: %s\n' % evt.GetDate())
  31
  32     def OnChangeMonth(self, evt=None):
  33         cur_month = self.cal.GetDate().GetMonth() + 1   # convert wxDateTime 0-11 => 1-12
  34         for month, day in self.holidays:
  35             if month == cur_month:
  36                 self.cal.SetHoliday(day)
  37         if cur_month == 8:
  38             attr = wx.calendar.CalendarDateAttr(border=wx.calendar.CAL_BORDER_SQUARE,
  39                                                 colBorder="blue")
  40             self.cal.SetAttr(14, attr)
  41         else:
  42             self.cal.ResetAttr(14)
  43
  44     def OnCalSelChanged(self, evt):
  45         cal = evt.GetEventObject()
  46         self.log.write("OnCalSelChanged:\n\t%s: %s\n\t%s: %s\n\t%s: %s\n\t" %
  47                        ("EventObject", cal,
  48                         "Date       ", cal.GetDate(),
  49                         "Ticks      ", cal.GetDate().GetTicks(),
  50                         ))
  51
  52 #----------------------------------------------------------------------
  53
  54 def runTest(frame, nb, log):
  55     win = TestPanel(nb, -1, log)
  56     return win
  57
  58 #----------------------------------------------------------------------
  59
  60
  61 overview = """\
  62 <html><body>
  63 <h2>CalendarCtrl</h2>
  64
  65 Yet <i>another</i> calendar control.  This one is a wrapper around the C++
  66 version described in the docs.  This one will probably be a bit more efficient
  67 than the one in wxPython.lib.calendar, but I like a few things about it better,
  68 so I think both will stay in wxPython.
  69 """
  70
  71
  72 if __name__ == '__main__':
  73     import sys,os
  74     import run
  75     run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
  76