wxPython/wx/lib/calendar.py

   1 #----------------------------------------------------------------------------
   2 # Name:         calendar.py
   3 # Purpose:      Calendar display control
   4 #
   5 # Author:       Lorne White (email: lorne.white@telusplanet.net)
   6 #
   7 # Created:
   8 # Version       0.92
   9 # Date:         Nov 26, 2001
  10 # Licence:      wxWindows license
  11 #----------------------------------------------------------------------------
  12 # 12/01/2003 - Jeff Grimmett (grimmtooth@softhome.net)
  13 #
  14 # o Updated for wx namespace
  15 # o Tested with updated demo
  16 # o Added new event type EVT_CALENDAR. The reason for this is that the original
  17 #   library used a hardcoded ID of 2100 for generating events. This makes it
  18 #   very difficult to fathom when trying to decode the code since there's no
  19 #   published API. Creating the new event binder might seem like overkill -
  20 #   after all, you might ask, why not just use a new event ID and be done with
  21 #   it? However, a consistent interface is very useful at times; also it makes
  22 #   it clear that we're not just hunting for mouse clicks -- we're hunting
  23 #   wabbit^H^H^H^H (sorry bout that) for calender-driven mouse clicks. So
  24 #   that's my sad story. Shoot me if you must :-)
  25 # o There's still one deprecation warning buried in here somewhere, but I
  26 #   haven't been able to find it yet. It only occurs when displaying a
  27 #   print preview, and only the first time. It *could* be an error in the
  28 #   demo, I suppose.
  29 #
  30 #   Here's the traceback:
  31 #
  32 #   C:\Python\lib\site-packages\wx\core.py:949: DeprecationWarning:
  33 #       integer argument expected, got float
  34 #   newobj = _core.new_Rect(*args, **kwargs)
  35 #
  36 # 12/17/2003 - Jeff Grimmett (grimmtooth@softhome.net)
  37 #
  38 # o A few style-guide nips and tucks
  39 # o Renamed wxCalendar to Calendar
  40 # o Couple of bugfixes
  41 #
  42
  43 import  wx
  44
  45 from CDate import *
  46
  47 CalDays = [6, 0, 1, 2, 3, 4, 5]
  48 AbrWeekday = {6:"Sun", 0:"Mon", 1:"Tue", 2:"Wed", 3:"Thu", 4:"Fri", 5:"Sat"}
  49 _MIDSIZE = 180
  50
  51 BusCalDays = [0, 1, 2, 3, 4, 5, 6]
  52
  53 # Calendar click event - added 12/1/03 by jmg (see above)
  54 wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED = wx.NewEventType()
  55 EVT_CALENDAR = wx.PyEventBinder(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, 1)
  56
  57 def GetMonthList():
  58     monthlist = []
  59     for i in range(13):
  60         name = Month[i]
  61         if name != None:
  62             monthlist.append(name)
  63     return monthlist
  64
  65 # calendar drawing routing
  66
  67 class CalDraw:
  68     def __init__(self, parent):
  69         self.pwidth = 1
  70         self.pheight = 1
  71         try:
  72             self.scale = parent.scale
  73         except:
  74             self.scale = 1
  75
  76         self.DefParms()
  77
  78     def DefParms(self):
  79         self.num_auto = True    # auto scale of the cal number day size
  80         self.num_size = 12      # default size of calendar if no auto size
  81         self.max_num_size = 12  # maximum size for calendar number
  82
  83         self.num_align_horz = wx.ALIGN_CENTRE    # alignment of numbers
  84         self.num_align_vert = wx.ALIGN_CENTRE
  85         self.num_indent_horz = 0     # points indent from position, used to offset if not centered
  86         self.num_indent_vert = 0
  87
  88         self.week_auto = True       # auto scale of week font text
  89         self.week_size = 10
  90         self.max_week_size = 12
  91
  92         self.grid_color = 'BLACK'       # grid and selection colors
  93         self.back_color = 'WHITE'
  94         self.sel_color = 'RED'
  95
  96         self.high_color = 'LIGHT BLUE'
  97         self.border_color = 'BLACK'
  98         self.week_color = 'LIGHT GREY'
  99
 100         self.week_font_color = 'BLACK'      # font colors
 101         self.day_font_color = 'BLACK'
 102
 103         self.font = wx.SWISS
 104         self.bold = wx.NORMAL
 105
 106         self.hide_title = False
 107         self.hide_grid = False
 108         self.outer_border = True
 109
 110         self.title_offset = 0
 111         self.cal_week_scale = 0.7
 112         self.show_weekend = False
 113         self.cal_type = "NORMAL"
 114
 115     def SetWeekColor(self, font_color, week_color):     # set font and background color for week title
 116         self.week_font_color = font_color
 117         self.week_color = week_color
 118
 119     def SetSize(self, size):
 120         self.set_sizew = size[0]
 121         self.set_sizeh = size[1]
 122
 123     def InitValues(self):       # default dimensions of various elements of the calendar
 124         self.rg = {}
 125         self.cal_sel = {}
 126         self.set_cy_st = 0      # start position
 127         self.set_cx_st = 0
 128
 129         self.set_y_mrg = 15      # start of vertical draw default
 130         self.set_x_mrg = 10
 131         self.set_y_end = 10
 132
 133     def SetPos(self, xpos, ypos):
 134         self.set_cx_st = xpos
 135         self.set_cy_st = ypos
 136
 137     def SetMarg(self, xmarg, ymarg):
 138         self.set_x_st = xmarg
 139         self.set_y_st = ymarg
 140         self.set_y_end = ymarg
 141
 142     def InitScale(self):        # scale position values
 143         self.sizew = self.set_sizew * self.pwidth
 144         self.sizeh = self.set_sizeh * self.pheight
 145
 146         self.cx_st = self.set_cx_st * self.pwidth       # draw start position
 147         self.cy_st = self.set_cy_st * self.pheight
 148
 149         self.x_mrg = self.set_x_mrg * self.pwidth         # calendar draw margins
 150         self.y_mrg = self.set_y_mrg * self.pheight
 151         self.y_end = self.set_y_end * self.pheight
 152
 153     def DrawCal(self, DC, sel_lst=[]):
 154         self.DC = DC
 155         self.InitScale()
 156
 157         self.DrawBorder()
 158
 159         if self.hide_title is False:
 160             self.DrawMonth()
 161
 162         self.Center()
 163
 164         self.DrawGrid()
 165         self.GetRect()
 166
 167         if self.show_weekend is True:       # highlight weekend dates
 168             self.SetWeekEnd()
 169
 170         self.AddSelect(sel_lst)     # overrides the weekend highlight
 171
 172         self.DrawSel()      # highlighted days
 173         self.DrawWeek()
 174         self.DrawNum()
 175
 176     def AddSelect(self, list, cfont=None, cbackgrd = None):
 177         if cfont is None:
 178             cfont = self.sel_color      # font digit color
 179
 180         if cbackgrd is None:
 181             cbackgrd = self.high_color     # select background color
 182
 183         for val in list:
 184             self.cal_sel[val] = (cfont, cbackgrd)
 185
 186     # draw border around the outside of the main display rectangle
 187     def DrawBorder(self):
 188         brush = wx.Brush(wx.NamedColour(self.back_color), wx.SOLID)
 189         self.DC.SetBrush(brush)
 190         self.DC.SetPen(wx.Pen(wx.NamedColour(self.border_color), 1))
 191
 192         if self.outer_border is True:
 193             # full display window area
 194             rect = wx.Rect(self.cx_st, self.cy_st, self.sizew, self.sizeh)
 195             self.DC.DrawRectangleRect(rect)
 196
 197     def DrawNumVal(self):
 198         self.DrawNum()
 199
 200     # calculate the calendar days and offset position
 201
 202     def SetCal(self, year, month):
 203         self.InitValues()       # reset initial values
 204
 205         self.year = year
 206         self.month = month
 207
 208         day = 1
 209         t = Date(year, month, day)
 210         dow = self.dow = t.day_of_week     # start day in month
 211         dim = self.dim = t.days_in_month   # number of days in month
 212
 213         if self.cal_type == "NORMAL":
 214             start_pos = dow+1
 215         else:
 216             start_pos = dow
 217
 218         self.st_pos = start_pos
 219
 220         self.cal = []
 221         for i in range(start_pos):
 222             self.cal.append('')
 223
 224         i = 1
 225         while i <= dim:
 226             self.cal.append(str(i))
 227             i = i + 1
 228
 229         return start_pos
 230
 231     def SetWeekEnd(self, font_color='BLACK', backgrd = 'LIGHT GREY'):
 232         date = 6 - int(self.dow)     # start day of first saturday
 233
 234         while date <= self.dim:
 235             self.cal_sel[date] = (font_color, backgrd)  # Saturday
 236             date = date + 1
 237
 238             if date <= self.dim:
 239                 self.cal_sel[date] = (font_color, backgrd)      # Sunday
 240                 date = date + 6
 241             else:
 242                 date = date + 7
 243
 244     # get the display rectange list of the day grid
 245     def GetRect(self):
 246         cnt = 0
 247         for y in self.gridy[1:-1]:
 248             for x in self.gridx[:-1]:
 249                 rect = wx.Rect(x, y, self.dl_w, self.dl_h)  # create rect region
 250                 self.rg[cnt] = rect
 251                 cnt = cnt + 1
 252
 253         return self.rg
 254
 255     def GetCal(self):
 256         return self.cal
 257
 258     def GetOffset(self):
 259         return self.st_pos
 260
 261     # month and year title
 262     def DrawMonth(self):
 263         month = Month[self.month]
 264
 265         sizef = 11
 266         if self.sizeh < _MIDSIZE:
 267             sizef = 10
 268
 269         f = wx.Font(sizef, self.font, wx.NORMAL, self.bold)
 270         self.DC.SetFont(f)
 271
 272         tw,th = self.DC.GetTextExtent(month)
 273         adjust = self.cx_st + (self.sizew-tw)/2
 274         self.DC.DrawText(month, (adjust, self.cy_st + th))
 275
 276         year = str(self.year)
 277         tw,th = self.DC.GetTextExtent(year)
 278         adjust =  self.sizew - tw - self.x_mrg
 279
 280         self.title_offset = th * 2
 281
 282         f = wx.Font(sizef, self.font, wx.NORMAL, self.bold)
 283         self.DC.SetFont(f)
 284         self.DC.DrawText(year, (self.cx_st + adjust, self.cy_st + th))
 285
 286     def DrawWeek(self):     # draw the week days
 287         width = self.gridx[1]-self.gridx[0]
 288         height = self.gridy[1] - self.gridy[0]
 289         rect_w = self.gridx[7]-self.gridx[0]
 290
 291         f = wx.Font(10, self.font, wx.NORMAL, self.bold)      # initial font setting
 292
 293         if self.week_auto == True:
 294             test_size = self.max_week_size      # max size
 295             test_day = ' Sun '
 296             while test_size > 2:
 297                 f.SetPointSize(test_size)
 298                 self.DC.SetFont(f)
 299                 tw,th = self.DC.GetTextExtent(test_day)
 300
 301                 if tw < width and th < height:
 302                     break
 303
 304                 test_size = test_size - 1
 305         else:
 306             f.SetPointSize(self.week_size)   # set fixed size
 307             self.DC.SetFont(f)
 308
 309         self.DC.SetTextForeground(wx.NamedColour(self.week_font_color))
 310
 311         cnt_x = 0
 312         cnt_y = 0
 313
 314         brush = wx.Brush(wx.NamedColour(self.week_color), wx.SOLID)
 315         self.DC.SetBrush(brush)
 316         self.DC.DrawRectangle((self.gridx[0], self.gridy[0]), (rect_w+1, height))
 317
 318         if self.cal_type == "NORMAL":
 319             cal_days = CalDays
 320         else:
 321             cal_days = BusCalDays
 322
 323         for val in cal_days:
 324             day = AbrWeekday[val]
 325
 326             if self.sizew < 200:
 327                 day = day[0]
 328
 329             dw,dh = self.DC.GetTextExtent(day)
 330             diffx = (width-dw)/2
 331             diffy = (height-dh)/2
 332
 333             x = self.gridx[cnt_x]
 334             y = self.gridy[cnt_y]
 335             self.DC.DrawRectangle((self.gridx[cnt_x], self.gridy[0]), (width+1, height))
 336             self.DC.DrawText(day, (x+diffx, y+diffy))
 337             cnt_x = cnt_x + 1
 338
 339     # draw the day numbers
 340     def DrawNum(self):
 341         f = wx.Font(10, self.font, wx.NORMAL, self.bold)      # initial font setting
 342
 343         if self.num_auto == True:
 344             test_size = self.max_num_size      # max size
 345             test_day = ' 99 '
 346
 347             while test_size > 2:
 348                 f.SetPointSize(test_size)
 349                 self.DC.SetFont(f)
 350                 tw,th = self.DC.GetTextExtent(test_day)
 351
 352                 if tw < self.dl_w and th < self.dl_h:
 353                     sizef = test_size
 354                     break
 355                 test_size = test_size - 1
 356         else:
 357             f.SetPointSize(self.num_size)   # set fixed size
 358             self.DC.SetFont(f)
 359
 360         cnt_x = 0
 361         cnt_y = 1
 362         for val in self.cal:
 363             x = self.gridx[cnt_x]
 364             y = self.gridy[cnt_y]
 365
 366             try:
 367                 num_val = int(val)
 368                 num_color = self.cal_sel[num_val][0]
 369             except:
 370                 num_color = self.day_font_color
 371
 372             self.DC.SetTextForeground(wx.NamedColour(num_color))
 373             self.DC.SetFont(f)
 374
 375             tw,th = self.DC.GetTextExtent(val)
 376
 377             if self.num_align_horz == wx.ALIGN_CENTRE:
 378                 adj_h = (self.dl_w - tw)/2
 379             elif self.num_align_horz == wx.ALIGN_RIGHT:
 380                 adj_h = self.dl_w - tw
 381             else:
 382                 adj_h = 0   # left alignment
 383
 384             adj_h = adj_h + self.num_indent_horz
 385
 386             if self.num_align_vert == wx.ALIGN_CENTRE:
 387                 adj_v = (self.dl_h - th)/2
 388             elif self.num_align_horz == wx.ALIGN_RIGHT:
 389                 adj_v = self.dl_h - th
 390             else:
 391                 adj_v = 0   # left alignment
 392
 393             adj_v = adj_v + self.num_indent_vert
 394
 395             self.DC.DrawText(val, (x+adj_h, y+adj_v))
 396
 397             if cnt_x < 6:
 398                 cnt_x = cnt_x + 1
 399             else:
 400                 cnt_x = 0
 401                 cnt_y = cnt_y + 1
 402
 403     # calculate the dimensions in the center of the drawing area
 404     def Center(self):
 405         bdw = self.x_mrg * 2
 406         bdh = self.y_mrg + self.y_end + self.title_offset
 407
 408         self.dl_w = int((self.sizew-bdw)/7)
 409         self.dl_h = int((self.sizeh-bdh)/7)
 410
 411         # week title adjustment
 412         self.dl_th = int(self.dl_h*self.cal_week_scale)
 413         self.cwidth = self.dl_w * 7
 414         self.cheight = self.dl_h * 6 + self.dl_th
 415
 416     # highlighted selected days
 417     def DrawSel(self):
 418         for key in self.cal_sel.keys():
 419             sel_color = self.cal_sel[key][1]
 420             brush = wx.Brush(wx.NamedColour(sel_color), wx.SOLID)
 421             self.DC.SetBrush(brush)
 422
 423             if self.hide_grid is False:
 424                 self.DC.SetPen(wx.Pen(wx.NamedColour(self.grid_color), 0))
 425             else:
 426                 self.DC.SetPen(wx.Pen(wx.NamedColour(self.back_color), 0))
 427
 428             nkey = key + self.st_pos -1
 429             rect = self.rg[nkey]
 430             self.DC.DrawRectangle((rect.x, rect.y), (rect.width+1, rect.height+1))
 431
 432     # calculate and draw the grid lines
 433     def DrawGrid(self):
 434         self.DC.SetPen(wx.Pen(wx.NamedColour(self.grid_color), 0))
 435
 436         self.gridx = []
 437         self.gridy = []
 438
 439         self.x_st = self.cx_st + self.x_mrg
 440         # start postion of draw
 441         self.y_st = self.cy_st + self.y_mrg + self.title_offset
 442
 443         x1 = self.x_st
 444         y1 = self.y_st
 445         y2 = y1 + self.cheight
 446
 447         for i in range(8):
 448             if self.hide_grid is False:
 449                 self.DC.DrawLine((x1, y1), (x1, y2))
 450             self.gridx.append(x1)
 451             x1 = x1 + self.dl_w
 452
 453         x1 = self.x_st
 454         y1 = self.y_st
 455         x2 = x1 + self.cwidth
 456
 457         for i in range(8):
 458             if self.hide_grid is False:
 459                 self.DC.DrawLine((x1, y1), (x2, y1))
 460
 461             self.gridy.append(y1)
 462
 463             if i == 0:
 464                 y1 = y1 + self.dl_th
 465             else:
 466                 y1 = y1 + self.dl_h
 467
 468
 469 class PrtCalDraw(CalDraw):
 470     def InitValues(self):
 471         self.rg = {}
 472         self.cal_sel = {}
 473         # start draw border location
 474         self.set_cx_st = 1.0
 475         self.set_cy_st = 1.0
 476
 477         # draw offset position
 478         self.set_y_mrg = 0.2
 479         self.set_x_mrg = 0.2
 480         self.set_y_end = 0.2
 481
 482     # calculate the dimensions in the center of the drawing area
 483     def SetPSize(self, pwidth, pheight):
 484         self.pwidth = int(pwidth)/self.scale
 485         self.pheight = int(pheight)/self.scale
 486
 487     def SetPreview(self, preview):
 488         self.preview = preview
 489
 490 class Calendar(wx.Window):
 491     def __init__(self, parent, id, pos=wx.DefaultPosition, size=wx.DefaultSize):
 492         wx.Window.__init__(self, parent, id, pos, size)
 493
 494         # set the calendar control attributes
 495
 496         self.grid_color = 'BLACK'
 497         self.back_color = 'WHITE'
 498         self.hide_grid = False
 499         self.sel_color = 'RED'
 500         self.hide_title = False
 501         self.show_weekend = False
 502         self.cal_type = "NORMAL"
 503
 504         # font colors
 505         self.week_color = 'LIGHT GREY'
 506         self.week_font_color = 'BLACK'
 507
 508         self.select_list = []
 509
 510         self.SetBackgroundColour(self.back_color)
 511         self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftEvent)
 512         self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDEvent)
 513         self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightEvent)
 514         self.Bind(wx.EVT_RIGHT_DCLICK, self.OnRightDEvent)
 515
 516         self.sel_key = None      #  last used by
 517         self.sel_lst = []        # highlighted selected days
 518
 519         # default calendar for current month
 520         self.SetNow()
 521
 522         self.size = None
 523         self.set_day = None
 524
 525         self.Bind(wx.EVT_PAINT, self.OnPaint)
 526         self.Bind(wx.EVT_SIZE, self.OnSize)
 527
 528     # control some of the main calendar attributes
 529
 530     def HideTitle(self):
 531         self.hide_title = True
 532
 533     def HideGrid(self):
 534         self.hide_grid = True
 535
 536     # determine the calendar rectangle click area and draw a selection
 537
 538     def ProcessClick(self, event):
 539         self.x, self.y = event.GetX(), event.GetY()
 540         key = self.GetDayHit(self.x, self.y)
 541         self.SelectDay(key)
 542
 543     # tab mouse click events and process
 544
 545     def OnLeftEvent(self, event):
 546         self.click = 'LEFT'
 547         self.shiftkey = event.ShiftDown()
 548         self.ctrlkey = event.ControlDown()
 549         self.ProcessClick(event)
 550
 551     def OnLeftDEvent(self, event):
 552         self.click = 'DLEFT'
 553         self.ProcessClick(event)
 554
 555     def OnRightEvent(self, event):
 556         self.click = 'RIGHT'
 557         self.ProcessClick(event)
 558
 559     def OnRightDEvent(self, event):
 560         self.click = 'DRIGHT'
 561         self.ProcessClick(event)
 562
 563     def SetSize(self, set_size):
 564         self.size = set_size
 565
 566     def SetSelDay(self, sel):
 567         # list of highlighted days
 568         self.sel_lst = sel
 569
 570     # get the current date
 571     def SetNow(self):
 572         dt = now()
 573         self.month = dt.month
 574         self.year = dt.year
 575         self.day = dt.day
 576
 577     # set the current day
 578     def SetCurrentDay(self):
 579         self.SetNow()
 580         self.set_day = self.day
 581
 582     # get the date, day, month, year set in calendar
 583
 584     def GetDate(self):
 585         return self.day, self.month, self.year
 586
 587     def GetDay(self):
 588         return self.day
 589
 590     def GetMonth(self):
 591         return self.month
 592
 593     def GetYear(self):
 594         return self.year
 595
 596     # set the day, month, and year
 597
 598     def SetDayValue(self, day):
 599         self.set_day = day
 600
 601     def SetMonth(self, month):
 602         if month >= 1 and month <= 12:
 603             self.month = month
 604         else:
 605             self.month = 1
 606         self.set_day = None
 607
 608     def SetYear(self, year):
 609         self.year = year
 610
 611     # increment year and month
 612
 613     def IncYear(self):
 614         self.year = self.year + 1
 615         self.set_day = None
 616
 617     def DecYear(self):
 618         self.year = self.year - 1
 619         self.set_day = None
 620
 621     def IncMonth(self):
 622         self.month = self.month + 1
 623         if self.month > 12:
 624             self.month = 1
 625             self.year = self.year + 1
 626         self.set_day = None
 627
 628     def DecMonth(self):
 629         self.month = self.month - 1
 630         if self.month < 1:
 631             self.month = 12
 632             self.year = self.year - 1
 633         self.set_day = None
 634
 635     # test to see if the selection has a date and create event
 636
 637     def TestDay(self, key):
 638         try:
 639             self.day = int(self.cal[key])
 640         except:
 641             return None
 642         if self.day == "":
 643             return None
 644         else:
 645             # Changed 12/1/03 by jmg (see above) to support 2.5 event binding
 646             evt = wx.PyCommandEvent(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, self.GetId())
 647             evt.click, evt.day, evt.month, evt.year = self.click, self.day, self.month, self.year
 648             evt.shiftkey = self.shiftkey
 649             evt.ctrlkey = self.ctrlkey
 650             self.GetEventHandler().ProcessEvent(evt)
 651
 652             self.set_day = self.day
 653             return key
 654
 655     # find the clicked area rectangle
 656
 657     def GetDayHit(self, mx, my):
 658         for key in self.rg.keys():
 659             val = self.rg[key]
 660             ms_rect = wx.Rect(mx, my, 1, 1)
 661             if wx.IntersectRect(ms_rect, val) is not None:
 662                 result = self.TestDay(key)
 663                 return result
 664
 665         return None
 666
 667     # calendar drawing
 668
 669     def SetWeekColor(self, font_color, week_color):
 670         # set font and background color for week title
 671         self.week_font_color = font_color
 672         self.week_color = week_color
 673
 674     def AddSelect(self, list, font_color, back_color):
 675         list_val = [list, font_color, back_color]
 676         self.select_list.append(list_val)
 677
 678     def ShowWeekEnd(self):
 679         # highlight weekend
 680         self.show_weekend = True
 681
 682     def SetBusType(self):
 683         self.cal_type = "BUS"
 684
 685     def OnSize(self, evt):
 686         self.Refresh(False)
 687         evt.Skip()
 688
 689     def OnPaint(self, event):
 690         DC = wx.PaintDC(self)
 691         self.DoDrawing(DC)
 692
 693     def DoDrawing(self, DC):
 694         DC = wx.PaintDC(self)
 695         DC.BeginDrawing()
 696
 697         self.cal = cal = CalDraw(self)
 698
 699         cal.grid_color = self.grid_color
 700         cal.back_color = self.back_color
 701         cal.hide_grid = self.hide_grid
 702         cal.grid_color = self.grid_color
 703         cal.hide_title = self.hide_title
 704         cal.show_weekend = self.show_weekend
 705         cal.cal_type = self.cal_type
 706
 707         if self.size is None:
 708             size = self.GetClientSize()
 709         else:
 710             size = self.size
 711
 712         # drawing attributes
 713
 714         cal.week_font_color = self.week_font_color
 715         cal.week_color = self.week_color
 716
 717         cal.SetSize(size)
 718         cal.SetCal(self.year, self.month)
 719
 720         for val in self.select_list:
 721             cal.AddSelect(val[0], val[1], val[2])
 722
 723         cal.DrawCal(DC, self.sel_lst)
 724
 725         self.rg = cal.GetRect()
 726         self.cal = cal.GetCal()
 727         self.st_pos = cal.GetOffset()
 728         self.ymax = DC.MaxY()
 729
 730         if self.set_day != None:
 731             self.SetDay(self.set_day)
 732
 733         DC.EndDrawing()
 734
 735     # draw the selection rectangle
 736     def DrawRect(self, key, color = 'BLACK', width = 0):
 737         if key == None:
 738             return
 739
 740         DC = wx.ClientDC(self)
 741         DC.BeginDrawing()
 742
 743         brush = wx.Brush(wx.Colour(0, 0xFF, 0x80), wx.TRANSPARENT)
 744         DC.SetBrush(brush)
 745         DC.SetPen(wx.Pen(wx.NamedColour(color), width))
 746
 747         rect = self.rg[key]
 748         DC.DrawRectangle((rect.x, rect.y), (rect.width+1, rect.height+1))
 749
 750         DC.EndDrawing()
 751
 752     # set the day selection
 753
 754     def SetDay(self, day):
 755         day = day + self.st_pos - 1
 756         self.SelectDay(day)
 757
 758     def SelectDay(self, key):
 759         sel_size = 1
 760         # clear large selection
 761         self.DrawRect(self.sel_key, self.back_color, sel_size)
 762
 763         if self.hide_grid is False:
 764             self.DrawRect(self.sel_key, self.grid_color)
 765
 766         self.DrawRect(key, self.sel_color, sel_size)
 767         # store last used by
 768         self.sel_key = key
 769         self.select_day = None
 770
 771     def ClearDsp(self):
 772         self.Clear()
 773
 774 class CalenDlg(wx.Dialog):
 775     def __init__(self, parent, month=None, day = None, year=None):
 776         wx.Dialog.__init__(self, parent, -1, "Event Calendar", wx.DefaultPosition, (280, 360))
 777
 778         # set the calendar and attributes
 779         self.calend = Calendar(self, -1, (20, 60), (240, 200))
 780
 781         if month == None:
 782             self.calend.SetCurrentDay()
 783             start_month = self.calend.GetMonth()
 784             start_year = self.calend.GetYear()
 785         else:
 786             self.calend.month = start_month = month
 787             self.calend.year = start_year = year
 788             self.calend.SetDayValue(day)
 789
 790         self.calend.HideTitle()
 791         self.ResetDisplay()
 792
 793         # get month list from DateTime
 794         monthlist = GetMonthList()
 795
 796         # select the month
 797         self.date = wx.ComboBox(self, -1, Month[start_month], (20, 20), (90, -1),
 798                                 monthlist, wx.CB_DROPDOWN)
 799         self.Bind(wx.EVT_COMBOBOX, self.EvtComboBox, self.date)
 800
 801         # alternate spin button to control the month
 802         h = self.date.GetSize().height
 803         self.m_spin = wx.SpinButton(self, -1, (130, 20), (h*2, h), wx.SP_VERTICAL)
 804         self.m_spin.SetRange(1, 12)
 805         self.m_spin.SetValue(start_month)
 806         self.Bind(wx.EVT_SPIN, self.OnMonthSpin, self.m_spin)
 807
 808         # spin button to control the year
 809         self.dtext = wx.TextCtrl(self, -1, str(start_year), (160, 20), (60, -1))
 810         h = self.dtext.GetSize().height
 811
 812         self.y_spin = wx.SpinButton(self, -1, (220, 20), (h*2, h), wx.SP_VERTICAL)
 813         self.y_spin.SetRange(1980, 2010)
 814         self.y_spin.SetValue(start_year)
 815
 816         self.Bind(wx.EVT_SPIN, self.OnYrSpin, self.y_spin)
 817         self.Bind(EVT_CALENDAR, self.MouseClick, self.calend)
 818
 819         x_pos = 50
 820         y_pos = 280
 821         but_size = (60, 25)
 822
 823         btn = wx.Button(self, -1, ' Ok ', (x_pos, y_pos), but_size)
 824         self.Bind(wx.EVT_BUTTON, self.OnOk, btn)
 825
 826         btn = wx.Button(self, -1, ' Close ', (x_pos + 120, y_pos), but_size)
 827         self.Bind(wx.EVT_BUTTON, self.OnCancel, btn)
 828
 829     def OnOk(self, event):
 830         self.EndModal(wx.ID_OK)
 831
 832     def OnCancel(self, event):
 833         self.EndModal(wx.ID_CANCEL)
 834
 835     # log the mouse clicks
 836     def MouseClick(self, evt):
 837         self.month = evt.month
 838         # result click type and date
 839         self.result = [evt.click, str(evt.day), Month[evt.month], str(evt.year)]
 840
 841         if evt.click == 'DLEFT':
 842             self.EndModal(wx.ID_OK)
 843
 844     # month and year spin selection routines
 845     def OnMonthSpin(self, event):
 846         month = event.GetPosition()
 847         self.date.SetValue(Month[month])
 848         self.calend.SetMonth(month)
 849         self.calend.Refresh()
 850
 851     def OnYrSpin(self, event):
 852         year = event.GetPosition()
 853         self.dtext.SetValue(str(year))
 854         self.calend.SetYear(year)
 855         self.calend.Refresh()
 856
 857     def EvtComboBox(self, event):
 858         name = event.GetString()
 859         monthval = self.date.FindString(name)
 860         self.m_spin.SetValue(monthval+1)
 861
 862         self.calend.SetMonth(monthval+1)
 863         self.ResetDisplay()
 864
 865     # set the calendar for highlighted days
 866
 867     def ResetDisplay(self):
 868         month = self.calend.GetMonth()
 869         self.calend.Refresh()
 870
 871
 872