micropython-nano-gui/extras/widgets/calendar.py

62 wiersze
2.5 KiB
Python

# calendar.py Calendar object
# Released under the MIT License (MIT). See LICENSE.
# Copyright (c) 2023 Peter Hinch
from extras.widgets.grid import Grid
from gui.widgets.label import Label, ALIGN_CENTER
from extras.date import DateCal
class Calendar:
def __init__(
self, wri, row, col, colwidth, fgcolor, bgcolor, today_c, cur_c, sun_c, today_inv=False, cur_inv=False
):
self.fgcolor = fgcolor
self.bgcolor = bgcolor
self.today_c = today_c # Color of "today" cell
self.today_inv = today_inv
self.cur_c = cur_c # Calendar currency
self.cur_inv = cur_inv
self.sun_c = sun_c # Sundays
self.date = DateCal()
self.date.callback = self.show
rows = 6
cols = 7
self.ncells = cols * (rows - 1) # Row 0 has day labels
self.last_cell = cols * rows
lw = (colwidth + 4) * cols # Label width = width of grid
kwargs = {"align": ALIGN_CENTER, "fgcolor": fgcolor, "bgcolor": bgcolor}
self.lbl = Label(wri, row, col, lw, **kwargs)
row += self.lbl.height + 3 # Two border widths
self.grid = Grid(wri, row, col, colwidth, rows, cols, **kwargs)
self.grid.show() # Draw grid lines
self.grid[0, 0:7] = iter([d[:3] for d in DateCal.days]) # 3-char day names
self.show()
def days(self, month_length): # Produce content for every cell
for n in range(self.ncells + 1):
yield str(n + 1) if n < month_length else ""
def show(self):
grid = self.grid
cur = self.date # Currency
self.lbl.value(f"{DateCal.months[cur.month - 1]} {cur.year}")
values = self.days(cur.month_length) # Instantiate generator
idx_1 = 7 + cur.wday_n(1) # Index of 1st of month
grid[idx_1 : self.last_cell] = values
grid[7 : idx_1] = values
# Assign colors. Last to be applied has priority.
grid[1:6, 6] = {"fgcolor": self.sun_c} # Sunday color
idx_cur = idx_1 + cur.mday - 1 # Currency (overrides Sunday)
if self.cur_inv:
grid[idx_cur] = {"fgcolor": self.bgcolor, "bgcolor": self.cur_c}
else:
grid[idx_cur] = {"fgcolor": self.cur_c}
today = DateCal()
if cur.year == today.year and cur.month == today.month: # Today is in current month
idx = idx_1 + today.mday - 1
if self.today_inv:
grid[idx] = {"fgcolor": self.bgcolor, "bgcolor": self.fgcolor}
else:
grid[idx] = {"fgcolor": self.today_c}