mirror of https://github.com/jumpserver/jumpserver
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
# 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)
|
|
]
|