mirror of https://github.com/jumpserver/jumpserver
commit
8a3a9c87a8
|
@ -1,66 +0,0 @@
|
||||||
# Generated by Django 3.2.16 on 2023-04-11 10:14
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
|
||||||
|
|
||||||
from assets.const import AllTypes
|
|
||||||
|
|
||||||
|
|
||||||
def migrate_automation_push_account_params(apps, schema_editor):
|
|
||||||
platform_automation_model = apps.get_model('assets', 'PlatformAutomation')
|
|
||||||
platform_automation_methods = AllTypes.get_automation_methods()
|
|
||||||
methods_id_data_map = {
|
|
||||||
i['id']: None if i['params_serializer'] is None else i['params_serializer']({}).data
|
|
||||||
for i in platform_automation_methods
|
|
||||||
if i['method'] == 'push_account'
|
|
||||||
}
|
|
||||||
automation_objs = []
|
|
||||||
for automation in platform_automation_model.objects.all():
|
|
||||||
push_account_method = automation.push_account_method
|
|
||||||
if not push_account_method:
|
|
||||||
continue
|
|
||||||
value = methods_id_data_map.get(push_account_method)
|
|
||||||
if value is None:
|
|
||||||
continue
|
|
||||||
automation.push_account_params = value
|
|
||||||
automation_objs.append(automation)
|
|
||||||
platform_automation_model.objects.bulk_update(automation_objs, ['push_account_params'])
|
|
||||||
|
|
||||||
|
|
||||||
class Migration(migrations.Migration):
|
|
||||||
dependencies = [
|
|
||||||
('assets', '0112_auto_20230404_1631'),
|
|
||||||
]
|
|
||||||
|
|
||||||
operations = [
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='change_secret_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Change secret params'),
|
|
||||||
),
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='gather_accounts_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Gather facts params'),
|
|
||||||
),
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='gather_facts_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Gather facts params'),
|
|
||||||
),
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='ping_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Ping params'),
|
|
||||||
),
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='push_account_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Push account params'),
|
|
||||||
),
|
|
||||||
migrations.AddField(
|
|
||||||
model_name='platformautomation',
|
|
||||||
name='verify_account_params',
|
|
||||||
field=models.JSONField(default=dict, verbose_name='Verify account params'),
|
|
||||||
),
|
|
||||||
migrations.RunPython(migrate_automation_push_account_params),
|
|
||||||
]
|
|
|
@ -2,11 +2,33 @@
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
from assets.const import AllTypes
|
||||||
|
|
||||||
|
|
||||||
|
def migrate_automation_push_account_params(apps, schema_editor):
|
||||||
|
platform_automation_model = apps.get_model('assets', 'PlatformAutomation')
|
||||||
|
platform_automation_methods = AllTypes.get_automation_methods()
|
||||||
|
methods_id_data_map = {
|
||||||
|
i['id']: None if i['params_serializer'] is None else i['params_serializer']({}).data
|
||||||
|
for i in platform_automation_methods
|
||||||
|
if i['method'] == 'push_account'
|
||||||
|
}
|
||||||
|
automation_objs = []
|
||||||
|
for automation in platform_automation_model.objects.all():
|
||||||
|
push_account_method = automation.push_account_method
|
||||||
|
if not push_account_method:
|
||||||
|
continue
|
||||||
|
value = methods_id_data_map.get(push_account_method)
|
||||||
|
if value is None:
|
||||||
|
continue
|
||||||
|
automation.push_account_params = value
|
||||||
|
automation_objs.append(automation)
|
||||||
|
platform_automation_model.objects.bulk_update(automation_objs, ['push_account_params'])
|
||||||
|
|
||||||
|
|
||||||
class Migration(migrations.Migration):
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
dependencies = [
|
dependencies = [
|
||||||
('assets', '0113_auto_20230411_1814'),
|
('assets', '0113_auto_20230411_1917'),
|
||||||
]
|
]
|
||||||
|
|
||||||
operations = [
|
operations = [
|
||||||
|
@ -15,4 +37,35 @@ class Migration(migrations.Migration):
|
||||||
name='params',
|
name='params',
|
||||||
field=models.JSONField(default=dict, verbose_name='Params'),
|
field=models.JSONField(default=dict, verbose_name='Params'),
|
||||||
),
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='change_secret_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Change secret params'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='gather_accounts_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Gather facts params'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='gather_facts_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Gather facts params'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='ping_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Ping params'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='push_account_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Push account params'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='platformautomation',
|
||||||
|
name='verify_account_params',
|
||||||
|
field=models.JSONField(default=dict, verbose_name='Verify account params'),
|
||||||
|
),
|
||||||
|
migrations.RunPython(migrate_automation_push_account_params),
|
||||||
]
|
]
|
||||||
|
|
Loading…
Reference in New Issue