From 102640f14de8e382c3eb6cb5975eba194e86c23b Mon Sep 17 00:00:00 2001
From: Mo8it <mo8it@proton.me>
Date: Sun, 15 May 2022 19:07:08 +0200
Subject: [PATCH] Use handle_view_exception

---
 advlabdb/customClasses.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/advlabdb/customClasses.py b/advlabdb/customClasses.py
index 3f4ceb7..36c5917 100644
--- a/advlabdb/customClasses.py
+++ b/advlabdb/customClasses.py
@@ -134,12 +134,11 @@ class CustomModelView(ModelView):
             model = self.customCreateModel(form)
 
             self.session.add(model)
-
             self.on_model_change(form, model, True)
-
             self.session.commit()
         except Exception as ex:
-            flash(str(ex), "error")
+            if not self.handle_view_exception(ex):
+                flash(str(ex), "error")
 
             self.session.rollback()
         else: