#!/usr/bin/env python3
-import gkeepapi
+import logging
import os
import re
-from typing import List, Dict
+from typing import Dict
-import constants
+import gkeepapi # type: ignore
+
+import kiosk_constants
import file_writer
import renderer
-import secrets
+import kiosk_secrets as secrets
+
+logger = logging.getLogger(__name__)
-class gkeep_renderer(renderer.debuggable_abstaining_renderer):
+
+class gkeep_renderer(renderer.abstaining_renderer):
def __init__(self, name_to_timeout_dict: Dict[str, int]) -> None:
- super(gkeep_renderer, self).__init__(name_to_timeout_dict, True)
+ super().__init__(name_to_timeout_dict)
self.colors_by_name = {
"white": "#002222",
"green": "#345920",
"teal": "#16504B",
}
self.keep = gkeepapi.Keep()
- success = self.keep.login(
- secrets.google_keep_username, secrets.google_keep_password
- )
- if success:
- self.debug_print("Connected with gkeep.")
- else:
- self.debug_print("Error connecting with gkeep.")
+ self.token_file = "./.google_keep_token"
+ if os.path.exists(self.token_file):
+ logger.debug("Attempting to reuse persisted Google Keep login token...")
+ try:
+ with open(self.token_file, "r") as rf:
+ token = "".join(rf.readlines()).strip()
+ self.keep.resume(secrets.google_keep_username, token)
+ logger.debug("Successfully reused existing login token.")
+ except gkeepapi.exception.LoginException:
+ logger.warning("Invalid token, attempting to re-login.")
+
+ if not self.keep.login(
+ secrets.google_keep_username,
+ secrets.google_keep_password,
+ secrets.google_keep_mac,
+ ):
+ raise Exception("Error connecting with Google Keep?!")
+ logger.debug("Successfully logged in with Google Keep")
def debug_prefix(self) -> str:
return "gkeep"
- def periodic_render(self: str, key) -> bool:
+ def periodic_render(self, key: str) -> bool:
strikethrough = re.compile("(\u2611[^\n]*)\n", re.UNICODE)
- linkify = re.compile(r".*(https?:\/\/\S+).*")
+ # linkify = re.compile(r".*(https?:\/\/\S+).*")
self.keep.sync()
result_list = self.keep.find(labels=[self.keep.findLabel("kiosk")])
filename = f"{title}_2_3600.html"
contents = note.text + "\n"
- self.debug_print(f"Note title '{title}'")
+ logger.debug(f"Note title '{title}'")
if contents != "" and not contents.isspace():
contents = strikethrough.sub("", contents)
- self.debug_print(f"Note contents:\n{contents}")
+ logger.debug(f"Note contents:\n{contents}")
contents = contents.replace(
"\u2610 ", '<LI><INPUT TYPE="checkbox"> '
)
- contents = linkify.sub(r'<a href="\1">\1</a>', contents)
+
# contents = linkify.sub(r'<a href="\1">\1</a>', contents)
individual_lines = contents.split("\n")
num_lines = len(individual_lines)
- max_length = 0
contents = ""
for x in individual_lines:
- length = len(x)
- if length > max_length:
- max_length = length
leading_spaces = len(x) - len(x.lstrip(" "))
- leading_spaces /= 2
+ leading_spaces //= 2
leading_spaces = int(leading_spaces)
x = x.lstrip(" ")
- # self.debug_print(" * (%d) '%s'" % (leading_spaces, x))
+ # logger.debug(" * (%d) '%s'" % (leading_spaces, x))
for y in range(0, leading_spaces):
x = "<UL>" + x
for y in range(0, leading_spaces):
if color in list(self.colors_by_name.keys()):
color = self.colors_by_name[color]
else:
- self.debug_print(f"Unknown color '{color}'")
+ logger.debug(f"Unknown color '{color}'")
print(f"TITLE: {color} {note.title}")
with file_writer.file_writer(filename) as f:
- f.write("""
+ f.write(
+ """
<STYLE type="text/css">
a:link { color:#88bfbf; }
ul { list-style-type:none; }
</STYLE>
<DIV STYLE="border-radius:25px; border-style:solid; padding:20px; background-color:%s; color:#eeeeee; font-size:x-large;">
-""" % color
- )
- f.write(f"""
+"""
+ % color
+ )
+ f.write(
+ f"""
<p style="color:#ffffff; font-size:larger"><B>{note.title}</B></p>
<HR style="border-top:3px solid white;">
"""
- )
- if num_lines >= 12 and max_length < 120:
- self.debug_print(
- f"{num_lines} lines (max={max_length} chars): two columns"
- )
+ )
+ if num_lines >= 10:
+ logger.debug(f"{num_lines} lines: two column mode")
f.write('<TABLE BORDER=0 WIDTH=100%><TR valign="top">')
f.write(
'<TD WIDTH=50% style="color:#eeeeee; font-size:large">\n'
'<TD WIDTH=50% style="color:#eeeeee; font-size:large">\n'
)
f.write("<FONT><UL STYLE='list-style-type:none'>")
- f.write("</UL></FONT></TD></TR></TABLE></DIV>\n")
+ f.write("</UL></FONT></TD></TR></TABLE></DIV>")
else:
- self.debug_print(
- f"{num_lines} lines (max={max_length} chars): one column"
- )
+ logger.debug(f"{num_lines} lines: one column mode")
f.write(f"<FONT><UL>{contents}</UL></FONT>")
f.write("</DIV>")
else:
- self.debug_print(f"Note is empty, deleting {filename}.")
- _ = os.path.join(constants.pages_dir, filename)
+ logger.debug(f"Note is empty, deleting {filename}.")
+ _ = os.path.join(kiosk_constants.pages_dir, filename)
try:
os.remove(_)
except:
pass
+
+ if self.token_file:
+ token = self.keep.getMasterToken()
+ os.umask(0)
+ descriptor = os.open(
+ path=self.token_file,
+ flags=(os.O_WRONLY | os.O_CREAT | os.O_TRUNC),
+ mode=0o600,
+ )
+ with open(descriptor, "w") as wf:
+ print(token, file=wf)
+ logger.debug("Saved Google Keep token successfully.")
return True
# Test
-#x = gkeep_renderer({"Test", 1234})
-#x.periodic_render("Test")
+# logger.setLevel(logging.DEBUG)
+# ch = logging.StreamHandler()
+# ch.setLevel(logging.DEBUG)
+# formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+# ch.setFormatter(formatter)
+# logger.addHandler(ch)
+# x = gkeep_renderer({"Test", 1234})
+# x.periodic_render("Test")
projects / kiosk.git / blobdiff