mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
|
from django.db import migrations
|
||
|
|
import utilities.fields
|
||
|
|
import utilities.ordering
|
||
|
|
|
||
|
|
|
||
|
|
def _update_model_names(model):
|
||
|
|
# Update each unique field value in bulk
|
||
|
|
for name in model.objects.values_list('name', flat=True).order_by('name').distinct():
|
||
|
|
model.objects.filter(name=name).update(_name=utilities.ordering.naturalize(name))
|
||
|
|
|
||
|
|
|
||
|
|
def naturalize_sites(apps, schema_editor):
|
||
|
|
_update_model_names(apps.get_model('dcim', 'Site'))
|
||
|
|
|
||
|
|
|
||
|
|
def naturalize_racks(apps, schema_editor):
|
||
|
|
_update_model_names(apps.get_model('dcim', 'Rack'))
|
||
|
|
|
||
|
|
|
||
|
|
def naturalize_devices(apps, schema_editor):
|
||
|
|
_update_model_names(apps.get_model('dcim', 'Device'))
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
('dcim', '0094_device_component_template_ordering'),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AlterModelOptions(
|
||
|
|
name='device',
|
||
|
|
options={'ordering': ('_name', 'pk'), 'permissions': (('napalm_read', 'Read-only access to devices via NAPALM'), ('napalm_write', 'Read/write access to devices via NAPALM'))},
|
||
|
|
),
|
||
|
|
migrations.AlterModelOptions(
|
||
|
|
name='rack',
|
||
|
|
options={'ordering': ('site', 'group', '_name', 'pk')},
|
||
|
|
),
|
||
|
|
migrations.AlterModelOptions(
|
||
|
|
name='site',
|
||
|
|
options={'ordering': ('_name',)},
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='device',
|
||
|
|
name='_name',
|
||
|
|
field=utilities.fields.NaturalOrderingField('target_field', blank=True, max_length=100, naturalize_function=utilities.ordering.naturalize),
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='rack',
|
||
|
|
name='_name',
|
||
|
|
field=utilities.fields.NaturalOrderingField('target_field', blank=True, max_length=100, naturalize_function=utilities.ordering.naturalize),
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='site',
|
||
|
|
name='_name',
|
||
|
|
field=utilities.fields.NaturalOrderingField('target_field', blank=True, max_length=100, naturalize_function=utilities.ordering.naturalize),
|
||
|
|
),
|
||
|
|
migrations.RunPython(
|
||
|
|
code=naturalize_sites
|
||
|
|
),
|
||
|
|
migrations.RunPython(
|
||
|
|
code=naturalize_racks
|
||
|
|
),
|
||
|
|
migrations.RunPython(
|
||
|
|
code=naturalize_devices
|
||
|
|
),
|
||
|
|
]
|