17419 rebuild module bay tree on upgrade

2025-08-05 07:08:16 -06:00 · 2024-09-19 13:40:05 -07:00 · 2024-09-19 13:40:05 -07:00 · d716d29c0a
commit d716d29c0a
parent 6d0a3485a1
1 changed files with 20 additions and 0 deletions
--- a/netbox/dcim/migrations/0191_module_bay_rebuild.py
+++ b/netbox/dcim/migrations/0191_module_bay_rebuild.py
@ -0,0 +1,20 @@
+from django.db import migrations
+
+
+def rebuild_mptt(apps, schema_editor):
+    ModuleBay = apps.get_model('dcim', 'ModuleBay')
+    ModuleBay.objects.rebuild()
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('dcim', '0190_nested_modules'),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            code=rebuild_mptt,
+            reverse_code=migrations.RunPython.noop
+        ),
+    ]