mirror of https://github.com/jumpserver/jumpserver
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
|
# Generated by Django 4.1.10 on 2023-08-03 08:28
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.db import migrations, models
|
||
|
|
|
||
|
|
|
||
|
|
def migrate_recipients(apps, schema_editor):
|
||
|
|
account_backup_model = apps.get_model('accounts', 'AccountBackupAutomation')
|
||
|
|
execution_model = apps.get_model('accounts', 'AccountBackupExecution')
|
||
|
|
for account_backup in account_backup_model.objects.all():
|
||
|
|
recipients = list(account_backup.recipients.all())
|
||
|
|
if not recipients:
|
||
|
|
continue
|
||
|
|
account_backup.recipients_part_one.set(recipients)
|
||
|
|
|
||
|
|
execution_bojs = []
|
||
|
|
for execution in execution_model.objects.all():
|
||
|
|
snapshot = execution.snapshot
|
||
|
|
recipients = snapshot.pop('recipients', {})
|
||
|
|
snapshot.update({'recipients_part_one': recipients, 'recipients_part_two': {}})
|
||
|
|
execution_bojs.append(execution)
|
||
|
|
execution_model.objects.bulk_update(execution_bojs, ['snapshot'])
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
dependencies = [
|
||
|
|
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
||
|
|
('accounts', '0012_auto_20230621_1456'),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='accountbackupautomation',
|
||
|
|
name='recipients_part_one',
|
||
|
|
field=models.ManyToManyField(
|
||
|
|
blank=True, related_name='recipient_part_one_plans',
|
||
|
|
to=settings.AUTH_USER_MODEL, verbose_name='Recipient part one'
|
||
|
|
),
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='accountbackupautomation',
|
||
|
|
name='recipients_part_two',
|
||
|
|
field=models.ManyToManyField(
|
||
|
|
blank=True, related_name='recipient_part_two_plans',
|
||
|
|
to=settings.AUTH_USER_MODEL, verbose_name='Recipient part two'
|
||
|
|
),
|
||
|
|
),
|
||
|
|
migrations.RenameField(
|
||
|
|
model_name='accountbackupexecution',
|
||
|
|
old_name='plan_snapshot',
|
||
|
|
new_name='snapshot',
|
||
|
|
),
|
||
|
|
migrations.RunPython(migrate_recipients),
|
||
|
|
migrations.RemoveField(
|
||
|
|
model_name='accountbackupautomation',
|
||
|
|
name='recipients',
|
||
|
|
),
|
||
|
|
|
||
|
|
]
|