mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
from django.db import migrations, models
|
|
import django.db.models.deletion
|
|
|
|
|
|
def set_assigned_object_type(apps, schema_editor):
|
|
ContentType = apps.get_model('contenttypes', 'ContentType')
|
|
IPAddress = apps.get_model('ipam', 'IPAddress')
|
|
|
|
device_ct = ContentType.objects.get(app_label='dcim', model='interface').pk
|
|
IPAddress.objects.update(assigned_object_type=device_ct)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('contenttypes', '0002_remove_content_type_name'),
|
|
('ipam', '0036_standardize_description'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RenameField(
|
|
model_name='ipaddress',
|
|
old_name='interface',
|
|
new_name='assigned_object_id',
|
|
),
|
|
migrations.AlterField(
|
|
model_name='ipaddress',
|
|
name='assigned_object_id',
|
|
field=models.PositiveIntegerField(blank=True, null=True),
|
|
),
|
|
migrations.AddField(
|
|
model_name='ipaddress',
|
|
name='assigned_object_type',
|
|
field=models.ForeignKey(blank=True, limit_choices_to=models.Q(models.Q(models.Q(('app_label', 'dcim'), ('model', 'interface')), models.Q(('app_label', 'virtualization'), ('model', 'vminterface')), _connector='OR')), null=True, on_delete=django.db.models.deletion.PROTECT, related_name='+', to='contenttypes.ContentType'),
|
|
preserve_default=False,
|
|
),
|
|
migrations.RunPython(
|
|
code=set_assigned_object_type
|
|
),
|
|
]
|