fix migration

This commit is contained in:
Arthur
2026-01-21 12:28:01 -08:00
parent 46f9a12a87
commit 6a5aced4bc

View File

@@ -1,6 +1,4 @@
from django.db import migrations from django.db import migrations
import mptt
import mptt.managers
def rebuild_mptt(apps, schema_editor): def rebuild_mptt(apps, schema_editor):
@@ -8,12 +6,8 @@ def rebuild_mptt(apps, schema_editor):
Rebuild the MPTT tree for ModuleBay to apply new ordering by 'name' Rebuild the MPTT tree for ModuleBay to apply new ordering by 'name'
instead of 'module'. instead of 'module'.
""" """
manager = mptt.managers.TreeManager()
ModuleBay = apps.get_model('dcim', 'ModuleBay') ModuleBay = apps.get_model('dcim', 'ModuleBay')
manager.model = ModuleBay ModuleBay.objects.rebuild()
mptt.register(ModuleBay)
manager.contribute_to_class(ModuleBay, 'objects')
manager.rebuild()
class Migration(migrations.Migration): class Migration(migrations.Migration):