97 lines
3.6 KiB
Python
97 lines
3.6 KiB
Python
from DesignEntity import DesignEntity
|
|
import calendar as callib
|
|
from datetime import datetime, timedelta
|
|
from TextDesign import TextDesign
|
|
from Assets import colors
|
|
from settings import week_starts_on
|
|
from BoxDesign import BoxDesign
|
|
|
|
daynumberboxsize = (0.143, 0.2)
|
|
dayhighlightboxsize = (0.143, 0.14)
|
|
daynumbersize = daynumberboxsize[0] * 0.45
|
|
day_number_ypadding = -0.002
|
|
|
|
|
|
class MonthBlockDesign (DesignEntity):
|
|
"""Creates a view containing one week of the month in
|
|
one row"""
|
|
|
|
def __init__(self, size, datetime_month, highlight_today=False):
|
|
super(MonthBlockDesign, self).__init__(size, mask=True)
|
|
self.month = datetime_month.month
|
|
self.year = datetime_month.year
|
|
self.highlight_today = highlight_today
|
|
self.__week_days__ = self.__get_week_days_ordered__()
|
|
|
|
def __finish_image__(self):
|
|
self.__draw_month_overview__()
|
|
|
|
def __draw_month_overview__(self):
|
|
"""Using the built-in calendar function, draw icons for each
|
|
number of the month (1,2,3,...29,30,31)"""
|
|
cal = callib.monthcalendar(self.year, self.month)
|
|
for week in cal:
|
|
for numbers in week:
|
|
self.__draw_day_number__(numbers, self.get_day_pos(
|
|
cal.index(week), week.index(numbers)))
|
|
|
|
if self.highlight_today:
|
|
self.__draw_highlight_box__(self.__abs_pos__(
|
|
dayhighlightboxsize), self.__get_today_box_pos__(), width=3)
|
|
|
|
def __draw_highlight_box__(self, size, pos, color=colors["fg"], width=1):
|
|
design = BoxDesign(size, outline=color, width=width)
|
|
design.pos = pos
|
|
self.draw_design(design)
|
|
|
|
def __draw_day_number__(self, number, pos):
|
|
if number <= 0:
|
|
return
|
|
txt = TextDesign(self.__abs_pos__(daynumberboxsize), fontsize=daynumbersize *
|
|
self.size[0], text=str(number), verticalalignment="center", horizontalalignment="center")
|
|
txt.pos = (pos[0], pos[1] + day_number_ypadding * self.size[1])
|
|
self.draw_design(txt)
|
|
|
|
def get_day_pos(self, week_in_month, day_of_week, rel_pos=(0, 0)):
|
|
maxwidth, maxheight = self.size
|
|
partialwidth = maxwidth / 7
|
|
partialheight = maxheight / 5
|
|
return (int(rel_pos[0] + day_of_week * partialwidth), int(rel_pos[1] + week_in_month * partialheight))
|
|
|
|
def __get_today_box_pos__(self):
|
|
x, y = self.get_day_pos(self.__get_week_of_month__(
|
|
datetime.now()), self.__get_day_of_week__(datetime.now()))
|
|
return (x, int(y + (self.__abs_pos__(daynumberboxsize)[1] - self.__abs_pos__(dayhighlightboxsize)[1]) / 2))
|
|
|
|
def __get_week_of_month__(self, date):
|
|
for wof, week in enumerate(callib.monthcalendar(date.year, date.month)):
|
|
if date.day in week:
|
|
return wof
|
|
return 0
|
|
|
|
def __get_day_of_week__(self, date):
|
|
return self.__week_days__.index(date.strftime("%a"))
|
|
|
|
def __get_week_days_ordered__(self):
|
|
cur_weekday = datetime.now().weekday()
|
|
correction = -cur_weekday
|
|
if week_starts_on == "Sunday":
|
|
correction -= 1
|
|
|
|
weekdays = []
|
|
for i in range(7):
|
|
weekdays.append(
|
|
(datetime.now() + timedelta(days=(i + correction))).strftime("%a"))
|
|
|
|
return weekdays
|
|
|
|
def __abs_pos__(self, pos, size=None):
|
|
if size is None:
|
|
size = self.size
|
|
return (int(pos[0] * size[0]), int(pos[1] * size[1]))
|
|
|
|
def get_real_height(self):
|
|
weeks_in_month = callib.monthcalendar(self.year, self.month)
|
|
num_size = self.__abs_pos__(daynumberboxsize)
|
|
num_pos = self.get_day_pos(len(weeks_in_month) - 1, 6)
|
|
return num_size[1] + num_pos[1]
|