E-Paper-Calendar/Calendar/MonthBlockDesign.py

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]