diff options
Diffstat (limited to 'jimbrella/umbrellas.py')
| -rw-r--r-- | jimbrella/umbrellas.py | 41 |
1 files changed, 28 insertions, 13 deletions
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) |