Improved formatting
This commit is contained in:
		| @@ -1,5 +1,5 @@ | ||||
| #!/usr/bin/python3 | ||||
| # -*- coding: utf-8 -*- | ||||
| #!python3 | ||||
|  | ||||
| """ | ||||
| Agenda module for Inky-Calendar Project | ||||
| Copyright by aceisace | ||||
| @@ -15,219 +15,219 @@ import arrow | ||||
| filename = os.path.basename(__file__).split('.py')[0] | ||||
| logger = logging.getLogger(filename) | ||||
|  | ||||
|  | ||||
| class Agenda(inkycal_module): | ||||
|   """Agenda class | ||||
|   Create agenda and show events from given icalendars | ||||
|   """ | ||||
|     """Agenda class | ||||
|     Create agenda and show events from given icalendars | ||||
|     """ | ||||
|  | ||||
|   name = "Agenda - Display upcoming events from given iCalendars" | ||||
|     name = "Agenda - Display upcoming events from given iCalendars" | ||||
|  | ||||
|   requires = { | ||||
|     "ical_urls" : { | ||||
|       "label":"iCalendar URL/s, separate multiple ones with a comma", | ||||
|       }, | ||||
|     requires = { | ||||
|         "ical_urls": { | ||||
|             "label": "iCalendar URL/s, separate multiple ones with a comma", | ||||
|         }, | ||||
|  | ||||
|     } | ||||
|  | ||||
|   optional = { | ||||
|     "ical_files" : { | ||||
|       "label":"iCalendar filepaths, separated with a comma", | ||||
|       }, | ||||
|     optional = { | ||||
|         "ical_files": { | ||||
|             "label": "iCalendar filepaths, separated with a comma", | ||||
|         }, | ||||
|  | ||||
|     "date_format":{ | ||||
|       "label":"Use an arrow-supported token for custom date formatting "+ | ||||
|       "see https://arrow.readthedocs.io/en/stable/#supported-tokens, e.g. ddd D MMM", | ||||
|       "default": "ddd D MMM", | ||||
|       }, | ||||
|  | ||||
|     "time_format":{ | ||||
|       "label":"Use an arrow-supported token for custom time formatting "+ | ||||
|       "see https://arrow.readthedocs.io/en/stable/#supported-tokens, e.g. HH:mm", | ||||
|       "default":"HH:mm", | ||||
|       }, | ||||
|         "date_format": { | ||||
|             "label": "Use an arrow-supported token for custom date formatting " + | ||||
|                      "see https://arrow.readthedocs.io/en/stable/#supported-tokens, e.g. ddd D MMM", | ||||
|             "default": "ddd D MMM", | ||||
|         }, | ||||
|  | ||||
|         "time_format": { | ||||
|             "label": "Use an arrow-supported token for custom time formatting " + | ||||
|                      "see https://arrow.readthedocs.io/en/stable/#supported-tokens, e.g. HH:mm", | ||||
|             "default": "HH:mm", | ||||
|         }, | ||||
|  | ||||
|     } | ||||
|  | ||||
|   def __init__(self, config): | ||||
|     """Initialize inkycal_agenda module""" | ||||
|     def __init__(self, config): | ||||
|         """Initialize inkycal_agenda module""" | ||||
|  | ||||
|     super().__init__(config) | ||||
|         super().__init__(config) | ||||
|  | ||||
|     config = config['config'] | ||||
|         config = config['config'] | ||||
|  | ||||
|     # Check if all required parameters are present | ||||
|     for param in self.requires: | ||||
|       if not param in config: | ||||
|         raise Exception(f'config is missing {param}') | ||||
|         logger.exception(f'config is missing "{param}"') | ||||
|         # Check if all required parameters are present | ||||
|         for param in self.requires: | ||||
|             if not param in config: | ||||
|                 raise Exception(f'config is missing {param}') | ||||
|                 logger.exception(f'config is missing "{param}"') | ||||
|  | ||||
|     # module specific parameters | ||||
|     self.date_format = config['date_format'] | ||||
|     self.time_format = config['time_format'] | ||||
|     self.language = config['language'] | ||||
|         # module specific parameters | ||||
|         self.date_format = config['date_format'] | ||||
|         self.time_format = config['time_format'] | ||||
|         self.language = config['language'] | ||||
|  | ||||
|     # Check if ical_files is an empty string | ||||
|     if config['ical_urls'] and isinstance(config['ical_urls'], str): | ||||
|       self.ical_urls = config['ical_urls'].split(',') | ||||
|     else: | ||||
|       self.ical_urls = config['ical_urls'] | ||||
|         # Check if ical_files is an empty string | ||||
|         if config['ical_urls'] and isinstance(config['ical_urls'], str): | ||||
|             self.ical_urls = config['ical_urls'].split(',') | ||||
|         else: | ||||
|             self.ical_urls = config['ical_urls'] | ||||
|  | ||||
|     # Check if ical_files is an empty string | ||||
|     if config['ical_files'] and isinstance(config['ical_files'], str): | ||||
|       self.ical_files = config['ical_files'].split(',') | ||||
|     else: | ||||
|       self.ical_files = config['ical_files'] | ||||
|         # Check if ical_files is an empty string | ||||
|         if config['ical_files'] and isinstance(config['ical_files'], str): | ||||
|             self.ical_files = config['ical_files'].split(',') | ||||
|         else: | ||||
|             self.ical_files = config['ical_files'] | ||||
|  | ||||
|     # Additional config | ||||
|     self.timezone = get_system_tz() | ||||
|         # Additional config | ||||
|         self.timezone = get_system_tz() | ||||
|  | ||||
|     # give an OK message | ||||
|     print(f'{filename} loaded') | ||||
|         # give an OK message | ||||
|         print(f'{filename} loaded') | ||||
|  | ||||
|   def generate_image(self): | ||||
|     """Generate image for this module""" | ||||
|     def generate_image(self): | ||||
|         """Generate image for this module""" | ||||
|  | ||||
|     # Define new image size with respect to padding | ||||
|     im_width = int(self.width - (2 * self.padding_left)) | ||||
|     im_height = int(self.height - (2 * self.padding_top)) | ||||
|     im_size = im_width, im_height | ||||
|         # Define new image size with respect to padding | ||||
|         im_width = int(self.width - (2 * self.padding_left)) | ||||
|         im_height = int(self.height - (2 * self.padding_top)) | ||||
|         im_size = im_width, im_height | ||||
|  | ||||
|     logger.info(f'Image size: {im_size}') | ||||
|         logger.info(f'Image size: {im_size}') | ||||
|  | ||||
|     # Create an image for black pixels and one for coloured pixels | ||||
|     im_black = Image.new('RGB', size = im_size, color = 'white') | ||||
|     im_colour = Image.new('RGB', size = im_size, color = 'white') | ||||
|         # Create an image for black pixels and one for coloured pixels | ||||
|         im_black = Image.new('RGB', size=im_size, color='white') | ||||
|         im_colour = Image.new('RGB', size=im_size, color='white') | ||||
|  | ||||
|     # Calculate the max number of lines that can fit on the image | ||||
|     line_spacing = 1 | ||||
|     line_height = int(self.font.getsize('hg')[1]) + line_spacing | ||||
|     line_width = im_width | ||||
|     max_lines = im_height // line_height | ||||
|     logger.debug(f'max lines: {max_lines}') | ||||
|         # Calculate the max number of lines that can fit on the image | ||||
|         line_spacing = 1 | ||||
|         line_height = int(self.font.getsize('hg')[1]) + line_spacing | ||||
|         line_width = im_width | ||||
|         max_lines = im_height // line_height | ||||
|         logger.debug(f'max lines: {max_lines}') | ||||
|  | ||||
|     # Create timeline for agenda | ||||
|     now = arrow.now() | ||||
|     today = now.floor('day') | ||||
|         # Create timeline for agenda | ||||
|         now = arrow.now() | ||||
|         today = now.floor('day') | ||||
|  | ||||
|     # Create a list of dates for the next days | ||||
|     agenda_events = [ | ||||
|       {'begin':today.shift(days=+_), | ||||
|        'title': today.shift(days=+_).format( | ||||
|          self.date_format,locale=self.language)} | ||||
|       for _ in range(max_lines)] | ||||
|         # Create a list of dates for the next days | ||||
|         agenda_events = [ | ||||
|             {'begin': today.shift(days=+_), | ||||
|              'title': today.shift(days=+_).format( | ||||
|                  self.date_format, locale=self.language)} | ||||
|             for _ in range(max_lines)] | ||||
|  | ||||
|     # Load icalendar from config | ||||
|     self.ical = iCalendar() | ||||
|     parser = self.ical | ||||
|         # Load icalendar from config | ||||
|         self.ical = iCalendar() | ||||
|         parser = self.ical | ||||
|  | ||||
|     if self.ical_urls: | ||||
|       parser.load_url(self.ical_urls) | ||||
|         if self.ical_urls: | ||||
|             parser.load_url(self.ical_urls) | ||||
|  | ||||
|     if self.ical_files: | ||||
|       parser.load_from_file(self.ical_files) | ||||
|         if self.ical_files: | ||||
|             parser.load_from_file(self.ical_files) | ||||
|  | ||||
|     # Load events from all icalendar in timerange | ||||
|     upcoming_events = parser.get_events(today, agenda_events[-1]['begin'], | ||||
|                                         self.timezone) | ||||
|         # Load events from all icalendar in timerange | ||||
|         upcoming_events = parser.get_events(today, agenda_events[-1]['begin'], | ||||
|                                             self.timezone) | ||||
|  | ||||
|     # Sort events by beginning time | ||||
|     parser.sort() | ||||
|     #parser.show_events() | ||||
|         # Sort events by beginning time | ||||
|         parser.sort() | ||||
|         # parser.show_events() | ||||
|  | ||||
|     # Set the width for date, time and event titles | ||||
|     date_width = int(max([self.font.getsize( | ||||
|           dates['begin'].format(self.date_format, locale=self.language))[0] | ||||
|           for dates in agenda_events]) * 1.2) | ||||
|     logger.debug(f'date_width: {date_width}') | ||||
|         # Set the width for date, time and event titles | ||||
|         date_width = int(max([self.font.getsize( | ||||
|             dates['begin'].format(self.date_format, locale=self.language))[0] | ||||
|                               for dates in agenda_events]) * 1.2) | ||||
|         logger.debug(f'date_width: {date_width}') | ||||
|  | ||||
|     # Calculate positions for each line | ||||
|     line_pos = [(0, int(line * line_height)) for line in range(max_lines)] | ||||
|     logger.debug(f'line_pos: {line_pos}') | ||||
|         # Calculate positions for each line | ||||
|         line_pos = [(0, int(line * line_height)) for line in range(max_lines)] | ||||
|         logger.debug(f'line_pos: {line_pos}') | ||||
|  | ||||
|     # Check if any events were filtered | ||||
|     if upcoming_events: | ||||
|       logger.info('Managed to parse events from urls') | ||||
|         # Check if any events were filtered | ||||
|         if upcoming_events: | ||||
|             logger.info('Managed to parse events from urls') | ||||
|  | ||||
|       # Find out how much space the event times take | ||||
|       time_width = int(max([self.font.getsize( | ||||
|           events['begin'].format(self.time_format, locale=self.language))[0] | ||||
|           for events in upcoming_events]) * 1.2) | ||||
|       logger.debug(f'time_width: {time_width}') | ||||
|             # Find out how much space the event times take | ||||
|             time_width = int(max([self.font.getsize( | ||||
|                 events['begin'].format(self.time_format, locale=self.language))[0] | ||||
|                                   for events in upcoming_events]) * 1.2) | ||||
|             logger.debug(f'time_width: {time_width}') | ||||
|  | ||||
|       # Calculate x-pos for time | ||||
|       x_time = date_width | ||||
|       logger.debug(f'x-time: {x_time}') | ||||
|             # Calculate x-pos for time | ||||
|             x_time = date_width | ||||
|             logger.debug(f'x-time: {x_time}') | ||||
|  | ||||
|       # Find out how much space is left for event titles | ||||
|       event_width = im_width - time_width - date_width | ||||
|       logger.debug(f'width for events: {event_width}') | ||||
|             # Find out how much space is left for event titles | ||||
|             event_width = im_width - time_width - date_width | ||||
|             logger.debug(f'width for events: {event_width}') | ||||
|  | ||||
|       # Calculate x-pos for event titles | ||||
|       x_event = date_width + time_width | ||||
|       logger.debug(f'x-event: {x_event}') | ||||
|             # Calculate x-pos for event titles | ||||
|             x_event = date_width + time_width | ||||
|             logger.debug(f'x-event: {x_event}') | ||||
|  | ||||
|       # Merge list of dates and list of events | ||||
|       agenda_events += upcoming_events | ||||
|             # Merge list of dates and list of events | ||||
|             agenda_events += upcoming_events | ||||
|  | ||||
|       # Sort the combined list in chronological order of dates | ||||
|       by_date = lambda event: event['begin'] | ||||
|       agenda_events.sort(key = by_date) | ||||
|             # Sort the combined list in chronological order of dates | ||||
|             by_date = lambda event: event['begin'] | ||||
|             agenda_events.sort(key=by_date) | ||||
|  | ||||
|       # Delete more entries than can be displayed (max lines) | ||||
|       del agenda_events[max_lines:] | ||||
|             # Delete more entries than can be displayed (max lines) | ||||
|             del agenda_events[max_lines:] | ||||
|  | ||||
|       self._agenda_events = agenda_events | ||||
|             self._agenda_events = agenda_events | ||||
|  | ||||
|       cursor = 0 | ||||
|       for _ in agenda_events: | ||||
|         title = _['title'] | ||||
|             cursor = 0 | ||||
|             for _ in agenda_events: | ||||
|                 title = _['title'] | ||||
|  | ||||
|         # Check if item is a date | ||||
|         if not 'end' in _: | ||||
|           ImageDraw.Draw(im_colour).line( | ||||
|             (0, line_pos[cursor][1], im_width, line_pos[cursor][1]), | ||||
|           fill = 'black') | ||||
|                 # Check if item is a date | ||||
|                 if not 'end' in _: | ||||
|                     ImageDraw.Draw(im_colour).line( | ||||
|                         (0, line_pos[cursor][1], im_width, line_pos[cursor][1]), | ||||
|                         fill='black') | ||||
|  | ||||
|           write(im_black, line_pos[cursor], (date_width, line_height), | ||||
|               title, font = self.font, alignment='left') | ||||
|                     write(im_black, line_pos[cursor], (date_width, line_height), | ||||
|                           title, font=self.font, alignment='left') | ||||
|  | ||||
|           cursor += 1 | ||||
|                     cursor += 1 | ||||
|  | ||||
|         # Check if item is an event | ||||
|         if 'end' in _: | ||||
|           time = _['begin'].format(self.time_format) | ||||
|                 # Check if item is an event | ||||
|                 if 'end' in _: | ||||
|                     time = _['begin'].format(self.time_format) | ||||
|  | ||||
|           # Check if event is all day, if not, add the time | ||||
|           if parser.all_day(_) == False: | ||||
|             write(im_black, (x_time, line_pos[cursor][1]), | ||||
|                 (time_width, line_height), time, | ||||
|                 font = self.font, alignment='left') | ||||
|                     # Check if event is all day, if not, add the time | ||||
|                     if parser.all_day(_) == False: | ||||
|                         write(im_black, (x_time, line_pos[cursor][1]), | ||||
|                               (time_width, line_height), time, | ||||
|                               font=self.font, alignment='left') | ||||
|  | ||||
|           write(im_black, (x_event, line_pos[cursor][1]), | ||||
|                 (event_width, line_height), | ||||
|                 '• '+title, font = self.font, alignment='left') | ||||
|           cursor += 1 | ||||
|                     write(im_black, (x_event, line_pos[cursor][1]), | ||||
|                           (event_width, line_height), | ||||
|                           '• ' + title, font=self.font, alignment='left') | ||||
|                     cursor += 1 | ||||
|  | ||||
|     # If no events were found, write only dates and lines | ||||
|     else: | ||||
|       logger.info('no events found') | ||||
|         # If no events were found, write only dates and lines | ||||
|         else: | ||||
|             logger.info('no events found') | ||||
|  | ||||
|       cursor = 0 | ||||
|       for _ in agenda_events: | ||||
|         title = _['title'] | ||||
|         ImageDraw.Draw(im_colour).line( | ||||
|             (0, line_pos[cursor][1], im_width, line_pos[cursor][1]), | ||||
|             fill = 'black') | ||||
|             cursor = 0 | ||||
|             for _ in agenda_events: | ||||
|                 title = _['title'] | ||||
|                 ImageDraw.Draw(im_colour).line( | ||||
|                     (0, line_pos[cursor][1], im_width, line_pos[cursor][1]), | ||||
|                     fill='black') | ||||
|  | ||||
|         write(im_black, line_pos[cursor], (date_width, line_height), | ||||
|               title, font = self.font, alignment='left') | ||||
|                 write(im_black, line_pos[cursor], (date_width, line_height), | ||||
|                       title, font=self.font, alignment='left') | ||||
|  | ||||
|         cursor += 1 | ||||
|                 cursor += 1 | ||||
|  | ||||
|         # return the images ready for the display | ||||
|         return im_black, im_colour | ||||
|  | ||||
|     # return the images ready for the display | ||||
|     return im_black, im_colour | ||||
|  | ||||
| if __name__ == '__main__': | ||||
|   print(f'running {filename} in standalone mode') | ||||
|     print(f'running {filename} in standalone mode') | ||||
|   | ||||
		Reference in New Issue
	
	Block a user