jumpserver/apps/assets/migrations/0114_remove_redundant_macos.py

# Generated by Django 3.2.14 on 2022-12-15 07:08

from django.db import migrations


def migrate_del_macos(apps, schema_editor):
    db_alias = schema_editor.connection.alias
    asset_model = apps.get_model('assets', 'Asset')
    platform_model = apps.get_model('assets', 'Platform')
    old_macos = platform_model.objects.using(db_alias).filter(
        name='MacOS', type='macos'
    ).first()
    new_macos = platform_model.objects.using(db_alias).filter(
        name='macOS', type='unix'
    ).first()

    if not old_macos or not new_macos:
        return

    asset_model.objects.using(db_alias).filter(
        platform=old_macos
    ).update(platform=new_macos)

    platform_model.objects.using(db_alias).filter(id=old_macos.id).delete()


class Migration(migrations.Migration):
    dependencies = [
        ('assets', '0113_auto_20221122_2015'),
    ]

    operations = [
        migrations.RunPython(migrate_del_macos),
    ]