refactoring update
Created new ical_parser refactored inkycal_calendar refactored inkycal_agenda fixed minor bug in write_text function
This commit is contained in:
Ace
307
inkycal/modules/inkycal_calendar.py
Normal file
307
inkycal/modules/inkycal_calendar.py
Normal file
@@ -0,0 +1,307 @@
|
||||
#!/usr/bin/python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Calendar module for Inky-Calendar Project
|
||||
Copyright by aceisace
|
||||
"""
|
||||
|
||||
from inkycal.custom import *
|
||||
import calendar as cal
|
||||
import arrow
|
||||
|
||||
size = (400, 520)
|
||||
config = {'week_starts_on': 'Monday', 'ical_urls': ['https://calendar.google.com/calendar/ical/en.usa%23holiday%40group.v.calendar.google.com/public/basic.ics']}
|
||||
|
||||
|
||||
class calendar:
|
||||
"""Calendar class
|
||||
Create monthly calendar and show events from given icalendars
|
||||
"""
|
||||
logger = logging.getLogger(__name__)
|
||||
logging.basicConfig(level=logging.DEBUG)
|
||||
|
||||
def __init__(self, section_size, section_config):
|
||||
"""Initialize inkycal_calendar module"""
|
||||
|
||||
self.name = os.path.basename(__file__).split('.py')[0]
|
||||
self.config = section_config
|
||||
self.width, self.height = section_size
|
||||
|
||||
self.background_colour = 'white'
|
||||
self.font_colour = 'black'
|
||||
self.fontsize = 12
|
||||
self.font = ImageFont.truetype(
|
||||
fonts['NotoSans-SemiCondensed'], size = self.fontsize)
|
||||
self.padding_x = 0.02
|
||||
self.padding_y = 0.05
|
||||
|
||||
self.weekstart = 'Monday'
|
||||
self.show_events = True
|
||||
self.event_format = "D MMM HH:mm"
|
||||
self.language = 'en' # Grab from settings file?
|
||||
|
||||
self.timezone = get_system_tz()
|
||||
# urls of icalendars
|
||||
self.ical_urls = config['ical_urls']
|
||||
# filepaths of icalendar files
|
||||
self.ical_files = []
|
||||
print('{0} loaded'.format(self.name))
|
||||
|
||||
def set(self, **kwargs):
|
||||
"""Manually set some parameters of this module"""
|
||||
|
||||
for key, value in kwargs.items():
|
||||
if key in self.__dict__:
|
||||
setattr(self, key, value)
|
||||
else:
|
||||
print('{0} does not exist'.format(key))
|
||||
pass
|
||||
|
||||
def get(self, **kwargs):
|
||||
"""Manually get some parameters of this module"""
|
||||
|
||||
for key, value in kwargs.items():
|
||||
if key in self.__dict__:
|
||||
getattr(self, key, value)
|
||||
else:
|
||||
print('{0} does not exist'.format(key))
|
||||
pass
|
||||
|
||||
def get_options(self):
|
||||
"""Get all options which can be changed"""
|
||||
|
||||
return self.__dict__
|
||||
|
||||
def generate_image(self):
|
||||
"""Generate image for this module"""
|
||||
|
||||
# Define new image size with respect to padding
|
||||
im_width = int(self.width - (self.width * 2 * self.padding_x))
|
||||
im_height = int(self.height - (self.height * 2 * self.padding_y))
|
||||
im_size = im_width, im_height
|
||||
|
||||
logging.info('Image size: {0}'.format(im_size))
|
||||
|
||||
# Create an image for black pixels and one for coloured pixels
|
||||
im_black = Image.new('RGB', size = im_size, color = self.background_colour)
|
||||
im_colour = Image.new('RGB', size = im_size, color = 'white')
|
||||
|
||||
# Allocate space for month-names, weekdays etc.
|
||||
month_name_height = int(self.height*0.1)
|
||||
weekdays_height = int(self.height*0.05)
|
||||
|
||||
if self.show_events == True:
|
||||
calendar_height = int(self.height*0.6)
|
||||
events_height = int(self.height*0.25)
|
||||
logging.debug('calendar-section size: {0} x {1} px'.format(
|
||||
im_width, calendar_height))
|
||||
logging.debug('events-section size: {0} x {1} px'.format(
|
||||
im_width, events_height))
|
||||
else:
|
||||
calendar_height = self.height - month_name_height - weekday_height
|
||||
logging.debug('calendar-section size: {0} x {1} px'.format(
|
||||
im_width, calendar_height))
|
||||
|
||||
# Create grid and calculate icon sizes
|
||||
calendar_rows, calendar_cols = 6, 7
|
||||
icon_width = self.width // calendar_cols
|
||||
icon_height = calendar_height // calendar_rows
|
||||
|
||||
# Calculate spacings for calendar area
|
||||
x_spacing_calendar = int((self.width % icon_width) / 2)
|
||||
y_spacing_calendar = int((self.height % calendar_rows) / 2)
|
||||
|
||||
# Calculate positions for days of month
|
||||
grid_start_y = (month_name_height + weekdays_height + y_spacing_calendar)
|
||||
grid_start_x = x_spacing_calendar
|
||||
|
||||
grid = [(grid_start_x + icon_width*x, grid_start_y + icon_height*y)
|
||||
for y in range(calendar_rows) for x in range(calendar_cols)]
|
||||
|
||||
weekday_pos = [(grid_start_x + icon_width*_, month_name_height) for _ in
|
||||
range(calendar_cols)]
|
||||
|
||||
|
||||
now = arrow.now(tz = self.timezone)
|
||||
|
||||
# Set weekstart of calendar to specified weekstart
|
||||
if self.weekstart == "Monday":
|
||||
cal.setfirstweekday(cal.MONDAY)
|
||||
weekstart = now.shift(days = - now.weekday())
|
||||
else:
|
||||
cal.setfirstweekday(cal.SUNDAY)
|
||||
weekstart = now.shift(days = - now.isoweekday())
|
||||
|
||||
# Write the name of current month
|
||||
write(
|
||||
im_black,
|
||||
(x_spacing_calendar,0),
|
||||
(self.width, month_name_height),
|
||||
str(now.format('MMMM',locale=self.language)),
|
||||
font = self.font,
|
||||
autofit = True)
|
||||
|
||||
# Set up weeknames in local language and add to main section
|
||||
weekday_names = [weekstart.shift(days=+_).format('ddd',locale=self.language)
|
||||
for _ in range(7)]
|
||||
logging.debug('weekday names: {}'.format(weekday_names))
|
||||
|
||||
for _ in range(len(weekday_pos)):
|
||||
write(
|
||||
im_black,
|
||||
weekday_pos[_],
|
||||
(icon_width, weekdays_height),
|
||||
weekday_names[_],
|
||||
font = self.font,
|
||||
autofit = True
|
||||
)
|
||||
|
||||
# Create a calendar template and flatten (remove nestings)
|
||||
flatten = lambda z: [x for y in z for x in y]
|
||||
calendar_flat = flatten(cal.monthcalendar(now.year, now.month))
|
||||
|
||||
# Add the numbers on the correct positions
|
||||
for i in range(len(calendar_flat)):
|
||||
if calendar_flat[i] not in (0, int(now.day)):
|
||||
write(
|
||||
im_black,
|
||||
grid[i],
|
||||
(icon_width,icon_height),
|
||||
str(calendar_flat[i]),
|
||||
font = self.font,
|
||||
)
|
||||
|
||||
# Draw a red/black circle with the current day of month in white
|
||||
icon = Image.new('RGBA', (icon_width, icon_height))
|
||||
current_day_pos = grid[calendar_flat.index(now.day)]
|
||||
x_circle,y_circle = int(icon_width/2), int(icon_height/2)
|
||||
radius = int(icon_width * 0.25)
|
||||
text_width, text_height = self.font.getsize(str(now.day))
|
||||
x_text = int((icon_width / 2) - (text_width / 2))
|
||||
y_text = int((icon_height / 2) - (text_height / 1.7))
|
||||
ImageDraw.Draw(icon).ellipse((x_circle-radius, y_circle-radius,
|
||||
x_circle+radius, y_circle+radius), fill= 'black', outline=None)
|
||||
ImageDraw.Draw(icon).text((x_text, y_text), str(now.day), fill='white',
|
||||
font=self.font)
|
||||
im_colour.paste(icon, current_day_pos, icon)
|
||||
|
||||
# If events should be loaded and shown...
|
||||
if self.show_events == True:
|
||||
|
||||
# import the ical-parser
|
||||
from ical_parser import icalendar
|
||||
|
||||
# find out how many lines can fit at max in the event section
|
||||
line_spacing = 0
|
||||
max_event_lines = events_height // (self.font.getsize('hg')[1] +
|
||||
line_spacing)
|
||||
|
||||
# generate list of coordinates for each line
|
||||
event_lines = [(0, grid[-1][1] + int(events_height/max_event_lines*_))
|
||||
for _ in range(max_event_lines)]
|
||||
|
||||
# timeline for filtering events within this month
|
||||
month_start = arrow.get(now.floor('month'))
|
||||
month_end = arrow.get(now.ceil('month'))
|
||||
|
||||
# fetch events from given icalendars
|
||||
parser = icalendar()
|
||||
if self.ical_urls:
|
||||
parser.load_url(self.ical_urls)
|
||||
if self.ical_files:
|
||||
parser.load_from_file(self.ical_files)
|
||||
|
||||
# Filter events for full month (even past ones) for drawing event icons
|
||||
month_events = parser.get_events(month_start, month_end)
|
||||
parser.sort()
|
||||
# parser.show_events() # uncomment to show events
|
||||
|
||||
# find out on which days of this month events are taking place
|
||||
days_with_events = [int(events['begin'].format('D')) for events in
|
||||
month_events]
|
||||
|
||||
# remove duplicates (more than one event in a single day)
|
||||
list(set(days_with_events)).sort()
|
||||
print('days with events:', days_with_events)
|
||||
|
||||
## # calculate sizes for event-markers
|
||||
## square_size = int(icon_width * 0.6)
|
||||
## center_x = int((icon_width - square_size) / 2)
|
||||
## center_y = int((icon_height - square_size) / 2)
|
||||
|
||||
# Draw a border with specified parameters around days with events
|
||||
for days in days_with_events:
|
||||
draw_border(
|
||||
im_colour,
|
||||
grid[calendar_flat.index(days)],
|
||||
(icon_width, icon_height),
|
||||
radius = 4,
|
||||
thickness= 1,
|
||||
shrinkage = (0.4, 0.4)
|
||||
)
|
||||
|
||||
##
|
||||
## draw_square((int(grid[calendar_flat.index(days)][0]+center_x),
|
||||
## int(grid[calendar_flat.index(days)][1] + center_y )),
|
||||
## 8, square_size , square_size, colour='black')
|
||||
|
||||
|
||||
|
||||
|
||||
# Filter upcoming events until 4 weeks in the future
|
||||
parser.clear_events()
|
||||
upcoming_events = parser.get_events(now, now.shift(weeks=4))
|
||||
parser.show_events()
|
||||
|
||||
# delete events which won't be able to fit (more events than lines)
|
||||
upcoming_events[max_event_lines:]
|
||||
|
||||
|
||||
# Check if any events were found in the given timerange
|
||||
if upcoming_events:
|
||||
|
||||
# Find out how much space (width) the date format requires
|
||||
fmt = self.event_format
|
||||
lang = self.language
|
||||
|
||||
date_width = int(max([self.font.getsize(
|
||||
events['begin'].format(fmt,locale=lang))[0]
|
||||
for events in upcoming_events]) * 1.05)
|
||||
|
||||
line_height = self.font.getsize('hg')[1] + line_spacing
|
||||
event_width = im_width - date_width
|
||||
|
||||
# Display upcoming events below calendar
|
||||
tomorrow = now.shift(days=1).floor('day')
|
||||
in_two_days = now.shift(days=2).floor('day')
|
||||
|
||||
# Write events and dates below calendar
|
||||
# TODO: check if events all-day and then don't display time
|
||||
|
||||
cursor = 0
|
||||
for event in upcoming_events:
|
||||
name, date = event['title'], event['begin'].format(fmt, locale=lang)
|
||||
print(date)
|
||||
if now < event['end']:
|
||||
write(im_colour, event_lines[cursor], (date_width, line_height),
|
||||
date, font=self.font, alignment = 'left')
|
||||
write(im_black, (date_width,event_lines[cursor][1]),
|
||||
(event_width, line_height), name, font=self.font,
|
||||
alignment = 'left')
|
||||
cursor += 1
|
||||
else:
|
||||
#leave section empty? or display ----- (dotted line)
|
||||
pass
|
||||
|
||||
|
||||
###################################################################
|
||||
## Exception handling
|
||||
#################################################################
|
||||
|
||||
# Save image of black and colour channel in image-folder
|
||||
im_black.save(images+self.name+'.png')
|
||||
im_colour.save(images+self.name+'_colour.png')
|
||||
|
||||
if __name__ == '__main__':
|
||||
print('running {0} in standalone mode'.format(
|
||||
os.path.basename(__file__).split('.py')[0]))
|
||||
Reference in New Issue
Block a user