From 16cf984ddc6170c2f81d30d3e7703313bf7051e3 Mon Sep 17 00:00:00 2001
From: Ching <loooching@gmail.com>
Date: Sun, 3 Mar 2024 20:32:28 +0800
Subject: [PATCH] refactor(api): Refactor error handling in add() function

---
 app.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index 1a3e0cc..a358efc 100644
--- a/app.py
+++ b/app.py
@@ -201,7 +201,10 @@ def add():
                 response_data = {"message": "Fail to add new item"}
                 return jsonify(response_data), 400
         except Exception as e:
-            error_message = str(e)
+            if hasattr(e, "message"):
+                error_message = e.message
+            else:
+                error_message = str(e)
             response_data = {"message": error_message}
             return jsonify(response_data), 400