From 1787a84d3052adea8912343b08443cce4bbc9246 Mon Sep 17 00:00:00 2001
From: Mo8it <mo8it@proton.me>
Date: Thu, 29 Jul 2021 21:13:49 +0200
Subject: [PATCH] Fixed 2 checks

---
 advlabdb/modelViews.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/advlabdb/modelViews.py b/advlabdb/modelViews.py
index 1150316..10af424 100644
--- a/advlabdb/modelViews.py
+++ b/advlabdb/modelViews.py
@@ -34,7 +34,7 @@ from advlabdb.utils import (
     setUserActiveSemester,
     userActiveSemester,
 )
-from advlabdb.exceptions import ModelViewException
+from advlabdb.exceptions import ModelViewException, DataBaseException
 
 
 class UserView(SecureModelView):
@@ -288,7 +288,7 @@ class PartStudentView(SecureModelView):
         return form(get_form_data(), obj=obj)
 
     def on_model_change(self, form, model, is_created):
-        PartStudent.check(model.group, model.group)
+        PartStudent.check(model.group, model.part)
 
     def update_model(self, form, model):
         if form.final_part_mark.data == -1:
@@ -331,9 +331,12 @@ class GroupView(SecureModelView):
             return model
 
     def update_model(self, form, model):
-        Group.check(form.part_students.data, model.program)
-
-        return super().update_model(form, model)
+        try:
+            Group.check(form.part_students.data, model.program)
+        except Exception as ex:
+            flash(str(ex), "error")
+        else:
+            return super().update_model(form, model)
 
 
 class ExperimentView(SecureModelView):