Reasonably-working Kimai API data fetch'n'display

hamstertools/db.py

@@ -200,3 +200,69 @@ class Fact(BaseORM):
         cursor.execute("SELECT * FROM facts")
         rows = cursor.fetchall()
         return [Fact(db_manager, row[0], row[1]) for row in rows]
+
+
+class KimaiCustomer(BaseORM):
+    def __init__(self, db_manager, id, name):
+        super().__init__(db_manager, "kimai_customers", id, name=name)
+
+    def save(self):
+        cursor = self.db_manager.get_cursor()
+        cursor.execute("SELECT id FROM kimai_customers WHERE id = ?", (self.id,))
+        row = cursor.fetchone()
+        if row:
+            cursor.execute("""
+                UPDATE kimai_customers SET name = ? WHERE id = ?
+            """, (self.name, self.id))
+        else:
+            cursor.execute("""
+                INSERT INTO kimai_customers (id, name) VALUES (?, ?)
+            """, (self.id, self.name))
+        self.db_manager.get_conn().commit()
+
+
+
+class KimaiProject(BaseORM):
+    def __init__(self, db_manager, id, name, customer_id, customer_name):
+        super().__init__(db_manager, "kimai_projects", id, name=name,
+                         customer_id=customer_id, customer_name=customer_name)
+
+    def save(self):
+        cursor = self.db_manager.get_cursor()
+        cursor.execute("SELECT id FROM kimai_projects WHERE id = ?", (self.id,))
+        row = cursor.fetchone()
+        if row:
+            cursor.execute("""
+                UPDATE kimai_projects SET name = ?, customer_id = ? WHERE id = ?
+            """, (self.name, self.customer_id, self.id))
+        else:
+            cursor.execute("""
+                INSERT INTO kimai_projects (id, name, customer_id) VALUES (?, ?, ?)
+            """, (self.id, self.name, self.customer_id))
+        self.db_manager.get_conn().commit()
+
+    @staticmethod
+    def list(db_manager):
+        cursor = db_manager.get_cursor()
+        cursor.execute("""
+            SELECT
+                kimai_projects.id,
+                COALESCE(kimai_projects.name, ""),
+                kimai_customers.id,
+                COALESCE(kimai_customers.name, "")
+            FROM
+                kimai_projects
+            LEFT JOIN
+                kimai_customers
+            ON
+                kimai_customers.id = kimai_projects.customer_id
+            GROUP BY
+                kimai_customers.id
+        """)
+
+        rows = cursor.fetchall()
+        return [KimaiProject(db_manager, row[0], row[1], row[2], row[3]) for row in rows]
+
+
+class KimaiACtivity(BaseORM):
+    pass