Database keeps admin log

1 files changed, 6 insertions, 1 deletions

diff --git a/jimbrella/database.py b/jimbrella/database.py
index 9ca3c5b..410985e 100644
--- a/jimbrella/database.py
+++ b/jimbrella/database.py
@@ -1,7 +1,8 @@
 from datetime import datetime, timedelta
 from .csv_table import CsvTable
+from .admin_log import AdminLog
 from .utils import human_datetime, human_timedelta
-from .config import DUE_HOURS
+from .config import DUE_HOURS, ADMIN_LOG_PATH
 from .exceptions import *
 
 STATUSES = ["available", "lent", "overdue", "maintenance", "withheld", "unknown"]
@@ -60,6 +61,7 @@ class Database(CsvTable):
                 },
             ],
         )
+        self.admin_log = AdminLog(ADMIN_LOG_PATH)
 
     def _find_by_serial(self, serial: int) -> dict:
         """Given a serial number, returns an umbrella with such serial.
@@ -208,6 +210,7 @@ class Database(CsvTable):
         umb["tenant_email"] = tenant_email
         umb["lent_at"] = date
         self._update(umb)
+        self.admin_log.log("TAKEAWAY", umb)
 
     def give_back(self, serial) -> None:
         """When a user has returned an umbrella."""
@@ -221,6 +224,7 @@ class Database(CsvTable):
             umb[key] = ""
         umb["lent_at"] = None
         self._update(umb)
+        self.admin_log.log("GIVEBACK", umb)
 
     def mark_overdue(self, serial) -> None:
         """When an umbrella is overdue, change its status to "overdue"."""
@@ -231,3 +235,4 @@ class Database(CsvTable):
             raise UmbrellaStatusError
         umb["status"] = "overdue"
         self._update(umb)
+        self.admin_log.log("OVERDUE", umb)