re2o/users/migrations/0067_serveurpermission.py

# -*- coding: utf-8 -*-
# Generated by Django 1.10.7 on 2017-12-31 19:53
from __future__ import unicode_literals

from django.db import migrations


class Migration(migrations.Migration):

    dependencies = [("users", "0066_grouppermissions")]

    def transfer_permissions(apps, schema_editor):
        db_alias = schema_editor.connection.alias
        contenttype = apps.get_model("contenttypes", "ContentType")
        rights = apps.get_model("users", "ListRight")
        permissions = apps.get_model("auth", "Permission")
        groups = apps.get_model("auth", "Group")
        machine = apps.get_model("machines", "Machine")
        perm = permissions.objects.using(db_alias).filter(codename="serveur").first()
        if not perm:
            perm = permissions.objects.using(db_alias).create(
                codename="serveur",
                name="Serveur",
                content_type=contenttype.objects.get_for_model(machine),
            )
        group_object = (
            rights.objects.using(db_alias).filter(unix_name="serveur").first()
        )
        if not group_object:
            last_gid = rights.objects.using(db_alias).all().order_by("gid").last().gid
            gid = last_gid + 1
            abstract_group = groups.objects.using(db_alias).create(name="serveur")
            group_object = rights.objects.using(db_alias).create(
                group_ptr=abstract_group, unix_name="serveur", gid=gid
            )
        group_object = group_object.group_ptr
        group_object.permissions.add(perm)
        group_object.save()

    def untransfer_permissions(apps, schema_editor):
        return

    operations = [migrations.RunPython(transfer_permissions, untransfer_permissions)]