peeringdb-peeringdb/peeringdb_server/migrations/0044_ixlan_ixf_fields.py

# Generated by Django 2.2.13 on 2020-07-13 07:17

from django.db import migrations, models
from django.conf import settings
from django_namespace_perms.constants import PERM_CRUD, PERM_READ


def add_permissions(apps, schema_editor):
    Group = apps.get_model("auth", "Group")
    Organization = apps.get_model("peeringdb_server", "Organization")
    GroupPermission = apps.get_model("django_namespace_perms", "GroupPermission")

    guest_group = Group.objects.filter(id=settings.GUEST_GROUP_ID).first()
    user_group = Group.objects.filter(id=settings.USER_GROUP_ID).first()

    namespace = (
        "peeringdb.organization.{org}.internetexchange"
        ".*.ixf_ixp_member_list_url.{visible}"
    )

    if guest_group:
        GroupPermission.objects.create(
            group=guest_group,
            namespace=namespace.format(org="*", visible="public"),
            permissions=PERM_READ,
        )

    if user_group:
        GroupPermission.objects.create(
            group=user_group,
            namespace=namespace.format(org="*", visible="public"),
            permissions=PERM_READ,
        )

        GroupPermission.objects.create(
            group=user_group,
            namespace=namespace.format(org="*", visible="users"),
            permissions=PERM_READ,
        )

    for org in Organization.handleref.all():

        GroupPermission.objects.create(
            group=Group.objects.get(name=f"org.{org.id}.admin"),
            namespace=namespace.format(org=org.id, visible="private"),
            permissions=PERM_CRUD,
        )
        GroupPermission.objects.create(
            group=Group.objects.get(name=f"org.{org.id}"),
            namespace=namespace.format(org=org.id, visible="private"),
            permissions=PERM_READ,
        )


class Migration(migrations.Migration):

    dependencies = [
        ("peeringdb_server", "0043_ix_ixf_fields"),
    ]

    operations = [
        migrations.RunPython(add_permissions),
        migrations.AddField(
            model_name="ixlan",
            name="ixf_ixp_member_list_url_visible",
            field=models.CharField(
                choices=[
                    ("Private", "Private"),
                    ("Users", "Users"),
                    ("Public", "Public"),
                ],
                default="Private",
                max_length=64,
                verbose_name="IX-F Member Export URL Visibility",
            ),
        ),
    ]