jumpserver/apps/assets/migrations/0124_auto_20231007_1437.py

# Generated by Django 4.1.10 on 2023-10-07 06:37

from django.db import migrations


def add_db2_platform(apps, schema_editor):
    platform_cls = apps.get_model('assets', 'Platform')
    automation_cls = apps.get_model('assets', 'PlatformAutomation')
    platform, _ = platform_cls.objects.update_or_create(
        name='DB2', defaults={
            'name': 'DB2', 'category': 'database',
            'internal': True, 'type': 'db2',
            'domain_enabled': True, 'su_enabled': False,
            'su_method': None, 'comment': 'DB2', 'created_by': 'System',
            'updated_by': 'System', 'custom_fields': []
        }
    )
    platform.protocols.update_or_create(name='db2', defaults={
        'name': 'db2', 'port': 50000, 'primary': True, 'setting': {}
    })
    automation_cls.objects.update_or_create(platform=platform, defaults={'ansible_enabled': False})


class Migration(migrations.Migration):
    dependencies = [
        ('assets', '0123_device_automation_ansible_enabled'),
    ]

    operations = [
        migrations.RunPython(add_db2_platform)
    ]