Coverage for product_risk_suite/threat_model/migrations/0003_threatmodelconnectionname_id.py: 78%

1# Generated by Django 6.0.5 on 2026-05-13 22:24 

2 

3from django.db import migrations, models 

4 

5def migrate_assign_ids(apps,schema_editor): 

6 ThreatModelConnectionName = apps.get_model('threat_model', 'ThreatModelConnectionName') 

7 for n, obj in enumerate(ThreatModelConnectionName.objects.all()): 

8 obj.id = n 

9 obj.save() 

10 

11 

12class Migration(migrations.Migration): 

13 

14 dependencies = [ 

15 ('threat_model', '0002_threatmodelconnectionname_human_name'), 

16 ] 

17 

18 operations = [ 

19 migrations.AddField( 

20 model_name='threatmodelconnectionname', 

21 name='id', 

22 field=models.IntegerField(default=0), 

23 ), 

24 migrations.RunPython(migrate_assign_ids), 

25 ]