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

Add winery reviewer

Browse Source
This commit is contained in:
Виктор Гладких 2019-12-09 09:38:18 +03:00
parent 8e63b969bf
commit af7aef6d23
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
apps/utils/permissions.py
View File

@ -423,15 +423,28 @@ class IsWineryReviewer(IsStandardUser):
rules = [
super().has_object_permission(request, view, obj)
]
if hasattr(obj, 'type_id'):
est = EstablishmentSubType.objects.filter(establishment_type_id=obj.type_id)
# AttributeError: 'Establishment' object has no attribute 'object_id'
if hasattr(obj, 'type_id') or hasattr(obj, 'establishment_type_id'):
type_id: int
if hasattr(obj, 'type_id'):
type_id = obj.type_id
else:
type_id = obj.establishment_type_id
est = EstablishmentSubType.objects.filter(establishment_type_id=type_id)
role = Role.objects.filter(role=Role.WINERY_REVIEWER,
establishment_subtype_id__in=[id for type.id in est],
country_id=obj.country_id).first()
object_id: int
if hasattr(obj, 'object_id'):
object_id = obj.object_id
else:
object_id = obj.establishment_id
rules = [
UserRole.objects.filter(user=request.user, role=role,
establishment_id=obj.object_id
establishment_id=object_id
).exists(),
super().has_object_permission(request, view, obj)
]