1e4a197e4c
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@19793 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
|
|
|
|
"""
|