gkeep_renderer.py

   1 # -*- coding: utf-8 -*-
   2
   3 import constants
   4 import file_writer
   5 import gkeepapi
   6 import os
   7 import re
   8 import renderer
   9 import secrets
  10
  11 class gkeep_renderer(renderer.debuggable_abstaining_renderer):
  12     def __init__(self, name_to_timeout_dict):
  13         super(gkeep_renderer, self).__init__(name_to_timeout_dict, True)
  14         self.keep = gkeepapi.Keep()
  15         success = self.keep.login(secrets.google_keep_username,
  16                                   secrets.google_keep_password)
  17         if success:
  18             self.debug_print("Connected with gkeep.")
  19         else:
  20             self.debug_print("Error connecting with gkeep.")
  21         self.colors_by_name = {
  22             'white' : '#002222',
  23             'green' : '#345920',
  24             'darkblue' : '#1F3A5F',
  25             'blue' : '#2D545E',
  26             'orange' : '#604A19',
  27             'red' : '#5C2B29',
  28             'purple' : '#42275E',
  29             'pink' : '#5B2245',
  30             'yellow' : '#635D19',
  31             'brown' : '#442F19',
  32             'gray' : '#3c3f4c',
  33             'teal' : '#16504B'
  34         }
  35
  36     def debug_prefix(self):
  37         return "gkeep"
  38
  39     def periodic_render(self, key):
  40         strikethrough = re.compile('\u2611([^\n]*)\n', re.UNICODE)
  41         linkify = re.compile(r'.*(https?:\/\/\S+).*')
  42
  43         self.keep.sync()
  44         result_list = self.keep.find(labels=[self.keep.findLabel('kiosk')])
  45         for note in result_list:
  46             title = note.title
  47             title = title.replace(" ", "-")
  48             title = title.replace("/", "")
  49
  50             filename = "%s_2_3600.html" % title
  51             contents = note.text + "\n"
  52             self.debug_print("Note title '%s'" % title)
  53             if contents != '' and not contents.isspace():
  54                 contents = strikethrough.sub(r'<font color="#999999">` <del>\1</del></font>\n', contents)
  55                 contents = contents.replace('`', '\u2611')
  56                 #self.debug_print("Note contents:\n%s" % contents)
  57                 contents = linkify.sub(r'<a href="\1">\1</a>', contents)
  58                 individual_lines = contents.split("\n")
  59                 num_lines = len(individual_lines)
  60                 max_length = 0
  61                 for x in individual_lines:
  62                     length = len(x)
  63                     if length > max_length:
  64                         max_length = length
  65                 contents = contents.replace("\n", "<BR>\n")
  66                 color = note.color.name.lower()
  67                 if color in list(self.colors_by_name.keys()):
  68                     color = self.colors_by_name[color]
  69                 else:
  70                     self.debug_print("Unknown color '%s'" % color)
  71                 f = file_writer.file_writer(filename)
  72                 f.write("""
  73 <STYLE type="text/css">
  74 a:link { color:#88bfbf; }
  75 </STYLE>
  76 <DIV STYLE="border-radius: 25px; border-style: solid; padding: 20px; background-color: %s; color: #eeeeee; font-size: x-large;">
  77 <p style="color: #ffffff; font-size:larger"><B>%s</B></p>
  78 <HR style="border-top: 3px solid white;">""" % (color, note.title))
  79                 if num_lines >= 12 and max_length < 120:
  80                     self.debug_print("%d lines (max=%d chars): two columns" %
  81                                      (num_lines, max_length))
  82                     f.write("<TABLE BORDER=0 WIDTH=100%%><TR valign=\"top\">")
  83                     f.write("<TD WIDTH=50%% style=\"color:#eeeeee; font-size:large\">\n")
  84                     f.write("<FONT>")
  85                     count = 0
  86                     for x in individual_lines:
  87                         f.write(x + "<BR>\n")
  88                         count += 1
  89                         if count == num_lines / 2:
  90                             f.write("</FONT></TD>\n")
  91                             f.write("<TD WIDTH=50%% style=\"color:#eeeeee; font-size:large\">\n")
  92                             f.write("<FONT>")
  93                     f.write("</FONT></TD></TR></TABLE></DIV>\n");
  94                 else:
  95                     self.debug_print("%d lines (max=%d chars): one column" %
  96                                      (num_lines, max_length))
  97                     f.write("<FONT>%s</FONT>" % contents)
  98                 f.write("</DIV>")
  99                 f.close()
 100             else:
 101                 self.debug_print("Note is empty, deleting %s." % filename)
 102                 _ = os.path.join(constants.pages_dir, filename)
 103                 try:
 104                     os.remove(_)
 105                 except:
 106                     pass
 107         return True
 108
 109 # Test
 110 #x = gkeep_renderer({"Test", 1234})
 111 #x.periodic_render("Test")