#!/usr/bin/env python3 import logging import os import re from typing import Dict import gkeepapi # type: ignore import kiosk_constants import file_writer import renderer import kiosk_secrets as secrets logger = logging.getLogger(__name__) class gkeep_renderer(renderer.abstaining_renderer): def __init__(self, name_to_timeout_dict: Dict[str, int]) -> None: super().__init__(name_to_timeout_dict) self.colors_by_name = { "white": "#002222", "green": "#345920", "darkblue": "#1F3A5F", "blue": "#2D545E", "orange": "#604A19", "red": "#5C2B29", "purple": "#42275E", "pink": "#5B2245", "yellow": "#635D19", "brown": "#442F19", "gray": "#3c3f4c", "teal": "#16504B", } self.keep = gkeepapi.Keep() 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, key: str) -> bool: strikethrough = re.compile("(\u2611[^\n]*)\n", re.UNICODE) # linkify = re.compile(r".*(https?:\/\/\S+).*") self.keep.sync() result_list = self.keep.find(labels=[self.keep.findLabel("kiosk")]) for note in result_list: title = note.title title = title.replace(" ", "-") title = title.replace("/", "") filename = f"{title}_2_3600.html" contents = note.text + "\n" logger.debug(f"Note title '{title}'") if contents != "" and not contents.isspace(): contents = strikethrough.sub("", contents) logger.debug(f"Note contents:\n{contents}") contents = contents.replace( "\u2610 ", '
{note.title}
\n'
)
f.write("
| \n")
f.write(
'\n'
)
f.write("
|