mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
|
from django.db import migrations, models
|
||
|
|
import django.db.models.deletion
|
||
|
|
|
||
|
|
|
||
|
|
def device_to_generic_assignment(apps, schema_editor):
|
||
|
|
ContentType = apps.get_model('contenttypes', 'ContentType')
|
||
|
|
Secret = apps.get_model('secrets', 'Secret')
|
||
|
|
|
||
|
|
device_ct = ContentType.objects.get(app_label='dcim', model='device')
|
||
|
|
Secret.objects.update(assigned_object_type=device_ct, assigned_object_id=models.F('device_id'))
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
|
||
|
|
dependencies = [
|
||
|
|
('contenttypes', '0002_remove_content_type_name'),
|
||
|
|
('secrets', '0010_custom_field_data'),
|
||
|
|
]
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AlterModelOptions(
|
||
|
|
name='secret',
|
||
|
|
options={'ordering': ('role', 'name', 'pk')},
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='secret',
|
||
|
|
name='assigned_object_id',
|
||
|
|
field=models.PositiveIntegerField(blank=True, null=True),
|
||
|
|
preserve_default=False,
|
||
|
|
),
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='secret',
|
||
|
|
name='assigned_object_type',
|
||
|
|
field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, to='contenttypes.contenttype'),
|
||
|
|
preserve_default=False,
|
||
|
|
),
|
||
|
|
migrations.AlterUniqueTogether(
|
||
|
|
name='secret',
|
||
|
|
unique_together={('assigned_object_type', 'assigned_object_id', 'role', 'name')},
|
||
|
|
),
|
||
|
|
migrations.RunPython(
|
||
|
|
code=device_to_generic_assignment,
|
||
|
|
reverse_code=migrations.RunPython.noop
|
||
|
|
),
|
||
|
|
migrations.RemoveField(
|
||
|
|
model_name='secret',
|
||
|
|
name='device',
|
||
|
|
),
|
||
|
|
]
|