plomlompom.com Git - plomtask/blob - plomtask/todos.py

   1 """Actionables."""
   2 from __future__ import annotations
   3 from sqlite3 import Row
   4 from plomtask.db import DatabaseConnection
   5 from plomtask.days import Day
   6 from plomtask.processes import Process
   7 from plomtask.exceptions import NotFoundException
   8
   9
  10 class Todo:
  11     """Individual actionable."""
  12
  13     def __init__(self, id_: int | None, process: Process,
  14                  is_done: bool, day: Day) -> None:
  15         self.id_ = id_
  16         self.process = process
  17         self.is_done = is_done
  18         self.day = day
  19
  20     def __eq__(self, other: object) -> bool:
  21         return isinstance(other, self.__class__) and self.id_ == other.id_
  22
  23     @classmethod
  24     def from_table_row(cls, row: Row, db_conn: DatabaseConnection) -> Todo:
  25         """Make Todo from database row."""
  26         return cls(id_=row[0],
  27                    process=Process.by_id(db_conn, row[1]),
  28                    is_done=row[2],
  29                    day=Day.by_date(db_conn, row[3]))
  30
  31     @classmethod
  32     def by_date(cls, db_conn: DatabaseConnection, date: str) -> list[Todo]:
  33         """Collect all Todos for Day of date."""
  34         todos = []
  35         for row in db_conn.exec('SELECT * FROM todos WHERE day = ?', (date,)):
  36             todos += [cls.from_table_row(row, db_conn)]
  37         return todos
  38
  39     def save(self, db_conn: DatabaseConnection) -> None:
  40         """Write self to DB."""
  41         if self.process.id_ is None:
  42             raise NotFoundException('Process of Todo without ID (not saved?)')
  43         cursor = db_conn.exec('REPLACE INTO todos VALUES (?,?,?,?)',
  44                               (self.id_, self.process.id_,
  45                                self.is_done, self.day.date))
  46         self.id_ = cursor.lastrowid