64dfb023eb
true/false or TRUE/FALSE to prepare for the new boolean type and constants being added to Python. Added code to wx.py to test for the existence of the new constants and to create suitable values if not present. git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/branches/WX_2_4_BRANCH@19335 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
643 lines
18 KiB
Python
643 lines
18 KiB
Python
#----------------------------------------------------------------------------
|
|
# Name: wxCalendar.py
|
|
# Purpose: Calendar control display testing on panel for wxPython demo
|
|
#
|
|
# Author: Lorne White (email: lwhite1@planet.eon.net)
|
|
#
|
|
# Version 0.9
|
|
# Date: Feb 26, 2001
|
|
# Licence: wxWindows license
|
|
#----------------------------------------------------------------------------
|
|
|
|
from wxPython.wx import *
|
|
from wxPython.lib.calendar import wxCalendar, Month, PrtCalDraw, CalenDlg
|
|
|
|
import images
|
|
import os
|
|
|
|
|
|
# highlighted days in month
|
|
|
|
test_days ={ 0: [],
|
|
1: [3, 7, 9, 21],
|
|
2: [2, 10, 4, 9],
|
|
3: [4, 20, 29],
|
|
4: [1, 12, 22],
|
|
5: [2, 10, 15],
|
|
6: [4, 8, 17],
|
|
7: [6, 7, 8],
|
|
8: [5, 10, 20],
|
|
9: [1, 2, 5, 29],
|
|
10: [2, 4, 6, 22],
|
|
11: [6, 9, 12, 28, 29],
|
|
12: [8, 9, 10, 11, 20] }
|
|
|
|
# test of full window calendar control functions
|
|
|
|
def GetMonthList():
|
|
monthlist = []
|
|
for i in range(13):
|
|
name = Month[i]
|
|
if name != None:
|
|
monthlist.append(name)
|
|
return monthlist
|
|
|
|
class TestPanel(wxPanel):
|
|
def __init__(self, parent, log, frame):
|
|
wxPanel.__init__(self, parent, -1)
|
|
|
|
self.log = log
|
|
self.frame = frame
|
|
|
|
self.calend = wxCalendar(self, -1, wxPoint(100, 50), wxSize(200, 180))
|
|
|
|
# start_month = 2 # preselect the date for calendar
|
|
# start_year = 2001
|
|
|
|
start_month = self.calend.GetMonth() # get the current month & year
|
|
start_year = self.calend.GetYear()
|
|
|
|
# month list from DateTime module
|
|
|
|
monthlist = GetMonthList()
|
|
|
|
mID = NewId()
|
|
self.date = wxComboBox(self, mID, "",
|
|
wxPoint(100, 20), wxSize(90, -1),
|
|
monthlist, wxCB_DROPDOWN)
|
|
self.date.SetSelection(start_month-1)
|
|
EVT_COMBOBOX(self, mID, self.EvtComboBox)
|
|
|
|
# set start month and year
|
|
|
|
self.calend.SetMonth(start_month)
|
|
self.calend.SetYear(start_year)
|
|
|
|
# set attributes of calendar
|
|
|
|
self.calend.hide_title = True
|
|
self.calend.HideGrid()
|
|
self.calend.SetWeekColor('WHITE', 'BLACK')
|
|
|
|
# display routine
|
|
|
|
self.ResetDisplay()
|
|
|
|
# mouse click event
|
|
|
|
self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
|
|
|
|
# scroll bar for month selection
|
|
|
|
mID = NewId()
|
|
self.scroll = wxScrollBar(self, mID, wxPoint(100, 240), wxSize(200, 20), wxSB_HORIZONTAL)
|
|
self.scroll.SetScrollbar(start_month-1, 1, 12, 1, True)
|
|
EVT_COMMAND_SCROLL(self, mID, self.Scroll)
|
|
|
|
# spin control for year selection
|
|
|
|
self.dtext = wxTextCtrl(self, -1, str(start_year), wxPoint(200, 20), wxSize(60, -1))
|
|
h = self.dtext.GetSize().height
|
|
|
|
mID = NewId()
|
|
self.spin = wxSpinButton(self, mID, wxPoint(270, 20), wxSize(h*2, h))
|
|
self.spin.SetRange(1980, 2010)
|
|
self.spin.SetValue(start_year)
|
|
EVT_SPIN(self, mID, self.OnSpin)
|
|
|
|
# button for calendar dialog test
|
|
|
|
wxStaticText(self, -1, "Test Calendar Dialog", wxPoint(350, 50), wxSize(150, -1))
|
|
|
|
mID = NewId()
|
|
bmp = images.getCalendarBitmap()
|
|
self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 80))#, wxSize(30, 30))
|
|
EVT_BUTTON(self, mID, self.TestDlg)
|
|
|
|
# button for calendar window test
|
|
|
|
wxStaticText(self, -1, "Test Calendar Window", wxPoint(350, 150), wxSize(150, -1))
|
|
|
|
mID = NewId()
|
|
self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 180))#, wxSize(30, 30))
|
|
EVT_BUTTON(self, mID, self.TestFrame)
|
|
|
|
wxStaticText(self, -1, "Test Calendar Print", wxPoint(350, 250), wxSize(150, -1))
|
|
|
|
mID = NewId()
|
|
self.but = wxBitmapButton(self, mID, bmp, wxPoint(380, 280))#, wxSize(30, 30))
|
|
EVT_BUTTON(self, mID, self.OnPreview)
|
|
|
|
# calendar dialog
|
|
|
|
def TestDlg(self, event): # test the date dialog
|
|
dlg = CalenDlg(self)
|
|
dlg.Centre()
|
|
if dlg.ShowModal() == wxID_OK:
|
|
result = dlg.result
|
|
day = result[1]
|
|
month = result[2]
|
|
year = result[3]
|
|
new_date = str(month) + '/'+ str(day) + '/'+ str(year)
|
|
self.log.WriteText('Date Selected: %s\n' % new_date)
|
|
else:
|
|
self.log.WriteText('No Date Selected')
|
|
|
|
# calendar window test
|
|
|
|
def TestFrame(self, event):
|
|
frame = CalendFrame(self, -1, "Test Calendar", self.log)
|
|
frame.Show(True)
|
|
return True
|
|
|
|
# calendar print preview
|
|
|
|
def OnPreview(self, event):
|
|
month = self.calend.GetMonth()
|
|
year = self.calend.GetYear()
|
|
|
|
prt = PrintCalend(self.frame, month, year)
|
|
prt.Preview()
|
|
|
|
# month and year control events
|
|
|
|
def OnSpin(self, event):
|
|
year = event.GetPosition()
|
|
self.dtext.SetValue(str(year))
|
|
self.calend.SetYear(year)
|
|
self.calend.Refresh()
|
|
|
|
def EvtComboBox(self, event):
|
|
name = event.GetString()
|
|
self.log.WriteText('EvtComboBox: %s\n' % name)
|
|
monthval = self.date.FindString(name)
|
|
self.scroll.SetScrollbar(monthval, 1, 12, 1, True)
|
|
|
|
self.calend.SetMonth(monthval+1)
|
|
self.ResetDisplay()
|
|
|
|
def Scroll(self, event):
|
|
value = self.scroll.GetThumbPosition()
|
|
monthval = int(value)+1
|
|
self.calend.SetMonth(monthval)
|
|
self.ResetDisplay()
|
|
self.log.WriteText('Month: %s\n' % value)
|
|
|
|
name = Month[monthval]
|
|
self.date.SetValue(name)
|
|
|
|
# log mouse events
|
|
|
|
def MouseClick(self, evt):
|
|
text = '%s CLICK %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year) # format date
|
|
self.log.WriteText('Date Selected: ' + text + '\n')
|
|
|
|
|
|
# set the highlighted days for the calendar
|
|
|
|
def ResetDisplay(self):
|
|
month = self.calend.GetMonth()
|
|
try:
|
|
set_days = test_days[month]
|
|
except:
|
|
set_days = [1, 5, 12]
|
|
|
|
self.calend.AddSelect([4, 11], 'BLUE', 'WHITE')
|
|
self.calend.SetSelDay(set_days)
|
|
self.calend.Refresh()
|
|
|
|
# increment and decrement toolbar controls
|
|
|
|
def OnIncYear(self, event):
|
|
self.calend.IncYear()
|
|
self.ResetDisplay()
|
|
|
|
def OnDecYear(self, event):
|
|
self.calend.DecYear()
|
|
self.ResetDisplay()
|
|
|
|
def OnIncMonth(self, event):
|
|
self.calend.IncMonth()
|
|
self.ResetDisplay()
|
|
|
|
def OnDecMonth(self, event):
|
|
self.calend.DecMonth()
|
|
self.ResetDisplay()
|
|
|
|
def OnCurrent(self, event):
|
|
self.calend.SetCurrentDay()
|
|
self.ResetDisplay()
|
|
|
|
# test of full window calendar control functions
|
|
|
|
class CalendFrame(wxFrame):
|
|
def __init__(self, parent, id, title, log):
|
|
wxFrame.__init__(self, parent, id, title, size=wxSize(400, 400),
|
|
style=wxDEFAULT_FRAME_STYLE|wxNO_FULL_REPAINT_ON_RESIZE)
|
|
EVT_CLOSE(self, self.OnCloseWindow)
|
|
|
|
self.log = log
|
|
self.CreateStatusBar()
|
|
self.mainmenu = wxMenuBar()
|
|
menu = wxMenu()
|
|
|
|
menu = self.MakeFileMenu()
|
|
self.mainmenu.Append(menu, '&File')
|
|
|
|
self.MakeToolMenu() # toolbar
|
|
|
|
self.SetMenuBar(self.mainmenu)
|
|
self.calend = wxCalendar(self, -1)
|
|
self.calend.SetCurrentDay()
|
|
self.calend.grid_color = 'BLUE'
|
|
self.calend.SetBusType()
|
|
# self.calend.ShowWeekEnd()
|
|
|
|
self.ResetDisplay()
|
|
|
|
self.Connect(self.calend.GetId(), -1, 2100, self.MouseClick)
|
|
|
|
def MouseClick(self, evt):
|
|
text = '%s CLICK %02d/%02d/%d' % (evt.click, evt.day, evt.month, evt.year) # format date
|
|
self.log.WriteText('Date Selected: ' + text + '\n')
|
|
|
|
def OnCloseWindow(self, event):
|
|
self.Destroy()
|
|
|
|
def ResetDisplay(self):
|
|
month = self.calend.GetMonth()
|
|
try:
|
|
set_days = test_days[month]
|
|
except:
|
|
set_days = [1, 5, 12]
|
|
|
|
self.calend.AddSelect([2, 16], 'GREEN', 'WHITE')
|
|
|
|
self.calend.SetSelDay(set_days)
|
|
self.calend.Refresh()
|
|
|
|
def OnIncYear(self, event):
|
|
self.calend.IncYear()
|
|
self.ResetDisplay()
|
|
|
|
def OnDecYear(self, event):
|
|
self.calend.DecYear()
|
|
self.ResetDisplay()
|
|
|
|
def OnIncMonth(self, event):
|
|
self.calend.IncMonth()
|
|
self.ResetDisplay()
|
|
|
|
def OnDecMonth(self, event):
|
|
self.calend.DecMonth()
|
|
self.ResetDisplay()
|
|
|
|
def OnCurrent(self, event):
|
|
self.calend.SetCurrentDay()
|
|
self.ResetDisplay()
|
|
|
|
def MakeFileMenu(self):
|
|
menu = wxMenu()
|
|
|
|
mID = NewId()
|
|
menu.Append(mID, 'Decrement', 'Next')
|
|
EVT_MENU(self, mID, self.OnDecMonth)
|
|
|
|
mID = NewId()
|
|
menu.Append(mID, 'Increment', 'Dec')
|
|
EVT_MENU(self, mID, self.OnIncMonth)
|
|
|
|
menu.AppendSeparator()
|
|
|
|
mID = NewId()
|
|
menu.Append(mID, 'E&xit', 'Exit')
|
|
EVT_MENU(self, mID, self.OnCloseWindow)
|
|
|
|
return menu
|
|
|
|
def MakeToolMenu(self):
|
|
tb = self.CreateToolBar(wxTB_HORIZONTAL|wxNO_BORDER)
|
|
|
|
mID = NewId()
|
|
SetToolPath(self, tb, mID, images.getDbDecBitmap(), 'Dec Year')
|
|
EVT_TOOL(self, mID, self.OnDecYear)
|
|
|
|
mID = NewId()
|
|
SetToolPath(self, tb, mID, images.getDecBitmap(), 'Dec Month')
|
|
EVT_TOOL(self, mID, self.OnDecMonth)
|
|
|
|
mID = NewId()
|
|
SetToolPath(self, tb, mID, images.getPtBitmap(), 'Current Month')
|
|
EVT_TOOL(self, mID, self.OnCurrent)
|
|
|
|
mID = NewId()
|
|
SetToolPath(self, tb, mID, images.getIncBitmap(), 'Inc Month')
|
|
EVT_TOOL(self, mID, self.OnIncMonth)
|
|
|
|
mID = NewId()
|
|
SetToolPath(self, tb, mID, images.getDbIncBitmap(), 'Inc Year')
|
|
EVT_TOOL(self, mID, self.OnIncYear)
|
|
|
|
tb.Realize()
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
# example class for printing/previewing calendars
|
|
|
|
class PrintCalend:
|
|
def __init__(self, parent, month, year):
|
|
self.frame = parent
|
|
self.month = month
|
|
self.year = year
|
|
|
|
self.SetParms()
|
|
self.SetCal()
|
|
self.printData = wxPrintData()
|
|
|
|
def SetCal(self):
|
|
self.grid_color = 'BLUE'
|
|
self.back_color = 'WHITE'
|
|
self.sel_color = 'RED'
|
|
self.high_color = 'LIGHT BLUE'
|
|
self.font = wxSWISS
|
|
self.bold = wxNORMAL
|
|
|
|
self.sel_key = None # last used by
|
|
self.sel_lst = [] # highlighted selected days
|
|
|
|
self.size = None
|
|
self.hide_title = False
|
|
self.hide_grid = False
|
|
self.set_day = None
|
|
|
|
def SetParms(self):
|
|
self.ymax = 1
|
|
self.xmax = 1
|
|
self.page = 1
|
|
self.total_pg = 1
|
|
|
|
self.preview = None
|
|
self.scale = 1.0
|
|
|
|
self.pagew = 8.5
|
|
self.pageh = 11.0
|
|
|
|
self.txt_marg = 0.1
|
|
self.lf_marg = 0
|
|
self.top_marg = 0
|
|
|
|
self.page = 0
|
|
|
|
def SetDates(self, month, year):
|
|
self.month = month
|
|
self.year = year
|
|
|
|
def SetStyleDef(self, desc):
|
|
self.style = desc
|
|
|
|
def SetCopies(self, copies): # number of copies of label
|
|
self.copies = copies
|
|
|
|
def SetStart(self, start): # start position of label
|
|
self.start = start
|
|
|
|
def Preview(self):
|
|
printout = SetPrintout(self)
|
|
printout2 = SetPrintout(self)
|
|
self.preview = wxPrintPreview(printout, printout2, self.printData)
|
|
if not self.preview.Ok():
|
|
wxMessageBox("There was a problem printing!", "Printing", wxOK)
|
|
return
|
|
|
|
self.preview.SetZoom(60) # initial zoom value
|
|
|
|
frame = wxPreviewFrame(self.preview, self.frame, "Print preview")
|
|
|
|
frame.Initialize()
|
|
frame.SetPosition(self.frame.GetPosition())
|
|
frame.SetSize(self.frame.GetSize())
|
|
frame.Show(True)
|
|
|
|
def Print(self):
|
|
pdd = wxPrintDialogData()
|
|
pdd.SetPrintData(self.printData)
|
|
printer = wxPrinter(pdd)
|
|
printout = SetPrintout(self)
|
|
frame = wxFrame(None, -1, "Test")
|
|
if not printer.Print(frame, printout):
|
|
wxMessageBox("There was a problem printing.\nPerhaps your current printer is not set correctly?", "Printing", wxOK)
|
|
else:
|
|
self.printData = printer.GetPrintDialogData().GetPrintData()
|
|
printout.Destroy()
|
|
|
|
def DoDrawing(self, DC):
|
|
size = DC.GetSizeTuple()
|
|
DC.BeginDrawing()
|
|
|
|
cal = PrtCalDraw(self)
|
|
|
|
if self.preview is None:
|
|
cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
|
|
cal.SetPreview(False)
|
|
|
|
else:
|
|
if self.preview == 1:
|
|
cal.SetPSize(size[0]/self.pagew, size[1]/self.pageh)
|
|
else:
|
|
cal.SetPSize(self.pwidth, self.pheight)
|
|
|
|
cal.SetPreview(self.preview)
|
|
|
|
cal.hide_title = self.hide_title # set the calendar parameters
|
|
cal.hide_grid = self.hide_grid
|
|
|
|
cal.grid_color = self.grid_color
|
|
cal.high_color = self.high_color
|
|
cal.back_color = self.back_color
|
|
cal.outer_border = False
|
|
cal.font = self.font
|
|
cal.bold = self.bold
|
|
|
|
cal_size = wxSize(3.0, 3.0)
|
|
cal.SetSize(cal_size)
|
|
|
|
year, month = self.year, self.month
|
|
|
|
x = 1.0
|
|
for i in range(2):
|
|
y = 0.5
|
|
for j in range(3):
|
|
cal.SetCal(year, month) # current month
|
|
cal.SetPos(x, y)
|
|
|
|
try:
|
|
set_days = test_days[month]
|
|
except:
|
|
set_days = [1, 5, 12]
|
|
|
|
cal.AddSelect([2, 16], 'GREEN', 'WHITE')
|
|
|
|
cal.DrawCal(DC, set_days)
|
|
|
|
year, month = self.IncMonth(year, month)
|
|
y = y + 3.5
|
|
x = x + 4.0 # next colum
|
|
|
|
DC.EndDrawing()
|
|
|
|
self.ymax = DC.MaxY()
|
|
self.xmax = DC.MaxX()
|
|
|
|
def IncMonth(self, year, month): # next month
|
|
month = month + 1
|
|
if month > 12:
|
|
month = 1
|
|
year = year + 1
|
|
|
|
return year, month
|
|
|
|
def GetTotalPages(self):
|
|
self.pg_cnt = 1
|
|
return self.pg_cnt
|
|
|
|
def SetPage(self, page):
|
|
self.page = page
|
|
|
|
def SetPageSize(self, width, height):
|
|
self.pwidth, self.pheight = width, height
|
|
|
|
def SetTotalSize(self, width, height):
|
|
self.ptwidth, self.ptheight = width, height
|
|
|
|
def SetPreview(self, preview, scale):
|
|
self.preview = preview
|
|
self.scale = scale
|
|
|
|
def SetTotalSize(self, width, height):
|
|
self.ptwidth = width
|
|
self.ptheight = height
|
|
|
|
def SetToolPath(self, tb, id, bmp, title):
|
|
tb.AddSimpleTool(id, bmp, title, title)
|
|
|
|
class SetPrintout(wxPrintout):
|
|
def __init__(self, canvas):
|
|
wxPrintout.__init__(self)
|
|
self.canvas = canvas
|
|
self.end_pg = 1
|
|
|
|
def OnBeginDocument(self, start, end):
|
|
return self.base_OnBeginDocument(start, end)
|
|
|
|
def OnEndDocument(self):
|
|
self.base_OnEndDocument()
|
|
|
|
def HasPage(self, page):
|
|
if page <= self.end_pg:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def GetPageInfo(self):
|
|
self.end_pg = self.canvas.GetTotalPages()
|
|
str_pg = 1
|
|
try:
|
|
end_pg = self.end_pg
|
|
except:
|
|
end_pg = 1
|
|
return (str_pg, end_pg, str_pg, end_pg)
|
|
|
|
def OnPreparePrinting(self):
|
|
self.base_OnPreparePrinting()
|
|
|
|
def OnBeginPrinting(self):
|
|
dc = self.GetDC()
|
|
|
|
self.preview = self.IsPreview()
|
|
if (self.preview):
|
|
self.pixelsPerInch = self.GetPPIScreen()
|
|
else:
|
|
self.pixelsPerInch = self.GetPPIPrinter()
|
|
|
|
(w, h) = dc.GetSizeTuple()
|
|
scaleX = float(w) / 1000
|
|
scaleY = float(h) / 1000
|
|
self.printUserScale = min(scaleX, scaleY)
|
|
|
|
self.base_OnBeginPrinting()
|
|
|
|
def GetSize(self):
|
|
self.psizew, self.psizeh = self.GetPPIPrinter()
|
|
return self.psizew, self.psizeh
|
|
|
|
def GetTotalSize(self):
|
|
self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
|
|
return self.ptsizew, self.ptsizeh
|
|
|
|
def OnPrintPage(self, page):
|
|
dc = self.GetDC()
|
|
(w, h) = dc.GetSizeTuple()
|
|
scaleX = float(w) / 1000
|
|
scaleY = float(h) / 1000
|
|
self.printUserScale = min(scaleX, scaleY)
|
|
dc.SetUserScale(self.printUserScale, self.printUserScale)
|
|
|
|
self.preview = self.IsPreview()
|
|
|
|
self.canvas.SetPreview(self.preview, self.printUserScale)
|
|
self.canvas.SetPage(page)
|
|
|
|
self.ptsizew, self.ptsizeh = self.GetPageSizePixels()
|
|
self.canvas.SetTotalSize(self.ptsizew, self.ptsizeh)
|
|
|
|
self.psizew, self.psizeh = self.GetPPIPrinter()
|
|
self.canvas.SetPageSize(self.psizew, self.psizeh)
|
|
|
|
self.canvas.DoDrawing(dc)
|
|
return True
|
|
|
|
class MyApp(wxApp):
|
|
def OnInit(self):
|
|
frame = CalendFrame(None, -1, "Test Calendar")
|
|
frame.Show(True)
|
|
self.SetTopWindow(frame)
|
|
return True
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def MessageDlg(self, message, type = 'Message'):
|
|
dlg = wxMessageDialog(self, message, type, wxOK | wxICON_INFORMATION)
|
|
dlg.ShowModal()
|
|
dlg.Destroy()
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def main():
|
|
app = MyApp(0)
|
|
app.MainLoop()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def runTest(frame, nb, log):
|
|
win = TestPanel(nb, log, frame)
|
|
return win
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
|
|
overview = """\
|
|
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.
|
|
|
|
Additional features include weekend highlighting and business type Monday-Sunday format.
|
|
|
|
See example for various methods used to set display month, year, and highlighted dates (different font and background colours).
|
|
|
|
by Lorne White
|
|
|
|
"""
|