E-Paper-Calendar/Calendar/EventListDesign.py

37 lines
No EOL
1.5 KiB
Python

from DesignEntity import DesignEntity
from TableTextDesign import TableTextDesign
col_sizes = [0.15, 0.85]
class EventListDesign (DesignEntity):
"""Creates a TableTextDesign filled with event
begin date and title"""
def __init__ (self, size, calendar, event_count = None, text_size = 16):
super(EventListDesign, self).__init__(size)
self.calendar = calendar
self.__event_matrix__ = []
self.text_size = text_size
def __finish_image__ (self):
self.__fill_event_matrix__()
max_col_size = [int(col_sizes[0] * self.size[0]), int(col_sizes[1] * self.size[0])]
col_hori_alignment = ['right', 'left']
table_design = TableTextDesign(self.size, line_spacing=3, col_spacing=10, text_matrix=self.__event_matrix__, fontsize = self.text_size, column_horizontal_alignments=col_hori_alignment, mask=False, max_col_size = max_col_size, truncate_cols=False)
self.draw_design(table_design)
def __get_formatted_event__ (self, event):
date = event.begin_datetime.strftime('%d %b')
date = self.__remove_leading_zero__(date)
return [ date, event.title ]
def __remove_leading_zero__(self, text):
while text[0] is '0':
text = text[1:]
return text
def __fill_event_matrix__ (self):
for event in self.calendar.get_upcoming_events():
row = self.__get_formatted_event__(event)
self.__event_matrix__.append(row)