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('', contents)
  55                 contents = contents.replace(u'\u2610',
  56                                             u'<LI><INPUT TYPE="checkbox">&nbsp;')
  57
  58                 #self.debug_print("Note contents:\n%s" % contents)
  59                 contents = linkify.sub(r'<a href="\1">\1</a>', contents)
  60
  61                 individual_lines = contents.split("\n")
  62                 num_lines = len(individual_lines)
  63                 max_length = 0
  64                 indent = 0
  65                 contents = ""
  66                 for x in individual_lines:
  67                     length = len(x)
  68                     if length > max_length:
  69                         max_length = length
  70                     spaces = len(x) - len(x.lstrip(' '))
  71                     spaces /= 2
  72                     x = x.lstrip(' ')
  73                     if spaces > indent:
  74                         x = "<UL>" + x
  75                     elif spaces < indent:
  76                         x = "</UL>" + x
  77                     indent = spaces
  78                     contents = contents + x + "\n"
  79
  80                 individual_lines = contents.split("\n")
  81                 color = note.color.name.lower()
  82                 if color in list(self.colors_by_name.keys()):
  83                     color = self.colors_by_name[color]
  84                 else:
  85                     self.debug_print("Unknown color '%s'" % color)
  86                 f = file_writer.file_writer(filename)
  87                 f.write("""
  88 <STYLE type="text/css">
  89   a:link { color:#88bfbf; }
  90   ul { list-style-type:none; }
  91 </STYLE>
  92 <DIV STYLE="border-radius: 25px; border-style: solid; padding: 20px; background-color: %s; color: #eeeeee; font-size: x-large;">
  93 <p style="color: #ffffff; font-size:larger"><B>%s</B></p>
  94 <HR style="border-top: 3px solid white;">""" % (color, note.title))
  95                 if num_lines >= 12 and max_length < 120:
  96                     self.debug_print("%d lines (max=%d chars): two columns" %
  97                                      (num_lines, max_length))
  98                     f.write("<TABLE BORDER=0 WIDTH=100%%><TR valign=\"top\">")
  99                     f.write("<TD WIDTH=50%% style=\"color:#eeeeee; font-size:large\">\n")
 100                     f.write("<FONT><UL>")
 101                     count = 0
 102                     for x in individual_lines:
 103                         f.write(x + "\n")
 104                         count += 1
 105                         if count == num_lines / 2:
 106                             f.write("</UL></FONT></TD>\n")
 107                             f.write("<TD WIDTH=50%% style=\"color:#eeeeee; font-size:large\">\n")
 108                             f.write("<FONT><UL>")
 109                     f.write("</UL></FONT></TD></TR></TABLE></DIV>\n");
 110                 else:
 111                     self.debug_print("%d lines (max=%d chars): one column" %
 112                                      (num_lines, max_length))
 113                     f.write("<FONT><UL>%s</UL></FONT>" % contents)
 114                 f.write("</DIV>")
 115                 f.close()
 116             else:
 117                 self.debug_print("Note is empty, deleting %s." % filename)
 118                 _ = os.path.join(constants.pages_dir, filename)
 119                 try:
 120                     os.remove(_)
 121                 except:
 122                     pass
 123         return True
 124
 125 # Test
 126 #x = gkeep_renderer({"Test", 1234})
 127 #x.periodic_render("Test")