X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/1fded56b375bf7a4687af1cdb182899614c1b2a8..3d2d9036037d9ee51f47d878c3dcef4af4a6ef41:/wxPython/wx/lib/calendar.py?ds=sidebyside diff --git a/wxPython/wx/lib/calendar.py b/wxPython/wx/lib/calendar.py index 8b5a466e1c..4c8c3b4a48 100644 --- a/wxPython/wx/lib/calendar.py +++ b/wxPython/wx/lib/calendar.py @@ -1,11 +1,1208 @@ +#---------------------------------------------------------------------------- +# Name: calendar.py +# Purpose: Calendar display control +# +# Author: Lorne White (email: lorne.white@telusplanet.net) +# +# Created: +# Version 0.92 +# Date: Nov 26, 2001 +# Licence: wxWindows license +#---------------------------------------------------------------------------- +# 12/01/2003 - Jeff Grimmett (grimmtooth@softhome.net) +# +# o Updated for wx namespace +# o Tested with updated demo +# o Added new event type EVT_CALENDAR. The reason for this is that the original +# library used a hardcoded ID of 2100 for generating events. This makes it +# very difficult to fathom when trying to decode the code since there's no +# published API. Creating the new event binder might seem like overkill - +# after all, you might ask, why not just use a new event ID and be done with +# it? However, a consistent interface is very useful at times; also it makes +# it clear that we're not just hunting for mouse clicks -- we're hunting +# wabbit^H^H^H^H (sorry bout that) for calender-driven mouse clicks. So +# that's my sad story. Shoot me if you must :-) +# o There's still one deprecation warning buried in here somewhere, but I +# haven't been able to find it yet. It only occurs when displaying a +# print preview, and only the first time. It *could* be an error in the +# demo, I suppose. +# +# Here's the traceback: +# +# C:\Python\lib\site-packages\wx\core.py:949: DeprecationWarning: +# integer argument expected, got float +# newobj = _core.new_Rect(*args, **kwargs) +# +# 12/17/2003 - Jeff Grimmett (grimmtooth@softhome.net) +# +# o A few style-guide nips and tucks +# o Renamed wxCalendar to Calendar +# o Couple of bugfixes +# +# 06/02/2004 - Joerg "Adi" Sieker adi@sieker.info +# +# o Changed color handling, use dictionary instead of members. +# This causes all color changes to be ignored if they manipluate the members directly. +# SetWeekColor and other method color methods were adapted to use the new dictionary. +# o Added COLOR_* constants +# o Added SetColor method for Calendar class +# o Added 3D look of week header +# o Added colors for 3D look of header +# o Fixed width calculation. +# Because of rounding difference the total width and height of the +# calendar could be up to 6 pixels to small. The last column and row +# are now wider/taller by the missing amount. +# o Added SetTextAlign method to wxCalendar. This exposes logic +# which was already there. +# o Fixed CalDraw.SetMarg which set set_x_st and set_y_st which don't get used anywhere. +# Instead set set_x_mrg and set_y_mrg +# o Changed default X and Y Margin to 0. +# o Added wxCalendar.SetMargin. +# +# 17/03/2004 - Joerg "Adi" Sieker adi@sieker.info +# o Added keyboard navigation to the control. +# Use the cursor keys to navigate through the ages. :) +# The Home key function as go to today +# o select day is now a filled rect instead of just an outline +# +# 15/04/2005 - Joe "shmengie" Brown joebrown@podiatryfl.com +# o Adjusted spin control size/placement (On Windows ctrls were overlapping). +# o Set Ok/Cancel buttons to wx.ID_OK & wx.ID_CANCEL to provide default dialog +# behaviour. +# o If no date has been clicked clicked, OnOk set the result to calend's date, +# important if keyboard only navigation is used. + + +import wx + +from CDate import * + +CalDays = [6, 0, 1, 2, 3, 4, 5] +AbrWeekday = {6:"Sun", 0:"Mon", 1:"Tue", 2:"Wed", 3:"Thu", 4:"Fri", 5:"Sat"} +_MIDSIZE = 180 + +COLOR_GRID_LINES = "grid_lines" +COLOR_BACKGROUND = "background" +COLOR_SELECTION_FONT = "selection_font" +COLOR_SELECTION_BACKGROUND = "selection_background" +COLOR_BORDER = "border" +COLOR_HEADER_BACKGROUND = "header_background" +COLOR_HEADER_FONT = "header_font" +COLOR_WEEKEND_BACKGROUND = "weekend_background" +COLOR_WEEKEND_FONT = "weekend_font" +COLOR_FONT = "font" +COLOR_3D_LIGHT = "3d_light" +COLOR_3D_DARK = "3d_dark" +COLOR_HIGHLIGHT_FONT = "highlight_font" +COLOR_HIGHLIGHT_BACKGROUND = "highlight_background" + +BusCalDays = [0, 1, 2, 3, 4, 5, 6] + +# Calendar click event - added 12/1/03 by jmg (see above) +wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED = wx.NewEventType() +EVT_CALENDAR = wx.PyEventBinder(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, 1) + +def GetMonthList(): + monthlist = [] + for i in range(13): + name = Month[i] + if name != None: + monthlist.append(name) + return monthlist + +def MakeColor(in_color): + try: + color = wxNamedColour(in_color) + except: + color = in_color + return color + +def DefaultColors(): + colors = {} + colors[COLOR_GRID_LINES] = 'BLACK' + colors[COLOR_BACKGROUND] = 'WHITE' + colors[COLOR_SELECTION_FONT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOWTEXT) + colors[COLOR_SELECTION_BACKGROUND] =wx.Colour(255,255,225) + colors[COLOR_BORDER] = 'BLACK' + colors[COLOR_HEADER_BACKGROUND] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_3DFACE) + colors[COLOR_HEADER_FONT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOWTEXT) + colors[COLOR_WEEKEND_BACKGROUND] = 'LIGHT GREY' + colors[COLOR_WEEKEND_FONT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOWTEXT) + colors[COLOR_FONT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOWTEXT) + colors[COLOR_3D_LIGHT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_BTNHIGHLIGHT) + colors[COLOR_3D_DARK] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_BTNSHADOW) + colors[COLOR_HIGHLIGHT_FONT] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHTTEXT) + colors[COLOR_HIGHLIGHT_BACKGROUND] = wx.SystemSettings_GetColour(wx.SYS_COLOUR_HIGHLIGHT) + return colors +# calendar drawing routing + +class CalDraw: + def __init__(self, parent): + self.pwidth = 1 + self.pheight = 1 + try: + self.scale = parent.scale + except: + self.scale = 1 + + self.gridx = [] + self.gridy = [] + + self.DefParms() + + def DefParms(self): + self.num_auto = True # auto scale of the cal number day size + self.num_size = 12 # default size of calendar if no auto size + self.max_num_size = 12 # maximum size for calendar number + + self.num_align_horz = wx.ALIGN_CENTRE # alignment of numbers + self.num_align_vert = wx.ALIGN_CENTRE + self.num_indent_horz = 0 # points indent from position, used to offset if not centered + self.num_indent_vert = 0 + + self.week_auto = True # auto scale of week font text + self.week_size = 10 + self.max_week_size = 12 + + self.colors = DefaultColors() + + self.font = wx.SWISS + self.bold = wx.NORMAL + + self.hide_title = False + self.hide_grid = False + self.outer_border = True + + self.title_offset = 0 + self.cal_week_scale = 0.7 + self.show_weekend = False + self.cal_type = "NORMAL" + + def SetWeekColor(self, font_color, week_color): + # set font and background color for week title + self.colors[COLOR_HEADER_FONT] = MakeColor(font_color) + self.colors[COLOR_HEADER_BACKGROUND] = MakeColor(week_color) + self.colors[COLOR_3D_LIGHT] = MakeColor(week_color) + self.colors[COLOR_3D_DARK] = MakeColor(week_color) + + def SetSize(self, size): + self.set_sizew = size[0] + self.set_sizeh = size[1] + + def InitValues(self): # default dimensions of various elements of the calendar + self.rg = {} + self.cal_sel = {} + self.set_cy_st = 0 # start position + self.set_cx_st = 0 + + self.set_y_mrg = 1 # start of vertical draw default + self.set_x_mrg = 1 + self.set_y_end = 1 + def SetPos(self, xpos, ypos): + self.set_cx_st = xpos + self.set_cy_st = ypos + + def SetMarg(self, xmarg, ymarg): + self.set_x_mrg = xmarg + self.set_y_mrg = ymarg + self.set_y_end = ymarg + + def InitScale(self): # scale position values + self.sizew = int(self.set_sizew * self.pwidth) + self.sizeh = int(self.set_sizeh * self.pheight) + + self.cx_st = int(self.set_cx_st * self.pwidth) # draw start position + self.cy_st = int(self.set_cy_st * self.pheight) + + self.x_mrg = int(self.set_x_mrg * self.pwidth) # calendar draw margins + self.y_mrg = int(self.set_y_mrg * self.pheight) + self.y_end = int(self.set_y_end * self.pheight) + + def DrawCal(self, DC, sel_lst=[]): + self.InitScale() + + self.DrawBorder(DC) + + if self.hide_title is False: + self.DrawMonth(DC) + + self.Center() + + self.DrawGrid(DC) + self.GetRect() + if self.show_weekend is True: # highlight weekend dates + self.SetWeekEnd() + + self.AddSelect(sel_lst) # overrides the weekend highlight + + self.DrawSel(DC) # highlighted days + self.DrawWeek(DC) + self.DrawNum(DC) + + def AddSelect(self, list, cfont=None, cbackgrd = None): + if cfont is None: + cfont = self.colors[COLOR_SELECTION_FONT] # font digit color + if cbackgrd is None: + cbackgrd = self.colors[COLOR_SELECTION_BACKGROUND] # select background color + + for val in list: + self.cal_sel[val] = (cfont, cbackgrd) + + # draw border around the outside of the main display rectangle + def DrawBorder(self, DC, transparent = False): + if self.outer_border is True: + if transparent == False: + brush = wx.Brush(MakeColor(self.colors[COLOR_BACKGROUND]), wx.SOLID) + else: + brush = wx.TRANSPARENT_BRUSH + DC.SetBrush(brush) + DC.SetPen(wx.Pen(MakeColor(self.colors[COLOR_BORDER]))) + # full display window area + rect = wx.Rect(self.cx_st, self.cy_st, self.sizew, self.sizeh) + DC.DrawRectangleRect(rect) + + def DrawFocusIndicator(self, DC): + if self.outer_border is True: + DC.SetBrush(wx.TRANSPARENT_BRUSH) + DC.SetPen(wx.Pen(MakeColor(self.colors[COLOR_HIGHLIGHT_BACKGROUND]), style=wx.DOT)) + # full display window area + rect = wx.Rect(self.cx_st, self.cy_st, self.sizew, self.sizeh) + DC.DrawRectangleRect(rect) + + def DrawNumVal(self): + self.DrawNum() + + # calculate the calendar days and offset position + def SetCal(self, year, month): + self.InitValues() # reset initial values + + self.year = year + self.month = month + + day = 1 + t = Date(year, month, day) + dow = self.dow = t.day_of_week # start day in month + dim = self.dim = t.days_in_month # number of days in month + + if self.cal_type == "NORMAL": + start_pos = dow+1 + else: + start_pos = dow + + self.st_pos = start_pos + + self.cal_days = [] + for i in range(start_pos): + self.cal_days.append('') + + i = 1 + while i <= dim: + self.cal_days.append(str(i)) + i = i + 1 + + self.end_pos = start_pos + dim - 1 + + return start_pos + + def SetWeekEnd(self, font_color=None, backgrd = None): + if font_color != None: + self.SetColor(COLOR_WEEKEND_FONT, MakeColor(font_color)) + if backgrd != None: + self.SetColor(COLOR_WEEKEND_BACKGROUND, MakeColor(backgrd)) + + date = 6 - int(self.dow) # start day of first saturday + + while date <= self.dim: + self.cal_sel[date] = (self.GetColor(COLOR_WEEKEND_FONT), self.GetColor(COLOR_WEEKEND_BACKGROUND)) # Saturday + date = date + 1 + + if date <= self.dim: + self.cal_sel[date] = (self.GetColor(COLOR_WEEKEND_FONT), self.GetColor(COLOR_WEEKEND_BACKGROUND)) # Sunday + date = date + 6 + else: + date = date + 7 + + # get the display rectange list of the day grid + def GetRect(self): + cnt = 0 + h = 0 + w = 0 + for y in self.gridy[1:-1]: + if y == self.gridy[-2]: + h = h + self.restH + + for x in self.gridx[:-1]: + assert type(y) == int + assert type(x) == int + + w = self.cellW + h = self.cellH + + if x == self.gridx[-2]: + w = w + self.restW + + rect = wx.Rect(x, y, w+1, h+1) # create rect region + + self.rg[cnt] = rect + cnt = cnt + 1 + + return self.rg + + def GetCal(self): + return self.cal_days + + def GetOffset(self): + return self.st_pos + + # month and year title + def DrawMonth(self, DC): + month = Month[self.month] + + sizef = 11 + if self.sizeh < _MIDSIZE: + sizef = 10 + + f = wx.Font(sizef, self.font, wx.NORMAL, self.bold) + DC.SetFont(f) + + tw,th = DC.GetTextExtent(month) + adjust = self.cx_st + (self.sizew-tw)/2 + DC.DrawText(month, adjust, self.cy_st + th) + + year = str(self.year) + tw,th = DC.GetTextExtent(year) + adjust = self.sizew - tw - self.x_mrg + + self.title_offset = th * 2 + + f = wx.Font(sizef, self.font, wx.NORMAL, self.bold) + DC.SetFont(f) + DC.DrawText(year, self.cx_st + adjust, self.cy_st + th) + + def DrawWeek(self, DC): # draw the week days + # increase by 1 to include all gridlines + width = self.gridx[1] - self.gridx[0] + 1 + height = self.gridy[1] - self.gridy[0] + 1 + rect_w = self.gridx[-1] - self.gridx[0] + + f = wx.Font(10, self.font, wx.NORMAL, self.bold) # initial font setting + + if self.week_auto == True: + test_size = self.max_week_size # max size + test_day = ' Sun ' + while test_size > 2: + f.SetPointSize(test_size) + DC.SetFont(f) + tw,th = DC.GetTextExtent(test_day) + + if tw < width and th < height: + break + + test_size = test_size - 1 + else: + f.SetPointSize(self.week_size) # set fixed size + DC.SetFont(f) + + DC.SetTextForeground(MakeColor(self.colors[COLOR_HEADER_FONT])) + + cnt_x = 0 + cnt_y = 0 + + brush = wx.Brush(MakeColor(self.colors[COLOR_HEADER_BACKGROUND]), wx.SOLID) + DC.SetBrush(brush) + + if self.cal_type == "NORMAL": + cal_days = CalDays + else: + cal_days = BusCalDays + + for val in cal_days: + if val == cal_days[-1]: + width = width + self.restW + + day = AbrWeekday[val] + + if self.sizew < 200: + day = day[0] + + dw,dh = DC.GetTextExtent(day) + + diffx = (width-dw)/2 + diffy = (height-dh)/2 + + x = self.gridx[cnt_x] + y = self.gridy[cnt_y] + pointXY = (x, y) + pointWH = (width, height) + if self.hide_grid == False: + pen = wx.Pen(MakeColor(self.GetColor(COLOR_GRID_LINES)), 1, wx.SOLID) + else: + pen = wx.Pen(MakeColor(self.GetColor(COLOR_BACKGROUND)), 1, wx.SOLID) + DC.SetPen(pen) + DC.DrawRectanglePointSize( pointXY, pointWH) + + old_pen = DC.GetPen() + + pen = wx.Pen(MakeColor(self.colors[COLOR_3D_LIGHT]), 1, wx.SOLID) + DC.SetPen(pen) + # draw the horizontal hilight + startPoint = wx.Point(x + 1 , y + 1) + endPoint = wx.Point(x + width - 1, y + 1) + DC.DrawLinePoint(startPoint, endPoint ) + + # draw the vertical hilight + startPoint = wx.Point(x + 1 , y + 1) + endPoint = wx.Point(x + 1, y + height - 2) + DC.DrawLinePoint(startPoint, endPoint ) + + pen = wx.Pen(MakeColor(self.colors[COLOR_3D_DARK]), 1, wx.SOLID) + DC.SetPen(pen) + + # draw the horizontal lowlight + startPoint = wx.Point(x + 1, y + height - 2) + endPoint = wx.Point(x + width - 1, y + height - 2) + DC.DrawLinePoint(startPoint, endPoint ) + + # draw the vertical lowlight + startPoint = wx.Point(x + width - 2 , y + 2) + endPoint = wx.Point(x + width - 2, y + height - 2) + DC.DrawLinePoint(startPoint, endPoint ) + + pen = wx.Pen(MakeColor(self.colors[COLOR_FONT]), 1, wx.SOLID) + + DC.SetPen(pen) + + point = (x+diffx, y+diffy) + DC.DrawTextPoint(day, point) + cnt_x = cnt_x + 1 + + def _CalcFontSize(self, DC, f): + if self.num_auto == True: + test_size = self.max_num_size # max size + test_day = ' 99 ' + + while test_size > 2: + f.SetPointSize(test_size) + DC.SetFont(f) + tw,th = DC.GetTextExtent(test_day) + + if tw < self.cellW and th < self.cellH: + sizef = test_size + break + test_size = test_size - 1 + else: + f.SetPointSize(self.num_size) # set fixed size + DC.SetFont(f) + + # draw the day numbers + def DrawNum(self, DC): + f = wx.Font(10, self.font, wx.NORMAL, self.bold) # initial font setting + self._CalcFontSize(DC, f) + + cnt_x = 0 + cnt_y = 1 + for val in self.cal_days: + x = self.gridx[cnt_x] + y = self.gridy[cnt_y] + + self._DrawDayText(x, y, val, f, DC) + + if cnt_x < 6: + cnt_x = cnt_x + 1 + else: + cnt_x = 0 + cnt_y = cnt_y + 1 + + def _DrawDayText(self, x, y, text, font, DC): + + try: + num_val = int(text) + num_color = self.cal_sel[num_val][0] + except: + num_color = self.colors[COLOR_FONT] + + DC.SetTextForeground(MakeColor(num_color)) + DC.SetFont(font) + + tw,th = DC.GetTextExtent(text) + + if self.num_align_horz == wx.ALIGN_CENTRE: + adj_h = (self.cellW - tw)/2 + elif self.num_align_horz == wx.ALIGN_RIGHT: + adj_h = self.cellW - tw + else: + adj_h = 0 # left alignment + + adj_h = adj_h + self.num_indent_horz + + if self.num_align_vert == wx.ALIGN_CENTRE: + adj_v = (self.cellH - th)/2 + elif self.num_align_vert == wx.ALIGN_BOTTOM: + adj_v = self.cellH - th + else: + adj_v = 0 # left alignment + + adj_v = adj_v + self.num_indent_vert + + DC.DrawTextPoint(text, (x+adj_h, y+adj_v)) + + def DrawDayText(self, DC, key): + f = wx.Font(10, self.font, wx.NORMAL, self.bold) # initial font setting + self._CalcFontSize(DC, f) + + if key > self.end_pos: + key = self.end_pos + + val = self.cal_days[key] + cnt_x = key % 7 + cnt_y = int(key / 7)+1 + x = self.gridx[cnt_x] + y = self.gridy[cnt_y] + self._DrawDayText(x, y, val, f, DC) + + + # calculate the dimensions in the center of the drawing area + def Center(self): + borderW = self.x_mrg * 2 + borderH = self.y_mrg + self.y_end + self.title_offset + + self.cellW = int((self.sizew - borderW)/7) + self.cellH = int((self.sizeh - borderH)/7) + + self.restW = ((self.sizew - borderW)%7 ) - 1 + + # week title adjustment + self.weekHdrCellH = int(self.cellH * self.cal_week_scale) + # recalculate the cell height exkl. the week header and + # subtracting the size + self.cellH = int((self.sizeh - borderH - self.weekHdrCellH)/6) + + self.restH = ((self.sizeh - borderH - self.weekHdrCellH)%6 ) - 1 + self.calW = self.cellW * 7 + self.calH = self.cellH * 6 + self.weekHdrCellH + + # highlighted selected days + def DrawSel(self, DC): + + for key in self.cal_sel.keys(): + sel_color = self.cal_sel[key][1] + brush = wx.Brush(MakeColor(sel_color), wx.SOLID) + DC.SetBrush(brush) + + if self.hide_grid is False: + DC.SetPen(wx.Pen(MakeColor(self.colors[COLOR_GRID_LINES]), 0)) + else: + DC.SetPen(wx.Pen(MakeColor(self.colors[COLOR_BACKGROUND]), 0)) + + nkey = key + self.st_pos -1 + rect = self.rg[nkey] + + DC.DrawRectangleRect(rect) + + # calculate and draw the grid lines + def DrawGrid(self, DC): + DC.SetPen(wx.Pen(MakeColor(self.colors[COLOR_GRID_LINES]), 0)) + + self.gridx = [] + self.gridy = [] + + self.x_st = self.cx_st + self.x_mrg + # start postion of draw + self.y_st = self.cy_st + self.y_mrg + self.title_offset + + x1 = self.x_st + y1 = self.y_st + y2 = y1 + self.calH + self.restH + + for i in range(8): + if i == 7: + x1 = x1 + self.restW + + if self.hide_grid is False: + DC.DrawLinePoint((x1, y1), (x1, y2)) + + self.gridx.append(x1) + + x1 = x1 + self.cellW + + x1 = self.x_st + y1 = self.y_st + x2 = x1 + self.calW + self.restW + + for i in range(8): + if i == 7: + y1 = y1 + self.restH + + if self.hide_grid is False: + DC.DrawLinePoint((x1, y1), (x2, y1)) + + self.gridy.append(y1) + + if i == 0: + y1 = y1 + self.weekHdrCellH + else: + y1 = y1 + self.cellH + + def GetColor(self, name): + return MakeColor(self.colors[name]) + + def SetColor(self, name, value): + self.colors[name] = MakeColor(value) + +class PrtCalDraw(CalDraw): + def InitValues(self): + self.rg = {} + self.cal_sel = {} + # start draw border location + self.set_cx_st = 1.0 + self.set_cy_st = 1.0 + + # draw offset position + self.set_y_mrg = 0.2 + self.set_x_mrg = 0.2 + self.set_y_end = 0.2 + + # calculate the dimensions in the center of the drawing area + def SetPSize(self, pwidth, pheight): + self.pwidth = int(pwidth)/self.scale + self.pheight = int(pheight)/self.scale + + def SetPreview(self, preview): + self.preview = preview + +class Calendar( wx.PyControl ): + def __init__(self, parent, id, pos=wx.DefaultPosition, size=wx.Size(200,200), + style= 0, validator=wx.DefaultValidator, + name= "calendar"): + wx.PyControl.__init__(self, parent, id, pos, size, style | wx.WANTS_CHARS, validator, name) + + self.hasFocus = False + # set the calendar control attributes + + self.hide_grid = False + self.hide_title = False + self.show_weekend = False + self.cal_type = "NORMAL" + self.outer_border = True + self.num_align_horz = wx.ALIGN_CENTRE + self.num_align_vert = wx.ALIGN_CENTRE + self.colors = DefaultColors() + self.set_x_mrg = 1 + self.set_y_mrg = 1 + self.set_y_end = 1 + + self.select_list = [] + + self.SetBackgroundColour(MakeColor(self.colors[COLOR_BACKGROUND])) + self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftEvent) + self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDEvent) + self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightEvent) + self.Bind(wx.EVT_RIGHT_DCLICK, self.OnRightDEvent) + self.Bind(wx.EVT_SET_FOCUS, self.OnSetFocus) + self.Bind(wx.EVT_KILL_FOCUS, self.OnKillFocus) + self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown) + + self.sel_key = None # last used by + self.sel_lst = [] # highlighted selected days + + # default calendar for current month + self.SetNow() + + self.size = None + self.set_day = None + + self.Bind(wx.EVT_PAINT, self.OnPaint) + self.Bind(wx.EVT_SIZE, self.OnSize) + + def AcceptsFocus(self): + return self.IsShown() and self.IsEnabled() + + def GetColor(self, name): + return MakeColor(self.colors[name]) + + def SetColor(self, name, value): + self.colors[name] = MakeColor(value) + + # control some of the main calendar attributes + + def HideTitle(self): + self.hide_title = True + + def HideGrid(self): + self.hide_grid = True + + # determine the calendar rectangle click area and draw a selection + + def ProcessClick(self, event): + self.x, self.y = event.GetX(), event.GetY() + key = self.GetDayHit(self.x, self.y) + self.SelectDay(key) + + # tab mouse click events and process + + def OnLeftEvent(self, event): + self.click = 'LEFT' + self.shiftkey = event.ShiftDown() + self.ctrlkey = event.ControlDown() + self.ProcessClick(event) + + def OnLeftDEvent(self, event): + self.click = 'DLEFT' + self.ProcessClick(event) + + def OnRightEvent(self, event): + self.click = 'RIGHT' + self.ProcessClick(event) + + def OnRightDEvent(self, event): + self.click = 'DRIGHT' + self.ProcessClick(event) + + def OnSetFocus(self, event): + self.hasFocus = True + self.DrawFocusIndicator(True) + + def OnKillFocus(self, event): + self.hasFocus = False + self.DrawFocusIndicator(False) + + def OnKeyDown(self, event): + if not self.hasFocus: + event.Skip() + return + + key_code = event.KeyCode() + + if key_code == wx.WXK_TAB: + forward = not event.ShiftDown() + ne = wx.NavigationKeyEvent() + ne.SetDirection(forward) + ne.SetCurrentFocus(self) + ne.SetEventObject(self) + self.GetParent().GetEventHandler().ProcessEvent(ne) + event.Skip() + return + + delta = None + + if key_code == wx.WXK_UP: + delta = -7 + elif key_code == wx.WXK_DOWN: + delta = 7 + elif key_code == wx.WXK_LEFT: + delta = -1 + elif key_code == wx.WXK_RIGHT: + delta = 1 + elif key_code == wx.WXK_HOME: + curDate = wx.DateTimeFromDMY(int(self.cal_days[self.sel_key]),self.month - 1,self.year) + newDate = wx.DateTime_Now() + ts = newDate - curDate + delta = ts.GetDays() + + if delta <> None: + curDate = wx.DateTimeFromDMY(int(self.cal_days[self.sel_key]),self.month - 1,self.year) + timeSpan = wx.TimeSpan_Days(delta) + newDate = curDate + timeSpan + + if curDate.GetMonth() == newDate.GetMonth(): + self.set_day = newDate.GetDay() + key = self.sel_key + delta + self.SelectDay(key) + else: + self.month = newDate.GetMonth() + 1 + self.year = newDate.GetYear() + self.set_day = newDate.GetDay() + self.sel_key = None + self.DoDrawing(wx.ClientDC(self)) + + event.Skip() + + def SetSize(self, set_size): + self.size = set_size + + def SetSelDay(self, sel): + # list of highlighted days + self.sel_lst = sel + + # get the current date + def SetNow(self): + dt = now() + self.month = dt.month + self.year = dt.year + self.day = dt.day + + # set the current day + def SetCurrentDay(self): + self.SetNow() + self.set_day = self.day + + # get the date, day, month, year set in calendar + + def GetDate(self): + return self.day, self.month, self.year + + def GetDay(self): + return self.day + + def GetMonth(self): + return self.month + + def GetYear(self): + return self.year + + # set the day, month, and year + + def SetDayValue(self, day): + self.set_day = day + + def SetMonth(self, month): + if month >= 1 and month <= 12: + self.month = month + else: + self.month = 1 + self.set_day = None + + def SetYear(self, year): + self.year = year + + # increment year and month + + def IncYear(self): + self.year = self.year + 1 + self.set_day = None + + def DecYear(self): + self.year = self.year - 1 + self.set_day = None + + def IncMonth(self): + self.month = self.month + 1 + if self.month > 12: + self.month = 1 + self.year = self.year + 1 + self.set_day = None + + def DecMonth(self): + self.month = self.month - 1 + if self.month < 1: + self.month = 12 + self.year = self.year - 1 + self.set_day = None + + # test to see if the selection has a date and create event + + def TestDay(self, key): + try: + self.day = int(self.cal_days[key]) + except: + return None + + if self.day == "": + return None + else: + # Changed 12/1/03 by jmg (see above) to support 2.5 event binding + evt = wx.PyCommandEvent(wxEVT_COMMAND_PYCALENDAR_DAY_CLICKED, self.GetId()) + evt.click, evt.day, evt.month, evt.year = self.click, self.day, self.month, self.year + evt.shiftkey = self.shiftkey + evt.ctrlkey = self.ctrlkey + self.GetEventHandler().ProcessEvent(evt) + + self.set_day = self.day + return key + + # find the clicked area rectangle + + def GetDayHit(self, mx, my): + for key in self.rg.keys(): + val = self.rg[key] + ms_rect = wx.Rect(mx, my, 1, 1) + if wx.IntersectRect(ms_rect, val) is not None: + result = self.TestDay(key) + return result + + return None + + # calendar drawing + + def SetWeekColor(self, font_color, week_color): + # set font and background color for week title + self.colors[COLOR_HEADER_FONT] = MakeColor(font_color) + self.colors[COLOR_HEADER_BACKGROUND] = MakeColor(week_color) + self.colors[COLOR_3D_LIGHT] = MakeColor(week_color) + self.colors[COLOR_3D_DARK] = MakeColor(week_color) + + def SetTextAlign(self, vert, horz): + self.num_align_horz = horz + self.num_align_vert = vert + + def AddSelect(self, list, font_color, back_color): + list_val = [list, font_color, back_color] + self.select_list.append(list_val) + + def ShowWeekEnd(self): + # highlight weekend + self.show_weekend = True + + def SetBusType(self): + self.cal_type = "BUS" + + def OnSize(self, evt): + self.Refresh(False) + evt.Skip() + + def OnPaint(self, event): + DC = wx.PaintDC(self) + self.DoDrawing(DC) + + def DoDrawing(self, DC): + #DC = wx.PaintDC(self) + DC.BeginDrawing() + + try: + cal = self.caldraw + except: + self.caldraw = CalDraw(self) + cal = self.caldraw + + cal.hide_grid = self.hide_grid + cal.hide_title = self.hide_title + cal.show_weekend = self.show_weekend + cal.cal_type = self.cal_type + cal.outer_border = self.outer_border + cal.num_align_horz = self.num_align_horz + cal.num_align_vert = self.num_align_vert + cal.colors = self.colors + + if self.size is None: + size = self.GetClientSize() + else: + size = self.size + + # drawing attributes + + cal.SetSize(size) + cal.SetCal(self.year, self.month) + + # these have to set after SetCal as SetCal would overwrite them again. + cal.set_x_mrg = self.set_x_mrg + cal.set_y_mrg = self.set_y_mrg + cal.set_y_end = self.set_y_end + + for val in self.select_list: + cal.AddSelect(val[0], val[1], val[2]) + + cal.DrawCal(DC, self.sel_lst) + + self.rg = cal.GetRect() + self.cal_days = cal.GetCal() + self.st_pos = cal.GetOffset() + self.ymax = DC.MaxY() + + if self.set_day != None: + self.SetDay(self.set_day) + + DC.EndDrawing() + + # draw the selection rectangle + def DrawFocusIndicator(self, draw): + DC = wx.ClientDC(self) + try: + if draw == True: + self.caldraw.DrawFocusIndicator(DC) + else: + self.caldraw.DrawBorder(DC,True) + except: + pass + + def DrawRect(self, key, bgcolor = 'WHITE', fgcolor= 'PINK',width = 0): + if key == None: + return + + DC = wx.ClientDC(self) + DC.BeginDrawing() + + brush = wx.Brush(MakeColor(bgcolor)) + DC.SetBrush(brush) + + DC.SetPen(wx.TRANSPARENT_PEN) + + rect = self.rg[key] + DC.DrawRectangle(rect.x+1, rect.y+1, rect.width-2, rect.height-2) + + self.caldraw.DrawDayText(DC,key) + + DC.EndDrawing() + + def DrawRectOrg(self, key, fgcolor = 'BLACK', width = 0): + if key == None: + return + + DC = wx.ClientDC(self) + DC.BeginDrawing() + + brush = wx.Brush(wx.Colour(0, 0xFF, 0x80), wx.TRANSPARENT) + DC.SetBrush(brush) + + try: + DC.SetPen(wx.Pen(MakeColor(fgcolor), width)) + except: + DC.SetPen(wx.Pen(MakeColor(self.GetColor(COLOR_GRID_LINES)), width)) + + rect = self.rg[key] + DC.DrawRectangleRect(rect) + + DC.EndDrawing() + + # set the day selection + + def SetDay(self, day): + d = day + self.st_pos - 1 + self.SelectDay(d) + + def IsDayInWeekend(self, key): + try: + t = Date(self.year, self.month, 1) + + day = self.cal_days[key] + day = int(day) + t.day_of_week + + if day % 7 == 6 or day % 7 == 0: + return True + except: + return False + + def SelectDay(self, key): + sel_size = 1 + # clear large selection + + if self.sel_key != None: + (cfont, bgcolor) = self.__GetColorsForDay(self.sel_key) + self.DrawRect(self.sel_key, bgcolor,cfont, sel_size) + + self.DrawRect(key, self.GetColor(COLOR_HIGHLIGHT_BACKGROUND), self.GetColor(COLOR_HIGHLIGHT_FONT), sel_size) + # store last used by + self.sel_key = key + + def ClearDsp(self): + self.Clear() + def SetMargin(self, xmarg, ymarg): + self.set_x_mrg = xmarg + self.set_y_mrg = ymarg + self.set_y_end = ymarg + def __GetColorsForDay(self, key): + cfont = self.GetColor(COLOR_FONT) + bgcolor = self.GetColor(COLOR_BACKGROUND) + + if self.IsDayInWeekend(key) is True and self.show_weekend is True: + cfont = self.GetColor(COLOR_WEEKEND_FONT) + bgcolor = self.GetColor(COLOR_WEEKEND_BACKGROUND) + + try: + dayIdx = int(self.cal_days[key]) + (cfont, bgcolor) = self.caldraw.cal_sel[dayIdx] + except: + pass + + return (cfont, bgcolor) + +class CalenDlg(wx.Dialog): + def __init__(self, parent, month=None, day = None, year=None): + wx.Dialog.__init__(self, parent, -1, "Event Calendar", wx.DefaultPosition, (280, 360)) + self.result = None + + # set the calendar and attributes + self.calend = Calendar(self, -1, (20, 60), (240, 200)) + + if month == None: + self.calend.SetCurrentDay() + start_month = self.calend.GetMonth() + start_year = self.calend.GetYear() + else: + self.calend.month = start_month = month + self.calend.year = start_year = year + self.calend.SetDayValue(day) + + self.calend.HideTitle() + self.ResetDisplay() + + # get month list from DateTime + monthlist = GetMonthList() + + # select the month + self.date = wx.ComboBox(self, -1, Month[start_month], (20, 20), (90, -1), + monthlist, wx.CB_DROPDOWN) + self.Bind(wx.EVT_COMBOBOX, self.EvtComboBox, self.date) + + # alternate spin button to control the month + h = self.date.GetSize().height + self.m_spin = wx.SpinButton(self, -1, (115, 20), (h*1.5, h), wx.SP_VERTICAL) + self.m_spin.SetRange(1, 12) + self.m_spin.SetValue(start_month) + self.Bind(wx.EVT_SPIN, self.OnMonthSpin, self.m_spin) + + # spin button to control the year + self.dtext = wx.TextCtrl(self, -1, str(start_year), (160, 20), (60, -1)) + h = self.dtext.GetSize().height + + self.y_spin = wx.SpinButton(self, -1, (225, 20), (h*1.5, h), wx.SP_VERTICAL) + self.y_spin.SetRange(1980, 2010) + self.y_spin.SetValue(start_year) + + self.Bind(wx.EVT_SPIN, self.OnYrSpin, self.y_spin) + self.Bind(EVT_CALENDAR, self.MouseClick, self.calend) + + x_pos = 50 + y_pos = 280 + but_size = (60, 25) + + btn = wx.Button(self, wx.ID_OK, ' Ok ', (x_pos, y_pos), but_size) + self.Bind(wx.EVT_BUTTON, self.OnOk, btn) + + btn = wx.Button(self, wx.ID_CANCEL, ' Close ', (x_pos + 120, y_pos), but_size) + self.Bind(wx.EVT_BUTTON, self.OnCancel, btn) + + def OnOk(self, evt): + self.result = ['None', str(self.calend.day), Month[self.calend.month], str(self.calend.year)] + self.EndModal(wx.ID_OK) + + def OnCancel(self, event): + self.EndModal(wx.ID_CANCEL) + + # log the mouse clicks + def MouseClick(self, evt): + self.month = evt.month + # result click type and date + self.result = [evt.click, str(evt.day), Month[evt.month], str(evt.year)] + + if evt.click == 'DLEFT': + self.EndModal(wx.ID_OK) + + # month and year spin selection routines + def OnMonthSpin(self, event): + month = event.GetPosition() + self.date.SetValue(Month[month]) + self.calend.SetMonth(month) + self.calend.Refresh() + + def OnYrSpin(self, event): + year = event.GetPosition() + self.dtext.SetValue(str(year)) + self.calend.SetYear(year) + self.calend.Refresh() + + def EvtComboBox(self, event): + name = event.GetString() + monthval = self.date.FindString(name) + self.m_spin.SetValue(monthval+1) + + self.calend.SetMonth(monthval+1) + self.ResetDisplay() + + # set the calendar for highlighted days + + def ResetDisplay(self): + month = self.calend.GetMonth() + self.calend.Refresh() -"""Renamer stub: provides a way to drop the wx prefix from wxPython objects.""" -__cvsid__ = "$Id$" -__revision__ = "$Revision$"[11:-2] -from wx import _rename -from wxPython.lib import calendar -_rename(globals(), calendar.__dict__, modulename='lib.calendar') -del calendar -del _rename