phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix role counter

Browse Source
This commit is contained in:
Anatoly 2020-01-20 11:43:35 +03:00
parent e6dfa6d74d
commit 765a66dc68
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/account/views/back.py
View File

@ -24,8 +24,9 @@ class RoleTabRetrieveView(generics.GenericAPIView):
"""Overridden get_queryset method."""
additional_filters = {}
if self.request.user.userrole_set.country_admin_role().exists():
additional_filters.update({'userrole__country__code': self.request.country_code})
if (self.request.user.userrole_set.country_admin_role().exists() and
hasattr(self.request.user, 'country_code')):
additional_filters.update({'country__code': self.request.country_code})
return models.Role.objects.filter(**additional_filters)\
.annotate_role_name()\