CaZzzer/access-controller
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Refactored some functions in AdminPageView

Browse Source
This commit is contained in:
Yuriy Kulakov 2021-03-12 13:01:41 +03:00
parent 72b70cc585
commit abe44fec5f
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main/views.py
View File

@ -184,10 +184,11 @@ class AdminPageView(LoginRequiredMixin, PermissionRequiredMixin, FormView):
""" """
Функция установки ролей пользователям Функция установки ролей пользователям
""" """
users = form.cleaned_data['users']
if 'engineer' in self.request.POST: if 'engineer' in self.request.POST:
self.make_engineers(form.cleaned_data['users']) self.make_engineers(users)
elif 'light_agent' in self.request.POST: elif 'light_agent' in self.request.POST:
self.make_light_agents(form.cleaned_data['users']) self.make_light_agents(users)
return super().form_valid(form) return super().form_valid(form)
@staticmethod @staticmethod
@ -202,8 +203,6 @@ class AdminPageView(LoginRequiredMixin, PermissionRequiredMixin, FormView):
""" """
Функция формирования контента страницы администратора (с проверкой прав доступа) Функция формирования контента страницы администратора (с проверкой прав доступа)
""" """
if self.request.user.userprofile.role != 'admin':
raise PermissionDenied
context = super().get_context_data(**kwargs) context = super().get_context_data(**kwargs)
users = get_list_or_404( users = get_list_or_404(
UserProfile, role='agent') UserProfile, role='agent')