Inkycal/modules/inkycal_calendar.py

#!/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()