jumpserver/apps/assets/migrations/0102_auto_20220803_1859.py

# Generated by Django 3.2.14 on 2022-08-03 10:59
import time
from django.db import migrations


def migrate_asset_protocols(apps, schema_editor):
    asset_model = apps.get_model('assets', 'Asset')
    protocol_model = apps.get_model('assets', 'Protocol')

    count = 0
    bulk_size = 1000
    print("\n\tStart migrate asset protocols")
    while True:
        start = time.time()
        assets = asset_model.objects.all()[count:count+bulk_size]
        if not assets:
            break
        count += len(assets)
        assets_protocols = []

        for asset in assets:
            old_protocols = asset._protocols or '{}/{}'.format(asset.protocol, asset.port) or 'ssh/22'

            if ',' in old_protocols:
                _protocols = old_protocols.split(',')
            else:
                _protocols = old_protocols.split()

            for name_port in _protocols:
                name_port_list = name_port.split('/')
                if len(name_port_list) != 2:
                    continue

                name, port = name_port_list
                protocol = protocol_model(**{'name': name, 'port': port, 'asset': asset})
                assets_protocols.append(protocol)

        protocol_model.objects.bulk_create(assets_protocols, ignore_conflicts=True)
        print("\t - Create asset protocols: {}-{} using: {:.2f}s".format(
            count - len(assets), count, time.time()-start
        ))


class Migration(migrations.Migration):

    dependencies = [
        ('assets', '0101_auto_20220803_1448'),
    ]

    operations = [
        migrations.RunPython(migrate_asset_protocols)
    ]