From d716d29c0a3ec61a12a29ece93578e2d52c1cb79 Mon Sep 17 00:00:00 2001 From: Arthur Hanson Date: Thu, 19 Sep 2024 13:40:05 -0700 Subject: [PATCH 1/2] 17419 rebuild module bay tree on upgrade --- .../migrations/0191_module_bay_rebuild.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 netbox/dcim/migrations/0191_module_bay_rebuild.py diff --git a/netbox/dcim/migrations/0191_module_bay_rebuild.py b/netbox/dcim/migrations/0191_module_bay_rebuild.py new file mode 100644 index 0000000000..86a57c300e --- /dev/null +++ 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 + ), + ] From b469e07ff9a6057c89f13b771c6bd28c5f4c2250 Mon Sep 17 00:00:00 2001 From: Arthur Hanson Date: Thu, 19 Sep 2024 14:24:08 -0700 Subject: [PATCH 2/2] 17419 rebuild module bay tree on upgrade --- netbox/dcim/migrations/0191_module_bay_rebuild.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/netbox/dcim/migrations/0191_module_bay_rebuild.py b/netbox/dcim/migrations/0191_module_bay_rebuild.py index 86a57c300e..7944c31aa3 100644 --- a/netbox/dcim/migrations/0191_module_bay_rebuild.py +++ b/netbox/dcim/migrations/0191_module_bay_rebuild.py @@ -2,7 +2,7 @@ def rebuild_mptt(apps, schema_editor): - ModuleBay = apps.get_model('dcim', 'ModuleBay') + from dcim.models import ModuleBay ModuleBay.objects.rebuild()