Updating dependencies check in account deletion

Zambom
1 parent ed2cd2be
Showing 2 changed files with 27 additions and 1 deletions Show diff stats
users/utils.py
users/views.py
@@ -0,0 +1,25 @@
+#    File used to store useful user functions    #
+
+from categories.models import Category
+from subjects.models import Subject
+
+def has_dependencies(user):
+	if user.is_staff: #Check admin function
+		return True
+
+	cats = Category.objects.filter(coordinators = user)
+
+	if len(cats) > 0: #Check coordinator function
+		return True
+
+	subs = Subject.objects.filter(professor = user)
+
+	if len(subs) > 0: #Check professor function
+		return True
+
+	subs = Subject.objects.filter(students = user)
+
+	if len(subs) > 0: #Check student function
+		return True
+
+	return False
 \ No newline at end of file
@@ -10,6 +10,7 @@ from django.db.models import Q
 from braces import views as braces_mixins
 from .models import User
+from .utils import has_dependencies
 from .forms import RegisterUserForm, ProfileForm, UserForm, ChangePassForm, PassResetRequest, SetPasswordForm
 #RECOVER PASS IMPORTS
@@ -168,7 +169,7 @@ class DeleteView(braces_mixins.LoginRequiredMixin, generic.DeleteView):
 		success_msg = _('User removed successfully!')
 		error_msg = _('Could not remove the account. The user is attach to one or more functions (administrator, coordinator, professor ou student) in the system.')
-		if user.has_dependencies():
+		if has_dependencies(user):
 			messages.error(self.request, error_msg)
 			return redirect(error_url)
	@@ -0,0 +1,25 @@		@@ -0,0 +1,25 @@
		1	+# File used to store useful user functions #
		2	+
		3	+from categories.models import Category
		4	+from subjects.models import Subject
		5	+
		6	+def has_dependencies(user):
		7	+ if user.is_staff: #Check admin function
		8	+ return True
		9	+
		10	+ cats = Category.objects.filter(coordinators = user)
		11	+
		12	+ if len(cats) > 0: #Check coordinator function
		13	+ return True
		14	+
		15	+ subs = Subject.objects.filter(professor = user)
		16	+
		17	+ if len(subs) > 0: #Check professor function
		18	+ return True
		19	+
		20	+ subs = Subject.objects.filter(students = user)
		21	+
		22	+ if len(subs) > 0: #Check student function
		23	+ return True
		24	+
		25	+ return False
0	\ No newline at end of file	26	\ No newline at end of file