В настоящее время я работаю над проектом, в котором одна комната имеет собственные разрешения для группы.
Когда кто-то создает новую комнату, система будет делать следующее:
Room Owner#room#{pk}
Вот моя модель комнаты:
class Room(models.Model):
# ...fields
PERMISSION_GROUPS = [
'owner': [
('view_room', _('...')),
('change_room', _('...')),
('delete_room', _('...')),
('open_room', _('...')),
('close_room', _('...'))
],
'staff': [
('view_room', _('...')),
('open_room', _('...')),
('close_room', _('...'))
]
]
Это мой код для создания комнаты и создания групповых разрешений:
from guardian.shortcuts import assign_perm
class CreateRoomView(generics.CreateAPIView):
def _get_permission_string(self, perm):
"""
Get string permission, the return is like 'room.add_room'
"""
return f'room.{perm.codename}'
def post(self):
# Create a room
body = {}
room = Room.objects.create(**body)
# Get content type
ctype = ContentType.objects.get_for_model(Room)
# Define name of owner and staff group for this room permission
owner_group_name = _('Room Owner#room#{pk}')
staff_group_name = _('Room Staff#room#{pk}')
# Create group
owner_group = Group.objects.create(
name=owner_group_name.format(pk=room.pk))
staff_group = Group.objects.create(
name=staff_group_name.format(pk=room.pk))
# Get permissions by group permissions
owner_perms = Permission.objects.filter(
codename__in=Room.PERMISSION_GROUPS.get('owner'), content_type=ctype)
staff_perms = Permission.objects.filter(
codename__in=Room.PERMISSION_GROUPS.get('staff'), content_type=ctype)
# Assign owner permissions to the group
for owner_perm in owner_perms:
perm = self._get_permission_string(owner_perm)
assign_perm(perm, owner_group, obj=room)
# Assign staff permissions to the group
for staff_perm in staff_perms:
perm = self._get_permission_string(staff_perm)
assign_perm(perm, staff_group, obj=room)
# Assign owner group permission to the user
request.user.groups.add(owner_group)
Когда я смотрю журналы запросов, в этом процессе выполняется много запросов.
На самом деле есть еще несколько групп и процесс назначения, как указано выше, но я включил только несколько примеров, поэтому их не слишком много. Если подсчитать, в этом процессе может выполняться более 50 запросов .
Итак, как я могу оптимизировать запрос для этой проблемы? Может кто поделится, как это сделать лучше.
Спасибо,
Используйте прямые внешние ключи, см. Https://django-guardian.readthedocs.io/en/stable/userguide/performance.html#direct -foreign-keys.
Вы можете передавать экземпляры Permission
в assign_perm
. perm = self._get_permission_string(owner_perm)
приводит к дополнительному запросу на разрешение.
Если этого недостаточно, вы можете назначить разрешения сразу, см. Код ниже. Но имейте в виду, что я не тестировал этот код. Может потребоваться корректировка, а может вообще не работать.
До:
# Assign owner permissions to the group
for owner_perm in owner_perms:
perm = self._get_permission_string(owner_perm)
assign_perm(perm, owner_group, obj=room)
# Assign staff permissions to the group
for staff_perm in staff_perms:
perm = self._get_permission_string(staff_perm)
assign_perm(perm, staff_group, obj=room)
После:
# RoomGroupObjectPermission is the model from suggestion 1, use direct foreign keys
room_group_object_permissions = [
RoomGroupObjectPermission(
permission=owner_perm, group=owner_group, content_object=room,
)
for owner_perm in owner_perms
] + [
RoomGroupObjectPermission(
permission=staff_perm, group=staff_group, content_object=room,
)
for staff_perm in staff_perms
]
RoomGroupObjectPermission.objects.bulk_create(room_group_object_permissions)
Мой Бог. Я скучал по этому. Решил воспользоваться первым. Спасибо,
Самый полезный комментарий
Используйте прямые внешние ключи, см. Https://django-guardian.readthedocs.io/en/stable/userguide/performance.html#direct -foreign-keys.
Вы можете передавать экземпляры
Permission
вassign_perm
.perm = self._get_permission_string(owner_perm)
приводит к дополнительному запросу на разрешение.Если этого недостаточно, вы можете назначить разрешения сразу, см. Код ниже. Но имейте в виду, что я не тестировал этот код. Может потребоваться корректировка, а может вообще не работать.
До:
После: