X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D/balance2?a=blobdiff_plain;f=plomtask%2Fdb.py;h=f53a94c4f085f3664d66f0b0cc38863b217741e1;hb=56b8aa166d51d139edb41f0ef250b1854a5e93e8;hp=929a733cc381a6d6d7a2e24feda04a5a014081a7;hpb=8ae8877e3e2588db76285e7e3ddfb8c7b9948a96;p=plomtask

diff --git a/plomtask/db.py b/plomtask/db.py
index 929a733..f53a94c 100644
--- a/plomtask/db.py
+++ b/plomtask/db.py
@@ -2,7 +2,7 @@
 from os.path import isfile
 from difflib import Differ
 from sqlite3 import connect as sql_connect, Cursor
-from typing import Any
+from typing import Any, Dict
 from plomtask.exceptions import HandledException
 
 PATH_DB_SCHEMA = 'scripts/init.sql'
@@ -23,7 +23,7 @@ class DatabaseFile:  # pylint: disable=too-few-public-methods
         self._check()
 
     def _check(self) -> None:
-        """Check file exists and is of proper schema."""
+        """Check file exists, and is of proper schema."""
         self.exists = isfile(self.path)
         if self.exists:
             self._validate_schema()
@@ -49,6 +49,11 @@ class DatabaseConnection:
     def __init__(self, db_file: DatabaseFile) -> None:
         self.file = db_file
         self.conn = sql_connect(self.file.path)
+        self.cached_todos: Dict[int, Any] = {}
+        self.cached_days: Dict[str, Any] = {}
+        self.cached_process_steps: Dict[int, Any] = {}
+        self.cached_processes: Dict[int, Any] = {}
+        self.cached_conditions: Dict[int, Any] = {}
 
     def commit(self) -> None:
         """Commit SQL transaction."""