# 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) ]