jumpserver/apps/labels/migrations/0002_auto_20231103_1659.py

53 lines
1.7 KiB
Python
Raw Normal View History

# 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:
2024-03-29 07:30:12 +00:00
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),
]