Skip to content

Commit 70f778c

Browse files
committed
Fix models for V2 importers
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 19487ec commit 70f778c

File tree

2 files changed

+24
-3
lines changed

2 files changed

+24
-3
lines changed

vulnerabilities/migrations/0106_alter_advisoryseverity_value_and_more.py renamed to vulnerabilities/migrations/0106_alter_advisoryreference_url_and_more.py

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Generated by Django 4.2.25 on 2025-12-18 14:37
1+
# Generated by Django 4.2.25 on 2025-12-19 08:31
22

33
from django.db import migrations, models
44

@@ -10,13 +10,33 @@ class Migration(migrations.Migration):
1010
]
1111

1212
operations = [
13+
migrations.AlterField(
14+
model_name="advisoryreference",
15+
name="url",
16+
field=models.URLField(help_text="URL to the vulnerability reference", max_length=1024),
17+
),
1318
migrations.AlterField(
1419
model_name="advisoryseverity",
1520
name="value",
1621
field=models.CharField(
1722
help_text="Example: 9.0, Important, High", max_length=50, null=True
1823
),
1924
),
25+
migrations.AlterField(
26+
model_name="advisoryweakness",
27+
name="cwe_id",
28+
field=models.IntegerField(help_text="CWE id", unique=True),
29+
),
30+
migrations.AlterUniqueTogether(
31+
name="advisoryreference",
32+
unique_together={("url", "reference_type")},
33+
),
34+
migrations.AlterUniqueTogether(
35+
name="advisoryseverity",
36+
unique_together={
37+
("url", "scoring_system", "value", "scoring_elements", "published_at")
38+
},
39+
),
2040
migrations.AddConstraint(
2141
model_name="advisoryseverity",
2242
constraint=models.CheckConstraint(

vulnerabilities/models.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2592,6 +2592,7 @@ class AdvisorySeverity(models.Model):
25922592
class Meta:
25932593
verbose_name_plural = "Advisory severities"
25942594
ordering = ["url", "scoring_system", "value"]
2595+
unique_together = ("url", "scoring_system", "value", "scoring_elements", "published_at")
25952596
constraints = [
25962597
models.CheckConstraint(
25972598
check=(
@@ -2622,7 +2623,7 @@ class AdvisoryWeakness(models.Model):
26222623
A weakness is a software weakness that is associated with a vulnerability.
26232624
"""
26242625

2625-
cwe_id = models.IntegerField(help_text="CWE id")
2626+
cwe_id = models.IntegerField(help_text="CWE id", unique=True)
26262627

26272628
cwe_by_id = {}
26282629

@@ -2669,7 +2670,6 @@ class AdvisoryReference(models.Model):
26692670
url = models.URLField(
26702671
max_length=1024,
26712672
help_text="URL to the vulnerability reference",
2672-
unique=True,
26732673
)
26742674

26752675
ADVISORY = "advisory"
@@ -2699,6 +2699,7 @@ class AdvisoryReference(models.Model):
26992699

27002700
class Meta:
27012701
ordering = ["reference_id", "url", "reference_type"]
2702+
unique_together = ("url", "reference_type")
27022703

27032704
def __str__(self):
27042705
reference_id = f" {self.reference_id}" if self.reference_id else ""

0 commit comments

Comments
 (0)