| """ | |
| Simple calendar using ttk Treeview together with calendar and datetime | |
| classes. | |
| """ | |
| import calendar | |
| import Tkinter | |
| import tkFont | |
| import ttk | |
| def get_calendar(locale, fwday): | |
| # instantiate proper calendar class | |
| if locale is None: | |
| return calendar.TextCalendar(fwday) | |
| else: | |
| return calendar.LocaleTextCalendar(fwday, locale) | |
| class Calendar(ttk.Frame): | |
| # XXX ToDo: cget and configure | |
| datetime = calendar.datetime.datetime | |
| timedelta = calendar.datetime.timedelta | |
| def __init__(self, master=None, **kw): | |
| """ | |
| WIDGET-SPECIFIC OPTIONS | |
| locale, firstweekday, year, month, selectbackground, | |
| selectforeground | |
| """ | |
| # remove custom options from kw before initializating ttk.Frame | |
| fwday = kw.pop('firstweekday', calendar.MONDAY) | |
| year = kw.pop('year', self.datetime.now().year) | |
| month = kw.pop('month', self.datetime.now().month) | |
| locale = kw.pop('locale', None) | |
| sel_bg = kw.pop('selectbackground', '#ecffc4') | |
| sel_fg = kw.pop('selectforeground', '#05640e') | |
| self._date = self.datetime(year, month, 1) | |
| self._selection = None # no date selected | |
| ttk.Frame.__init__(self, master, **kw) | |
| self._cal = get_calendar(locale, fwday) | |
| self.__setup_styles() # creates custom styles | |
| self.__place_widgets() # pack/grid used widgets | |
| self.__config_calendar() # adjust calendar columns and setup tags | |
| # configure a canvas, and proper bindings, for selecting dates | |
| self.__setup_selection(sel_bg, sel_fg) | |
| # store items ids, used for insertion later | |
| self._items = [self._calendar.insert('', 'end', values='') | |
| for _ in range(6)] | |
| # insert dates in the currently empty calendar | |
| self._build_calendar() | |
| # set the minimal size for the widget | |
| self._calendar.bind('<Map>', self.__minsize) | |
| def __setitem__(self, item, value): | |
| if item in ('year', 'month'): | |
| raise AttributeError("attribute '%s' is not writeable" % item) | |
| elif item == 'selectbackground': | |
| self._canvas['background'] = value | |
| elif item == 'selectforeground': | |
| self._canvas.itemconfigure(self._canvas.text, item=value) | |
| else: | |
| ttk.Frame.__setitem__(self, item, value) | |
| def __getitem__(self, item): | |
| if item in ('year', 'month'): | |
| return getattr(self._date, item) | |
| elif item == 'selectbackground': | |
| return self._canvas['background'] | |
| elif item == 'selectforeground': | |
| return self._canvas.itemcget(self._canvas.text, 'fill') | |
| else: | |
| r = ttk.tclobjs_to_py({item: ttk.Frame.__getitem__(self, item)}) | |
| return r[item] | |
| def __setup_styles(self): | |
| # custom ttk styles | |
| style = ttk.Style(self.master) | |
| arrow_layout = lambda dir: ( | |
| [('Button.focus', {'children': [('Button.%sarrow' % dir, None)]})] | |
| ) | |
| style.layout('L.TButton', arrow_layout('left')) | |
| style.layout('R.TButton', arrow_layout('right')) | |
| def __place_widgets(self): | |
| # header frame and its widgets | |
| hframe = ttk.Frame(self) | |
| lbtn = ttk.Button(hframe, style='L.TButton', command=self._prev_month) | |
| rbtn = ttk.Button(hframe, style='R.TButton', command=self._next_month) | |
| self._header = ttk.Label(hframe, width=15, anchor='center') | |
| # the calendar | |
| self._calendar = ttk.Treeview(show='', selectmode='none', height=7) | |
| # pack the widgets | |
| hframe.pack(in_=self, side='top', pady=4, anchor='center') | |
| lbtn.grid(in_=hframe) | |
| self._header.grid(in_=hframe, column=1, row=0, padx=12) | |
| rbtn.grid(in_=hframe, column=2, row=0) | |
| self._calendar.pack(in_=self, expand=1, fill='both', side='bottom') | |
| def __config_calendar(self): | |
| cols = self._cal.formatweekheader(3).split() | |
| self._calendar['columns'] = cols | |
| self._calendar.tag_configure('header', background='grey90') | |
| self._calendar.insert('', 'end', values=cols, tag='header') | |
| # adjust its columns width | |
| font = tkFont.Font() | |
| maxwidth = max(font.measure(col) for col in cols) | |
| for col in cols: | |
| self._calendar.column(col, width=maxwidth, minwidth=maxwidth, | |
| anchor='e') | |
| def __setup_selection(self, sel_bg, sel_fg): | |
| self._font = tkFont.Font() | |
| self._canvas = canvas = Tkinter.Canvas(self._calendar, | |
| background=sel_bg, borderwidth=0, highlightthickness=0) | |
| canvas.text = canvas.create_text(0, 0, fill=sel_fg, anchor='w') | |
| canvas.bind('<ButtonPress-1>', lambda evt: canvas.place_forget()) | |
| self._calendar.bind('<Configure>', lambda evt: canvas.place_forget()) | |
| self._calendar.bind('<ButtonPress-1>', self._pressed) | |
| def __minsize(self, evt): | |
| width, height = self._calendar.master.geometry().split('x') | |
| height = height[:height.index('+')] | |
| self._calendar.master.minsize(width, height) | |
| def _build_calendar(self): | |
| year, month = self._date.year, self._date.month | |
| # update header text (Month, YEAR) | |
| header = self._cal.formatmonthname(year, month, 0) | |
| self._header['text'] = header.title() | |
| # update calendar shown dates | |
| cal = self._cal.monthdayscalendar(year, month) | |
| for indx, item in enumerate(self._items): | |
| week = cal[indx] if indx < len(cal) else [] | |
| fmt_week = [('%02d' % day) if day else '' for day in week] | |
| self._calendar.item(item, values=fmt_week) | |
| def _show_selection(self, text, bbox): | |
| """Configure canvas for a new selection.""" | |
| x, y, width, height = bbox | |
| textw = self._font.measure(text) | |
| canvas = self._canvas | |
| canvas.configure(width=width, height=height) | |
| canvas.coords(canvas.text, width - textw, height / 2 - 1) | |
| canvas.itemconfigure(canvas.text, text=text) | |
| canvas.place(in_=self._calendar, x=x, y=y) | |
| # Callbacks | |
| def _pressed(self, evt): | |
| """Clicked somewhere in the calendar.""" | |
| x, y, widget = evt.x, evt.y, evt.widget | |
| item = widget.identify_row(y) | |
| column = widget.identify_column(x) | |
| if not column or not item in self._items: | |
| # clicked in the weekdays row or just outside the columns | |
| return | |
| item_values = widget.item(item)['values'] | |
| if not len(item_values): # row is empty for this month | |
| return | |
| text = item_values[int(column[1]) - 1] | |
| if not text: # date is empty | |
| return | |
| bbox = widget.bbox(item, column) | |
| if not bbox: # calendar not visible yet | |
| return | |
| # update and then show selection | |
| text = '%02d' % text | |
| self._selection = (text, item, column) | |
| self._show_selection(text, bbox) | |
| def _prev_month(self): | |
| """Updated calendar to show the previous month.""" | |
| self._canvas.place_forget() | |
| self._date = self._date - self.timedelta(days=1) | |
| self._date = self.datetime(self._date.year, self._date.month, 1) | |
| self._build_calendar() # reconstruct calendar | |
| def _next_month(self): | |
| """Update calendar to show the next month.""" | |
| self._canvas.place_forget() | |
| year, month = self._date.year, self._date.month | |
| self._date = self._date + self.timedelta( | |
| days=calendar.monthrange(year, month)[1] + 1) | |
| self._date = self.datetime(self._date.year, self._date.month, 1) | |
| self._build_calendar() # reconstruct calendar | |
| # Properties | |
| @property | |
| def selection(self): | |
| """Return a datetime representing the current selected date.""" | |
| if not self._selection: | |
| return None | |
| year, month = self._date.year, self._date.month | |
| return self.datetime(year, month, int(self._selection[0])) | |
| def test(): | |
| import sys | |
| root = Tkinter.Tk() | |
| root.title('Ttk Calendar') | |
| ttkcal = Calendar(firstweekday=calendar.SUNDAY) | |
| ttkcal.pack(expand=1, fill='both') | |
| if 'win' not in sys.platform: | |
| style = ttk.Style() | |
| style.theme_use('clam') | |
| root.mainloop() | |
| if __name__ == '__main__': | |
| test() |