mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-16 04:02:52 -06:00
78 lines
1.8 KiB
Python
78 lines
1.8 KiB
Python
from django.db import migrations, models
|
|
|
|
|
|
POWERFEED_TYPE_CHOICES = (
|
|
(1, 'primary'),
|
|
(2, 'redundant'),
|
|
)
|
|
|
|
POWERFEED_SUPPLY_CHOICES = (
|
|
(1, 'ac'),
|
|
(2, 'dc'),
|
|
)
|
|
|
|
POWERFEED_PHASE_CHOICES = (
|
|
(1, 'single-phase'),
|
|
(3, 'three-phase'),
|
|
)
|
|
|
|
|
|
def powerfeed_type_to_slug(apps, schema_editor):
|
|
PowerFeed = apps.get_model('dcim', 'PowerFeed')
|
|
for id, slug in POWERFEED_TYPE_CHOICES:
|
|
PowerFeed.objects.filter(type=id).update(type=slug)
|
|
|
|
|
|
def powerfeed_supply_to_slug(apps, schema_editor):
|
|
PowerFeed = apps.get_model('dcim', 'PowerFeed')
|
|
for id, slug in POWERFEED_SUPPLY_CHOICES:
|
|
PowerFeed.objects.filter(supply=id).update(supply=slug)
|
|
|
|
|
|
def powerfeed_phase_to_slug(apps, schema_editor):
|
|
PowerFeed = apps.get_model('dcim', 'PowerFeed')
|
|
for id, slug in POWERFEED_PHASE_CHOICES:
|
|
PowerFeed.objects.filter(phase=id).update(phase=slug)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
atomic = False
|
|
|
|
dependencies = [
|
|
('dcim', '0083_3569_cable_fields'),
|
|
]
|
|
|
|
operations = [
|
|
|
|
# PowerFeed.type
|
|
migrations.AlterField(
|
|
model_name='powerfeed',
|
|
name='type',
|
|
field=models.CharField(blank=True, max_length=50),
|
|
),
|
|
migrations.RunPython(
|
|
code=powerfeed_type_to_slug
|
|
),
|
|
|
|
# PowerFeed.supply
|
|
migrations.AlterField(
|
|
model_name='powerfeed',
|
|
name='supply',
|
|
field=models.CharField(blank=True, max_length=50),
|
|
),
|
|
migrations.RunPython(
|
|
code=powerfeed_supply_to_slug
|
|
),
|
|
|
|
# PowerFeed.phase
|
|
migrations.AlterField(
|
|
model_name='powerfeed',
|
|
name='phase',
|
|
field=models.CharField(blank=True, max_length=50),
|
|
),
|
|
migrations.RunPython(
|
|
code=powerfeed_phase_to_slug
|
|
),
|
|
|
|
]
|