mirror of https://github.com/jumpserver/jumpserver
feng
2 years ago
committed by
Jiangjie.Bai
2 changed files with 55 additions and 68 deletions
@ -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), |
|
||||||
] |
|
Loading…
Reference in new issue