wxPython/demo/wxCalendar.py

   1 #----------------------------------------------------------------------------
   2 # Name:         wxCalendar.py
   3 # Purpose:      Calendar control display testing on panel for wxPython demo
   4 #
   5 # Author:       Lorne White (email: lwhite1@planet.eon.net)
   6 #
   7 # Version       0.9
   8 # Date:         Feb 26, 2001
   9 # Licence:      wxWindows license
  10 #----------------------------------------------------------------------------
  11
  12 from wxPython.wx           import *
  13 from wxPython.lib.calendar import wxCalendar, Month, PrtCalDraw, CalenDlg
  14
  15 import os
  16 dir_path = os.getcwd()
  17
  18
  19 # highlighted days in month
  20
  21 test_days ={ 0: [],
  22             1: [3, 7, 9, 21],
  23             2: [2, 10, 4, 9],
  24             3: [4, 20, 29],
  25             4: [1, 12, 22],
  26             5: [2, 10, 15],
  27             6: [4, 8, 17],
  28             7: [6, 7, 8],
  29             8: [5, 10, 20],
  30             9: [1, 2, 5, 29],
  31            10: [2, 4, 6, 22],
  32            11: [6, 9, 12, 28, 29],
  33            12: [8, 9, 10, 11, 20] }
  34
  35 # test of full window calendar control functions
  36
  37 def GetMonthList():
  38     monthlist = []
  39     for i in range(13):
  40         name = Month[i]
  41         if name != None:
  42             monthlist.append(name)
  43     return monthlist
  44
  45 class TestPanel(wxPanel):
  46     def __init__(self, parent, log, frame):
  47         wxPanel.__init__(self, parent, -1)
  48
  49         self.log = log
  50         self.frame = frame
  51
  52         self.calend = wxCalendar(self, -1, wxPoint(100, 50), wxSize(200, 180))
  53
  54 #        start_month = 2        # preselect the date for calendar
  55 #        start_year = 2001
  56
  57         start_month = self.calend.GetMonth()        # get the current month & year
  58         start_year = self.calend.GetYear()
  59
  60 # month list from DateTime module
  61
  62         monthlist = GetMonthList()
  63
  64         mID = NewId()
  65         self.date = wxComboBox(self, mID, Month[start_month], wxPoint(100, 20), wxSize(90, -1), monthlist, wxCB_DROPDOWN)
  66         EVT_COMBOBOX(self, mID, self.EvtComboBox)
  67
  68 # set start month and year
  69
  70         self.calend.SetMonth(start_month)
  71         self.calend.SetYear(start_year)
  72
  73 # set attributes of calendar
  74
  75         self.calend.hide_title = TRUE
  76         self.calend.HideGrid()
  77         self.calend.SetWeekColor('WHITE', 'BLACK')
  78
  79 # display routine
  80
  81         self.ResetDisplay()
  82
  83 # mouse click event
  84
  85         self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
  86
  87 # scroll bar for month selection
  88
  89         mID = NewId()
  90         self.scroll = wxScrollBar(self, mID, wxPoint(100, 240), wxSize(200, 20), wxSB_HORIZONTAL)
  91         self.scroll.SetScrollbar(start_month-1, 1, 12, 1, TRUE)
  92         EVT_COMMAND_SCROLL(self, mID, self.Scroll)
  93
  94 # spin control for year selection
  95
  96         self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(200, 20), wxSize(60, -1))
  97         h = self.dtext.GetSize().height
  98
  99         mID = NewId()
 100         self.spin = wxSpinButton(self, mID, wxPoint(270, 20), wxSize(h*2, h))
 101         self.spin.SetRange(1980, 2010)
 102         self.spin.SetValue(start_year)
 103         EVT_SPIN(self, mID, self.OnSpin)
 104
 105 # button for calendar dialog test
 106
 107         wxStaticText(self, -1, "Test Calendar Dialog", wxPoint(350, 50), wxSize(150, -1))
 108
 109         mID = NewId()
 110         bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
 111         self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 80))#, wxSize(30, 30))
 112         EVT_BUTTON(self, mID, self.TestDlg)
 113
 114 # button for calendar window test
 115
 116         wxStaticText(self, -1, "Test Calendar Window", wxPoint(350, 150), wxSize(150, -1))
 117
 118         mID = NewId()
 119         bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
 120         self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 180))#, wxSize(30, 30))
 121         EVT_BUTTON(self, mID, self.TestFrame)
 122
 123         wxStaticText(self, -1, "Test Calendar Print", wxPoint(350, 250), wxSize(150, -1))
 124
 125         mID = NewId()
 126         bmp = wxBitmap('bitmaps/Calend.bmp', wxBITMAP_TYPE_BMP)
 127         self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 280))#, wxSize(30, 30))
 128         EVT_BUTTON(self, mID, self.OnPreview)
 129
 130 # calendar dialog
 131
 132     def TestDlg(self, event):       # test the date dialog
 133         dlg = CalenDlg(self)
 134         dlg.Centre()
 135         if dlg.ShowModal() == wxID_OK:
 136             result = dlg.result
 137             day = result[1]
 138             month = result[2]
 139             year = result[3]
 140             new_date = str(month) + '/'+ str(day) + '/'+ str(year)
 141             self.log.WriteText('Date Selected: %s\n' % new_date)
 142         else:
 143             self.log.WriteText('No Date Selected')
 144
 145 # calendar window test
 146
 147     def TestFrame(self, event):
 148         frame = CalendFrame(self, -1, "Test Calendar", self.log)
 149         frame.Show(true)
 150         return true
 151
 152 # calendar print preview
 153
 154     def OnPreview(self, event):
 155         month = self.calend.GetMonth()
 156         year = self.calend.GetYear()
 157
 158         prt = PrintCalend(self.frame, month, year)
 159         prt.Preview()
 160
 161 # month and year control events
 162
 163     def OnSpin(self, event):
 164         year = event.GetPosition()
 165         self.dtext.SetValue(str(year))
 166         self.calend.SetYear(year)
 167         self.calend.Refresh()
 168
 169     def EvtComboBox(self, event):
 170         name = event.GetString()
 171         self.log.WriteText('EvtComboBox: %s\n' % name)
 172         monthval = self.date.FindString(name)
 173         self.scroll.SetScrollbar(monthval, 1, 12, 1, TRUE)
 174
 175         self.calend.SetMonth(monthval+1)
 176         self.ResetDisplay()
 177
 178     def Scroll(self, event):
 179         value = self.scroll.GetThumbPosition()
 180         monthval = int(value)+1
 181         self.calend.SetMonth(monthval)
 182         self.ResetDisplay()
 183         self.log.WriteText('Month: %s\n' % value)
 184
 185         name = Month[monthval]
 186         self.date.SetValue(name)
 187
 188 # log mouse events
 189
 190     def MouseClick(self, evt):
 191         text = '%s CLICK   %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year)  # format date
 192         self.log.WriteText('Date Selected: ' + text + '\n')
 193
 194
 195 # set the highlighted days for the calendar
 196
 197     def ResetDisplay(self):
 198         month = self.calend.GetMonth()
 199         try:
 200             set_days = test_days[month]
 201         except:
 202             set_days = [1, 5, 12]
 203
 204         self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')
 205         self.calend.SetSelDay(set_days)
 206         self.calend.Refresh()
 207
 208 # increment and decrement toolbar controls
 209
 210     def OnIncYear(self, event):
 211         self.calend.IncYear()
 212         self.ResetDisplay()
 213
 214     def OnDecYear(self, event):
 215         self.calend.DecYear()
 216         self.ResetDisplay()
 217
 218     def OnIncMonth(self, event):
 219         self.calend.IncMonth()
 220         self.ResetDisplay()
 221
 222     def OnDecMonth(self, event):
 223         self.calend.DecMonth()
 224         self.ResetDisplay()
 225
 226     def OnCurrent(self, event):
 227         self.calend.SetCurrentDay()
 228         self.ResetDisplay()
 229
 230 # test of full window calendar control functions
 231
 232 class CalendFrame(wxFrame):
 233     def __init__(self, parent, id, title, log):
 234         wxFrame.__init__(self, parent, id, title, wxPyDefaultPosition, wxSize(400, 400))
 235         EVT_CLOSE(self, self.OnCloseWindow)
 236
 237         self.log = log
 238         self.CreateStatusBar()
 239         self.mainmenu = wxMenuBar()
 240         menu = wxMenu()
 241
 242         menu = self.MakeFileMenu()
 243         self.mainmenu.Append(menu, '&File')
 244
 245         self.MakeToolMenu()             # toolbar
 246
 247         self.SetMenuBar(self.mainmenu)
 248         self.calend = wxCalendar(self, -1)
 249         self.calend.SetCurrentDay()
 250         self.calend.grid_color = 'BLUE'
 251         self.calend.SetBusType()
 252 #        self.calend.ShowWeekEnd()
 253
 254         self.ResetDisplay()
 255
 256         self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
 257
 258     def MouseClick(self, evt):
 259         text = '%s CLICK   %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year)  # format date
 260         self.log.WriteText('Date Selected: ' + text + '\n')
 261
 262     def OnCloseWindow(self, event):
 263         self.Destroy()
 264
 265     def ResetDisplay(self):
 266         month = self.calend.GetMonth()
 267         try:
 268             set_days = test_days[month]
 269         except:
 270             set_days = [1, 5, 12]
 271
 272         self.calend.AddSelect([2, 16], 'GREEN', 'WHITE')
 273
 274         self.calend.SetSelDay(set_days)
 275         self.calend.Refresh()
 276
 277     def OnIncYear(self, event):
 278         self.calend.IncYear()
 279         self.ResetDisplay()
 280
 281     def OnDecYear(self, event):
 282         self.calend.DecYear()
 283         self.ResetDisplay()
 284
 285     def OnIncMonth(self, event):
 286         self.calend.IncMonth()
 287         self.ResetDisplay()
 288
 289     def OnDecMonth(self, event):
 290         self.calend.DecMonth()
 291         self.ResetDisplay()
 292
 293     def OnCurrent(self, event):
 294         self.calend.SetCurrentDay()
 295         self.ResetDisplay()
 296
 297     def MakeFileMenu(self):
 298         menu = wxMenu()
 299
 300         mID = NewId()
 301         menu.Append(mID, 'Decrement', 'Next')
 302         EVT_MENU(self, mID, self.OnDecMonth)
 303
 304         mID = NewId()
 305         menu.Append(mID, 'Increment', 'Dec')
 306         EVT_MENU(self, mID, self.OnIncMonth)
 307
 308         menu.AppendSeparator()
 309
 310         mID = NewId()
 311         menu.Append(mID, 'E&xit', 'Exit')
 312         EVT_MENU(self, mID, self.OnCloseWindow)
 313
 314         return menu
 315
 316     def MakeToolMenu(self):
 317         tb = self.CreateToolBar(wxTB_HORIZONTAL|wxNO_BORDER)
 318
 319         bmp_path = 'bitmaps/'
 320
 321         mID = NewId()
 322         SetToolPath(self, tb, mID, bmp_path + 'DbDec.bmp', 'Dec Year')
 323         EVT_TOOL(self, mID, self.OnDecYear)
 324
 325         mID = NewId()
 326         SetToolPath(self, tb, mID, bmp_path + 'Dec.bmp', 'Dec Month')
 327         EVT_TOOL(self, mID, self.OnDecMonth)
 328
 329         mID = NewId()
 330         SetToolPath(self, tb, mID, bmp_path + 'Pt.bmp', 'Current Month')
 331         EVT_TOOL(self, mID, self.OnCurrent)
 332
 333         mID = NewId()
 334         SetToolPath(self, tb, mID, bmp_path + 'Inc.bmp', 'Inc Month')
 335         EVT_TOOL(self, mID, self.OnIncMonth)
 336
 337         mID = NewId()
 338         SetToolPath(self, tb, mID, bmp_path + 'DbInc.bmp', 'Inc Year')
 339         EVT_TOOL(self, mID, self.OnIncYear)
 340
 341         tb.Realize()
 342
 343 #---------------------------------------------------------------------------
 344
 345 # example class for printing/previewing calendars
 346
 347 class PrintCalend:
 348     def __init__(self, parent, month, year):
 349         self.frame = parent
 350         self.month = month
 351         self.year = year
 352
 353         self.SetParms()
 354         self.SetCal()
 355         self.printData = wxPrintData()
 356
 357     def SetCal(self):
 358         self.grid_color = 'BLUE'
 359         self.back_color = 'WHITE'
 360         self.sel_color = 'RED'
 361         self.high_color = 'LIGHT BLUE'
 362         self.font = wxSWISS
 363         self.bold = wxNORMAL
 364
 365         self.sel_key = None      #  last used by
 366         self.sel_lst = []        # highlighted selected days
 367
 368         self.size = None
 369         self.hide_title = FALSE
 370         self.hide_grid = FALSE
 371         self.set_day = None
 372
 373     def SetParms(self):
 374         self.ymax = 1
 375         self.xmax = 1
 376         self.page = 1
 377         self.total_pg = 1
 378
 379         self.preview = None
 380         self.scale = 1.0
 381
 382         self.pagew = 8.5
 383         self.pageh = 11.0
 384
 385         self.txt_marg = 0.1
 386         self.lf_marg = 0
 387         self.top_marg = 0
 388
 389         self.page = 0
 390
 391     def SetDates(self, month, year):
 392         self.month = month
 393         self.year = year
 394
 395     def SetStyleDef(self, desc):
 396         self.style = desc
 397
 398     def SetCopies(self, copies):        # number of copies of label
 399         self.copies = copies
 400
 401     def SetStart(self, start):          # start position of label
 402         self.start = start
 403
 404     def Preview(self):
 405         printout = SetPrintout(self)
 406         printout2 = SetPrintout(self)
 407         self.preview = wxPrintPreview(printout, printout2, self.printData)
 408         if not self.preview.Ok():
 409             wxMessageBox("There was a problem printing!", "Printing", wxOK)
 410             return
 411
 412         self.preview.SetZoom(60)        # initial zoom value
 413
 414         frame = wxPreviewFrame(self.preview, self.frame, "Print preview")
 415
 416         frame.Initialize()
 417         frame.SetPosition(self.frame.GetPosition())
 418         frame.SetSize(self.frame.GetSize())
 419         frame.Show(true)
 420
 421     def Print(self):
 422         pdd = wxPrintDialogData()
 423         pdd.SetPrintData(self.printData)
 424         printer = wxPrinter(pdd)
 425         printout = SetPrintout(self)
 426         frame = wxFrame(NULL, -1, "Test")
 427         if not printer.Print(frame, printout):
 428             wxMessageBox("There was a problem printing.\nPerhaps your current printer is not set correctly?", "Printing", wxOK)
 429         else:
 430             self.printData = printer.GetPrintDialogData().GetPrintData()
 431         printout.Destroy()
 432
 433     def DoDrawing(self, DC):
 434         size = DC.GetSizeTuple()
 435         DC.BeginDrawing()
 436
 437         cal = PrtCalDraw(self)
 438
 439         if self.preview is None:
 440             cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
 441             cal.SetPreview(FALSE)
 442
 443         else:
 444             if self.preview == 1:
 445                 cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
 446             else:
 447                 cal.SetPSize(self.pwidth, self.pheight)
 448
 449             cal.SetPreview(self.preview)
 450
 451         cal.hide_title = self.hide_title        # set the calendar parameters
 452         cal.hide_grid = self.hide_grid
 453
 454         cal.grid_color = self.grid_color
 455         cal.high_color = self.high_color
 456         cal.back_color = self.back_color
 457         cal.outer_border = FALSE
 458         cal.font = self.font
 459         cal.bold = self.bold
 460
 461         cal_size = wxSize(3.0, 3.0)
 462         cal.SetSize(cal_size)
 463
 464         year, month = self.year, self.month
 465
 466         x = 1.0
 467         for i in range(2):
 468             y = 0.5
 469             for j in range(3):
 470                 cal.SetCal(year, month)       # current month
 471                 cal.SetPos(x, y)
 472
 473                 try:
 474                     set_days = test_days[month]
 475                 except:
 476                     set_days = [1, 5, 12]
 477
 478                 cal.AddSelect([2, 16], 'GREEN', 'WHITE')
 479
 480                 cal.DrawCal(DC, set_days)
 481
 482                 year, month = self.IncMonth(year, month)
 483                 y = y + 3.5
 484             x = x + 4.0     # next colum
 485
 486         DC.EndDrawing()
 487
 488         self.ymax = DC.MaxY()
 489         self.xmax = DC.MaxX()
 490
 491     def IncMonth(self, year, month):     # next month
 492         month = month + 1
 493         if month > 12:
 494             month = 1
 495             year = year + 1
 496
 497         return year, month
 498
 499     def GetTotalPages(self):
 500         self.pg_cnt = 1
 501         return self.pg_cnt
 502
 503     def SetPage(self, page):
 504         self.page = page
 505
 506     def SetPageSize(self, width, height):
 507         self.pwidth, self.pheight = width, height
 508
 509     def SetTotalSize(self, width, height):
 510         self.ptwidth, self.ptheight = width, height
 511
 512     def SetPreview(self, preview, scale):
 513         self.preview = preview
 514         self.scale = scale
 515
 516     def SetTotalSize(self, width, height):
 517         self.ptwidth = width
 518         self.ptheight = height
 519
 520 def SetToolPath(self, tb, id, bmp, title):
 521     global dir_path
 522     tb.AddSimpleTool(id, wxBitmap(os.path.join(dir_path, bmp), wxBITMAP_TYPE_BMP),
 523                      title, title)
 524
 525 class SetPrintout(wxPrintout):
 526     def __init__(self, canvas):
 527         wxPrintout.__init__(self)
 528         self.canvas = canvas
 529         self.end_pg = 1
 530
 531     def OnBeginDocument(self, start, end):
 532         return self.base_OnBeginDocument(start, end)
 533
 534     def OnEndDocument(self):
 535         self.base_OnEndDocument()
 536
 537     def HasPage(self, page):
 538         if page <= self.end_pg:
 539             return true
 540         else:
 541             return false
 542
 543     def GetPageInfo(self):
 544         self.end_pg = self.canvas.GetTotalPages()
 545         str_pg = 1
 546         try:
 547             end_pg = self.end_pg
 548         except:
 549             end_pg = 1
 550         return (str_pg, end_pg, str_pg, end_pg)
 551
 552     def OnPreparePrinting(self):
 553         self.base_OnPreparePrinting()
 554
 555     def OnBeginPrinting(self):
 556         dc = self.GetDC()
 557
 558         self.preview = self.IsPreview()
 559         if (self.preview):
 560             self.pixelsPerInch = self.GetPPIScreen()
 561         else:
 562             self.pixelsPerInch = self.GetPPIPrinter()
 563
 564         (w, h) = dc.GetSizeTuple()
 565         scaleX = float(w) / 1000
 566         scaleY = float(h) / 1000
 567         self.printUserScale = min(scaleX, scaleY)
 568
 569         self.base_OnBeginPrinting()
 570
 571     def GetSize(self):
 572         self.psizew, self.psizeh = self.GetPPIPrinter()
 573         return self.psizew, self.psizeh
 574
 575     def GetTotalSize(self):
 576         self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
 577         return self.ptsizew, self.ptsizeh
 578
 579     def OnPrintPage(self, page):
 580         dc = self.GetDC()
 581         (w, h) = dc.GetSizeTuple()
 582         scaleX = float(w) / 1000
 583         scaleY = float(h) / 1000
 584         self.printUserScale = min(scaleX, scaleY)
 585         dc.SetUserScale(self.printUserScale, self.printUserScale)
 586
 587         self.preview = self.IsPreview()
 588
 589         self.canvas.SetPreview(self.preview, self.printUserScale)
 590         self.canvas.SetPage(page)
 591
 592         self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
 593         self.canvas.SetTotalSize(self.ptsizew, self.ptsizeh)
 594
 595         self.psizew, self.psizeh = self.GetPPIPrinter()
 596         self.canvas.SetPageSize(self.psizew, self.psizeh)
 597
 598         self.canvas.DoDrawing(dc)
 599         return true
 600
 601 class MyApp(wxApp):
 602     def OnInit(self):
 603         frame = CalendFrame(NULL, -1, "Test Calendar")
 604         frame.Show(true)
 605         self.SetTopWindow(frame)
 606         return true
 607
 608 #---------------------------------------------------------------------------
 609
 610 def MessageDlg(self, message, type = 'Message'):
 611     dlg = wxMessageDialog(self, message, type, wxOK | wxICON_INFORMATION)
 612     dlg.ShowModal()
 613     dlg.Destroy()
 614
 615 #---------------------------------------------------------------------------
 616
 617 def main():
 618     app = MyApp(0)
 619     app.MainLoop()
 620
 621
 622 if __name__ == '__main__':
 623     main()
 624
 625
 626 #---------------------------------------------------------------------------
 627
 628 def runTest(frame, nb, log):
 629     win = TestPanel(nb, log, frame)
 630     return win
 631
 632 #---------------------------------------------------------------------------
 633
 634
 635 overview = """\
 636 This control provides a calendar control class for displaying and selecting dates.  In addition, the class is extended and can now be used for printing/previewing.
 637
 638 Additional features include weekend highlighting and business type Monday-Sunday format.
 639
 640 See example for various methods used to set display month, year, and highlighted dates (different font and background colours).
 641
 642 by Lorne White
 643
 644 """