mirror of https://github.com/jumpserver/jumpserver
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
# Generated by Django 4.1.10 on 2023-11-03 08:59
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
def migrate_assets_labels(apps, schema_editor):
|
|
old_label_model = apps.get_model('assets', 'Label')
|
|
new_label_model = apps.get_model('labels', 'Label')
|
|
asset_model = apps.get_model('assets', 'Asset')
|
|
labeled_item_model = apps.get_model('labels', 'LabeledResource')
|
|
|
|
old_labels = old_label_model.objects.all()
|
|
new_labels = []
|
|
old_new_label_map = {}
|
|
for label in old_labels:
|
|
new_label = new_label_model(name=label.name, value=label.value, org_id=label.org_id, id=label.id)
|
|
old_new_label_map[label.id] = new_label
|
|
new_labels.append(new_label)
|
|
new_label_model.objects.bulk_create(new_labels, ignore_conflicts=True)
|
|
|
|
label_relations = asset_model.labels.through.objects.all()
|
|
bulk_size = 1000
|
|
count = 0
|
|
content_type = apps.get_model('contenttypes', 'contenttype').objects.get_for_model(asset_model)
|
|
|
|
while True:
|
|
relations = label_relations[count:count + bulk_size]
|
|
if not relations:
|
|
break
|
|
count += bulk_size
|
|
|
|
tagged_items = []
|
|
for relation in relations:
|
|
new_label = old_new_label_map[relation.label_id]
|
|
tagged_item = labeled_item_model(
|
|
label_id=new_label.id, res_type=content_type,
|
|
res_id=relation.asset_id, org_id=new_label.org_id
|
|
)
|
|
tagged_items.append(tagged_item)
|
|
labeled_item_model.objects.bulk_create(tagged_items, ignore_conflicts=True)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('labels', '0001_initial'),
|
|
('assets', '0125_auto_20231011_1053')
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(migrate_assets_labels),
|
|
]
|