mirror of https://github.com/jumpserver/jumpserver
32 lines
924 B
Python
32 lines
924 B
Python
# Generated by Django 3.2.17 on 2023-04-11 11:17
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
def migrate_device_platform_su_method(apps, schema_editor):
|
|
platform_model = apps.get_model('assets', 'Platform')
|
|
device_map = {
|
|
'Huawei': 'super',
|
|
'Cisco': 'enable',
|
|
'H3C': 'super_level',
|
|
}
|
|
platforms = platform_model.objects.filter(name__in=device_map.keys())
|
|
print()
|
|
for platform in platforms:
|
|
print("Migrate platform su method: {}".format(platform.name))
|
|
if platform.name not in device_map:
|
|
continue
|
|
platform.su_method = device_map[platform.name]
|
|
platform.su_enabled = True
|
|
platform.save(update_fields=['su_method', 'su_enabled'])
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
dependencies = [
|
|
('assets', '0112_auto_20230404_1631'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate_device_platform_su_method)
|
|
]
|