NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-04 06:38:16 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

17419 use get_model

Browse Source
This commit is contained in:
Arthur Hanson 2024-09-24 14:46:54 -07:00
parent 0dd2a65007
commit b44b12dce9
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
netbox/dcim/migrations/0191_module_bay_rebuild.py
View File

@ -1,9 +1,15 @@
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):
from dcim.models import ModuleBay manager = mptt.managers.TreeManager()
ModuleBay.objects.rebuild() ModuleBay = apps.get_model('dcim', 'ModuleBay')
manager.model = ModuleBay
mptt.register(ModuleBay)
manager.contribute_to_class(ModuleBay, 'objects')
manager.rebuild()
class Migration(migrations.Migration): class Migration(migrations.Migration):