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
  18         self.Bind(wx.calendar.EVT_CALENDAR, self.OnCalSelected, id=cal.GetId())
  19
  20         b = wx.Button(self, -1, "Destroy the Calendar", pos = (250, 50))
  21         self.Bind(wx.EVT_BUTTON, self.OnButton, id= b.GetId())
  22         self.cal = cal
  23
  24         # Set up control to display a set of holidays:
  25         self.Bind(wx.calendar.EVT_CALENDAR_MONTH, self.OnChangeMonth, cal)
  26         self.holidays = [(1,1), (10,31), (12,25) ]    # (these don't move around)
  27         self.OnChangeMonth()
  28
  29     def OnButton(self, evt):
  30         self.cal.Destroy()
  31         self.cal = None
  32
  33     def OnCalSelected(self, evt):
  34         self.log.write('OnCalSelected: %s\n' % evt.GetDate())
  35
  36     def OnChangeMonth(self, evt=None):
  37         cur_month = self.cal.GetDate().GetMonth() + 1   # convert wxDateTime 0-11 => 1-12
  38
  39         for month, day in self.holidays:
  40             if month == cur_month:
  41                 self.cal.SetHoliday(day)
  42
  43 #----------------------------------------------------------------------
  44
  45 def runTest(frame, nb, log):
  46     win = TestPanel(nb, -1, log)
  47     return win
  48
  49 #----------------------------------------------------------------------
  50
  51
  52 overview = """\
  53 <html><body>
  54 <h2>CalendarCtrl</h2>
  55
  56 Yet <i>another</i> calendar control.  This one is a wrapper around the C++
  57 version described in the docs.  This one will probably be a bit more efficient
  58 than the one in wxPython.lib.calendar, but I like a few things about it better,
  59 so I think both will stay in wxPython.
  60 """
  61
  62
  63 if __name__ == '__main__':
  64     import sys,os
  65     import run
  66     run.main(['', os.path.basename(sys.argv[0])])
  67