jumpserver/apps/assets/migrations/0097_auto_20220426_1558.py

56 lines
1.8 KiB
Python

# Generated by Django 3.1.14 on 2022-04-26 07:58
from django.db import migrations
from assets.const import AllTypes
def create_internal_platforms(apps, *args):
platform_cls = apps.get_model('assets', 'Platform')
AllTypes.create_or_update_internal_platforms(platform_cls)
def update_user_platforms(apps, *args):
platform_cls = apps.get_model('assets', 'Platform')
AllTypes.update_user_create_platforms(platform_cls)
def migrate_macos_platform(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()
def migrate_connectivity(apps, schema_editor):
db_alias = schema_editor.connection.alias
asset_model = apps.get_model('assets', 'Asset')
asset_model.objects.using(db_alias).filter(connectivity='unknown').update(connectivity='-')
asset_model.objects.using(db_alias).filter(connectivity='failed').update(connectivity='err')
class Migration(migrations.Migration):
dependencies = [
('assets', '0096_auto_20220426_1550'),
]
operations = [
migrations.RunPython(create_internal_platforms),
migrations.RunPython(update_user_platforms),
migrations.RunPython(migrate_macos_platform),
migrations.RunPython(migrate_connectivity),
]