From 66264fe5730943576b847077dadf5e4531d4beb1 Mon Sep 17 00:00:00 2001
From: Frederick Yin <fkfd@fkfd.me>
Date: Wed, 2 Feb 2022 11:15:09 +0800
Subject: Adapt core to new db

---
 jimbrella/umbrellas.py | 41 ++++++++++++++++++++++++++++-------------
 1 file changed, 28 insertions(+), 13 deletions(-)

(limited to 'jimbrella/umbrellas.py')

diff --git a/jimbrella/umbrellas.py b/jimbrella/umbrellas.py
index e59bb54..6da7184 100644
--- a/jimbrella/umbrellas.py
+++ b/jimbrella/umbrellas.py
@@ -26,7 +26,7 @@ class Umbrellas:
         - tenant_id     | string. student or faculty ID.
         - tenant_phone  | string. phone number via which to contact tenant when the lease is due.
         - tenant_email  | string. for future compatibility. always None for the time being.
-        - lent_at       | an ISO 8601 date string "YYYY-MM-DDThh:mm:ss+08:00" if status is
+        - lent_at       | an ISO 8601 date string "YYYY-MM-DDThh:mm:ss.mmm+08:00" if status is
                         | "lent" or "overdue. is None otherwise.
 
         Schema:
@@ -43,13 +43,13 @@ class Umbrellas:
         self.path = path
 
     def read(self) -> list:
-        db = sqlite3.connect(path)
+        db = sqlite3.connect(self.path)
         db.row_factory = sqlite.Row
         umbrellas = db.execute("SELECT * FROM Umbrellas").fetchall()
         db.close()
         return umbrellas
 
-    def update(self, umb) -> None:
+    def update(self, umb) -> dict:
         """Update Umbrella table with new data given in `umb`.
 
         Not all fields in an umbrella dict need to be present in `umb`. Only `id` is required.
@@ -61,6 +61,9 @@ class Umbrellas:
         If `status` is not "lent" or "overdue", `tenant_*` and `lent_at` are automatically erased.
 
         `lent_at` may be either an ISO 8601 string or a datetime.datetime object. Must be UTC+8.
+
+        Returns a dict of <field>: (<prior_value>, <updated_value>) for each updated field unless
+        its erasure can be inferred. For AdminLog.
         """
         # `id` must be specified.
         try:
@@ -68,7 +71,7 @@ class Umbrellas:
         except (KeyError, ValueError):
             raise UmbrellaValueError("id")
 
-        db = sqlite3.connect(path)
+        db = sqlite3.connect(self.path)
         db.row_factory = sqlite.Row
 
         # check if umbrella #<id> exists in database
@@ -77,12 +80,18 @@ class Umbrellas:
         if umb_in_db is None:
             raise UmbrellaNotFoundError(umbid)
 
+        diff = {}
+
         status = umb_in_db["status"]
-        if "status" in umb and umb["status"] in STATUSES:
-            status = umb["status"]
-            db.execute("UPDATE Umbrellas SET status = ? WHERE id = ?", status, umbid)
-        else:
-            raise UmbrellaValueError("status")
+        if "status" in umb:
+            if umb["status"] in STATUSES:
+                status = umb["status"]
+                if umb_in_db["status"] != umb["status"]:
+                    diff["status"] = (umb_in_db["status"], umb["status"])
+
+                db.execute("UPDATE Umbrellas SET status = ? WHERE id = ?", status, umbid)
+            else:
+                raise UmbrellaValueError("status")
 
         if status in ("lent", "overdue"):
             for key in (
@@ -92,10 +101,13 @@ class Umbrellas:
                 "tenant_email",
             ):
                 if col in umb:
+                    if umb_in_db[col] != umb[col]:
+                        diff[col] = (umb_in_db[col], umb[col])
+
                     db.execute(
                         "UPDATE Umbrellas SET ? = ? WHERE id = ?",
                         col,
-                        umb[col] or None,
+                        umb[col],
                         umbid,
                     )
 
@@ -114,6 +126,8 @@ class Umbrellas:
                     # timezone must be +08:00
                     raise UmbrellaValueError("lent_at")
 
+                diff["lent_at"] = (umb_in_db["lent_at"], lent_at.isoformat(timespec="milliseconds"))
+
                 db.execute(
                     "UPDATE Umbrellas SET lent_at = ? WHERE id = ?",
                     lent_at.isoformat(timespec="milliseconds"),
@@ -133,12 +147,13 @@ class Umbrellas:
         # now that new data are validated, commit the SQL transaction
         db.commit()
         db.close()
+        return diff
 
     def take_away(
         self, umbid, date, tenant_name, tenant_id, tenant_phone="", tenant_email=""
     ) -> None:
         """When a user has borrowed an umbrella."""
-        db = sqlite3.connect(path)
+        db = sqlite3.connect(self.path)
         db.row_factory = sqlite3.Row
         umb = db.execute("SELECT * FROM Umbrellas WHERE id = ?", umbid)
         db.close()
@@ -166,7 +181,7 @@ class Umbrellas:
         `tenant_name` and `tenant_id` are used to verify if the umbrella is returned by the same
         person who borrowed it.
         """
-        db = sqlite3.connect(path)
+        db = sqlite3.connect(self.path)
         db.row_factory = sqlite3.Row
         umb = db.execute("SELECT * FROM Umbrellas WHERE id = ?", umbid)
         db.close()
@@ -187,7 +202,7 @@ class Umbrellas:
 
     def mark_overdue(self, umbid) -> None:
         """When an umbrella is overdue, change its status to "overdue"."""
-        db = sqlite3.connect(path)
+        db = sqlite3.connect(self.path)
         db.row_factory = sqlite3.Row
         umb = db.execute("SELECT * FROM Umbrellas WHERE id = ?", umbid)
 
-- 
cgit v1.2.3