9517017365
With the new folder structure, users can now easily clone the repo with git clone. The need for renaming and shifting folders is now gone.
195 lines
7.4 KiB
Python
195 lines
7.4 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Calendar module for Inky-Calendar Project
|
|
Copyright by aceisace
|
|
"""
|
|
from __future__ import print_function
|
|
import calendar
|
|
from configuration import *
|
|
from settings import *
|
|
import arrow
|
|
from PIL import Image, ImageDraw
|
|
|
|
print_events = False
|
|
show_events = True
|
|
max_event_lines = 4
|
|
style = "DD MMM"
|
|
event_icon = 'square' # dot #square
|
|
|
|
if show_events == True:
|
|
from inkycal_icalendar import fetch_events
|
|
|
|
"""Add a border to increase readability"""
|
|
border_top = int(middle_section_height * 0.02)
|
|
border_left = int(middle_section_width * 0.02)
|
|
|
|
main_area_height = middle_section_height-border_top*2
|
|
main_area_width = middle_section_width-border_left*2
|
|
|
|
"""Calculate height for each sub-section"""
|
|
month_name_height = int(main_area_height*0.1)
|
|
weekdays_height = int(main_area_height*0.05)
|
|
calendar_height = int(main_area_height*0.6)
|
|
events_height = int(main_area_height*0.25)
|
|
|
|
"""Set rows and coloumns in the calendar section and calculate sizes"""
|
|
calendar_rows, calendar_coloumns = 6, 7
|
|
icon_width = main_area_width // calendar_coloumns
|
|
icon_height = calendar_height // calendar_rows
|
|
|
|
"""Calculate paddings for calendar section"""
|
|
x_padding_calendar = int((main_area_width % icon_width) / 2)
|
|
y_padding_calendar = int((main_area_height % calendar_rows) / 2)
|
|
|
|
"""Add coordinates for number icons inside the calendar section"""
|
|
grid_start_y = (middle_section_offset + border_top + month_name_height +
|
|
weekdays_height + y_padding_calendar)
|
|
grid_start_x = border_left + x_padding_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_coloumns)]
|
|
|
|
weekday_pos = [(grid_start_x + icon_width*_, middle_section_offset +
|
|
month_name_height) for _ in range(calendar_coloumns)]
|
|
|
|
event_lines = [(border_left,(bottom_section_offset - events_height)+
|
|
int(events_height/max_event_lines*_)) for _ in
|
|
range(max_event_lines)]
|
|
|
|
def main():
|
|
try:
|
|
clear_image('middle_section')
|
|
print('Calendar module: Generating image...', end = '')
|
|
now = arrow.now(tz = get_tz())
|
|
|
|
"""Set up the Calendar template based on personal preferences"""
|
|
if week_starts_on == "Monday":
|
|
calendar.setfirstweekday(calendar.MONDAY)
|
|
weekstart = now.replace(days = - now.weekday())
|
|
else:
|
|
calendar.setfirstweekday(calendar.SUNDAY)
|
|
weekstart = now.replace(days = - now.isoweekday())
|
|
|
|
"""Write the name of the current month at the correct position"""
|
|
write_text(main_area_width, month_name_height,
|
|
str(now.format('MMMM',locale=language)), (border_left,
|
|
middle_section_offset), autofit = True)
|
|
|
|
"""Set up weeknames in local language and add to main section"""
|
|
weekday_names = [weekstart.replace(days=+_).format('ddd',locale=language)
|
|
for _ in range(7)]
|
|
|
|
for _ in range(len(weekday_pos)):
|
|
write_text(icon_width, weekdays_height, weekday_names[_],
|
|
weekday_pos[_], 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(calendar.monthcalendar(now.year, now.month))
|
|
|
|
"""Add the numbers on the correct positions"""
|
|
for i in range(len(calendar_flat)):
|
|
if calendar_flat[i] != 0:
|
|
write_text(icon_width, icon_height, str(calendar_flat[i]), grid[i])
|
|
|
|
"""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 = default.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= 'red' if
|
|
display_type == 'colour' else 'black', outline=None)
|
|
ImageDraw.Draw(icon).text((x_text, y_text), str(now.day), fill='white',
|
|
font=bold)
|
|
image.paste(icon, current_day_pos, icon)
|
|
|
|
"""Create some reference points for the current month"""
|
|
days_current_month = calendar.monthrange(now.year, now.month)[1]
|
|
month_start = now.replace(days =-now.day+1)
|
|
month_end = now.replace(days=+(days_current_month-now.day))
|
|
|
|
if show_events == True:
|
|
"""Filter events which begin before the end of this month"""
|
|
upcoming_events = fetch_events()
|
|
|
|
calendar_events = [events for events in upcoming_events if
|
|
events.begin < month_end and events.begin.month == now.month]
|
|
|
|
"""Find days with events in the current month"""
|
|
days_with_events = []
|
|
for events in calendar_events:
|
|
if events.duration.days <= 1:
|
|
days_with_events.append(int(events.begin.format('D')))
|
|
else:
|
|
for day in range(events.duration.days):
|
|
days_with_events.append(
|
|
int(events.begin.replace(days=+i).format('D')))
|
|
days_with_events = set(days_with_events)
|
|
|
|
if event_icon == 'dot':
|
|
for days in days_with_events:
|
|
write_text(icon_width, int(icon_height * 0.2), '•',
|
|
(grid[calendar_flat.index(days)][0],
|
|
int(grid[calendar_flat.index(days)][1] + icon_height*0.8)))
|
|
|
|
if event_icon == 'square':
|
|
square_size = int(icon_width *0.6)
|
|
center_x = int((icon_width - square_size) / 2)
|
|
center_y = int((icon_height - square_size) / 2)
|
|
for days in days_with_events:
|
|
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)
|
|
|
|
|
|
"""Add a small section showing events of today and tomorrow"""
|
|
event_list = ['{0} at {1} : {2}'.format('today', event.begin.format(
|
|
'HH:mm' if hours == 24 else 'hh:mm'), event.name)
|
|
for event in calendar_events if event.begin.day == now.day]
|
|
|
|
event_list += ['{0} at {1} : {2}'.format('tomorrow', event.begin.format(
|
|
'HH:mm' if hours == 24 else 'hh:mm'), event.name)
|
|
for event in calendar_events if event.begin.day == now.replace(days=+1).day]
|
|
|
|
del event_list[4:]
|
|
|
|
if event_list:
|
|
for lines in event_list:
|
|
write_text(main_area_width, int(events_height/max_event_lines), lines,
|
|
event_lines[event_list.index(lines)], alignment='left',
|
|
fill_height = 0.7)
|
|
else:
|
|
write_text(main_area_width, int(events_height/max_event_lines),
|
|
'No events today or tomorrow', event_lines[0], alignment='left',
|
|
fill_height = 0.7)
|
|
|
|
"""Set print_events_to True to print all events in this month"""
|
|
style = 'DD MMM YY HH:mm'
|
|
if print_events == True and calendar_events:
|
|
line_width = max(len(_.name) for _ in calendar_events)
|
|
for events in calendar_events:
|
|
print('{0} {1} | {2} | {3} | All day ='.format(events.name,
|
|
' ' * (line_width - len(events.name)), events.begin.format(style),
|
|
events.end.format(style)), events.all_day)
|
|
|
|
calendar_image = crop_image(image, 'middle_section')
|
|
calendar_image.save(image_path+'calendar.png')
|
|
|
|
print('Done')
|
|
|
|
except Exception as e:
|
|
"""If something went wrong, print a Error message on the Terminal"""
|
|
print('Failed!')
|
|
print('Error in Calendar module!')
|
|
print('Reason: ',e)
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|