mirror of
https://github.com/django-helpdesk/django-helpdesk.git
synced 2025-06-03 00:15:46 +02:00
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.db import migrations
|
|
from django.db.utils import IntegrityError
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
def create_and_assign_permissions(apps, schema_editor):
|
|
db_alias = schema_editor.connection.alias
|
|
Permission = apps.get_model("auth", "Permission")
|
|
ContentType = apps.get_model("contenttypes", "ContentType")
|
|
# Two steps:
|
|
# 1. Create the permission for existing Queues
|
|
# 2. Assign the permission to user according to QueueMembership objects
|
|
|
|
# First step: prepare the permission for each queue
|
|
Queue = apps.get_model("helpdesk", "Queue")
|
|
for q in Queue.objects.using(db_alias).all():
|
|
if not q.permission_name:
|
|
basename = "queue_access_%s" % q.slug
|
|
q.permission_name = "helpdesk.%s" % basename
|
|
else:
|
|
# Strip the `helpdesk.` prefix
|
|
basename = q.permission_name[9:]
|
|
|
|
try:
|
|
Permission.objects.using(db_alias).create(
|
|
name=_("Permission for queue: ") + q.title,
|
|
content_type=ContentType.objects.using(db_alias).get(model="queue"),
|
|
codename=basename,
|
|
)
|
|
except IntegrityError:
|
|
# Seems that it already existed, safely ignore it
|
|
pass
|
|
q.save()
|
|
|
|
# Second step: map the permissions according to QueueMembership
|
|
QueueMembership = apps.get_model("helpdesk", "QueueMembership")
|
|
for qm in QueueMembership.objects.using(db_alias).all():
|
|
user = qm.user
|
|
for q in qm.queues.all():
|
|
# Strip the `helpdesk.` prefix
|
|
p = Permission.objects.using(db_alias).get(codename=q.permission_name[9:])
|
|
user.user_permissions.add(p)
|
|
qm.delete()
|
|
|
|
|
|
def revert_queue_membership(apps, schema_editor):
|
|
db_alias = schema_editor.connection.alias
|
|
Permission = apps.get_model("auth", "Permission")
|
|
Queue = apps.get_model("helpdesk", "Queue")
|
|
QueueMembership = apps.get_model("helpdesk", "QueueMembership")
|
|
for p in Permission.objects.using(db_alias).all():
|
|
if p.codename.startswith("queue_access_"):
|
|
slug = p.codename[13:]
|
|
try:
|
|
q = Queue.objects.using(db_alias).get(slug=slug)
|
|
except ObjectDoesNotExist:
|
|
continue
|
|
|
|
for user in p.user_set.all():
|
|
qm, _ = QueueMembership.objects.using(db_alias).get_or_create(user=user)
|
|
qm.queues.add(q)
|
|
|
|
p.delete()
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
dependencies = [
|
|
("helpdesk", "0008_extra_for_permissions"),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(create_and_assign_permissions, revert_queue_membership)
|
|
]
|