Define application-specific exceptions

1 files changed, 5 insertions, 4 deletions

diff --git a/jimbrella/database.py b/jimbrella/database.py
index 014b5b5..63cba49 100644
--- a/jimbrella/database.py
+++ b/jimbrella/database.py
@@ -3,6 +3,7 @@ import os
 from datetime import datetime, timedelta
 from .utils import human_datetime, human_timedelta
 from .config import DUE_HOURS
+from .exceptions import *
 
 
 class Database:
@@ -201,9 +202,9 @@ class Database:
         """When a user has borrowed an umbrella."""
         umb = self._find_by_serial(serial)
         if umb is None:
-            raise ValueError(f"No umbrella with serial {serial} found.")
+            raise UmbrellaNotFoundError
         elif umb["status"] != "available":
-            raise ValueError(f"Umbrella with serial {serial} is inavailable.")
+            raise UmbrellaStatusError
         umb["status"] = "lent"
         umb["tenant_name"] = tenant_name
         umb["tenant_id"] = tenant_id
@@ -216,9 +217,9 @@ class Database:
         """When a user has returned an umbrella."""
         umb = self._find_by_serial(serial)
         if umb is None:
-            raise ValueError(f"No umbrella with serial {serial} found.")
+            raise UmbrellaNotFoundError
         elif umb["status"] not in ("lent", "overdue"):
-            raise ValueError(f"Umbrella with serial {serial} is not lent out.")
+            raise UmbrellaStatusError
         umb["status"] = "available"
         for key in ["tenant_name", "tenant_id", "tenant_phone", "tenant_email"]:
             umb[key] = ""