controller/rootfs/api/migrations/0013_migration_permissions_and_certificates.py at 5d564ab2ef291bd9242563d5d39213e16c66aeab · drycc/controller · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Generated by Django 4.2.15 on 2024-09-03 03:48

from django.db import migrations
from guardian.shortcuts import assign_perm, get_users_with_perms, remove_perm
from api.models.app import App, VIEW_APP_PERMISSION, CHANGE_APP_PERMISSION
from api.models.domain import Domain
from api.models.certificate import Certificate


def migration_permission(apps, schema_editor):
    for app in App.objects.all():
        for user in get_users_with_perms(app):
            remove_perm('use_app', user, app)
            assign_perm(VIEW_APP_PERMISSION.codename, user, app)
            assign_perm(CHANGE_APP_PERMISSION.codename, user, app)


def migration_certificate(apps, schema_editor):
    for domain in Domain.objects.all():
        if domain.certificate:
            certificate = domain.certificate
            certificate.pk = None
            certificate.app = domain.app
            certificate.save()
            domain.certificate = certificate
            domain.save()
    Certificate.objects.filter(app=None).delete()


class Migration(migrations.Migration):

    dependencies = [
        ('api', '0012_alter_app_options_rename_procfile_type_domain_ptype_and_more'),
    ]

    operations = [
        migrations.RunPython(migration_permission),
        migrations.RunPython(migration_certificate),
    ]