Skip to content

Commit a7af0b4

Browse files
committed
fix: delete AdvisorySet associated with malformed aliases
Signed-off-by: Keshav Priyadarshi <git@keshav.space>
1 parent 6973a54 commit a7af0b4

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

vulnerabilities/migrations/0124_advisoryv2_remove_malformed_aliases_and_dvisory_id.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ class Migration(migrations.Migration):
1010
def drop_malformed_advisory_v2(apps, _):
1111
AdvisoryV2 = apps.get_model("vulnerabilities", "AdvisoryV2")
1212
AdvisoryAlias = apps.get_model("vulnerabilities", "AdvisoryAlias")
13+
AdvisorySet = apps.get_model("vulnerabilities", "AdvisorySet")
1314

1415
valid_alias_prefix = [
1516
"cve-", "osv-", "xsa-", "vsv", "zbx-", "zf2", "vu#", "gms-", "usn-",
@@ -33,10 +34,14 @@ def drop_malformed_advisory_v2(apps, _):
3334
).exclude(query).values_list('id', flat=True).distinct()
3435
)
3536

36-
AdvisoryV2.objects.filter(
37+
advisories_to_delete=AdvisoryV2.objects.filter(
3738
datasource_id__in=target_importers,
3839
aliases__id__in=malformed_alias_ids
39-
).delete()
40+
)
41+
42+
AdvisorySet.objects.filter(primary_advisory__in=advisories_to_delete).delete()
43+
44+
advisories_to_delete.delete()
4045

4146
AdvisoryAlias.objects.filter(
4247
id__in=malformed_alias_ids,

0 commit comments

Comments
 (0)